metrics.go

package metrics

import (
	"github.com/stgnet/blocky/config"

	"github.com/go-chi/chi"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"
)

// nolint
var reg = prometheus.NewRegistry()

// nolint
var enabled bool

func RegisterMetric(c prometheus.Collector) {
	_ = reg.Register(c)
}

func Start(router *chi.Mux, cfg config.PrometheusConfig) {
	enabled = cfg.Enable

	if cfg.Enable {
		_ = reg.Register(prometheus.NewProcessCollector(prometheus.ProcessCollectorOpts{}))
		_ = reg.Register(prometheus.NewGoCollector())
		router.Handle(cfg.Path, promhttp.InstrumentMetricHandler(reg,
			promhttp.HandlerFor(reg, promhttp.HandlerOpts{})))
	}
}

func IsEnabled() bool {
	return enabled
}