sq

metrics/metrics.go

@@ -2,6 +2,7 @@ package metrics
 
 import (
 	"github.com/prometheus/client_golang/prometheus"
+	"github.com/prometheus/client_golang/prometheus/collectors"
 )
 
 //nolint:gochecknoglobals
@@ -11,3 +12,10 @@ var reg = prometheus.NewRegistry()
 func RegisterMetric(c prometheus.Collector) {
 	_ = reg.Register(c)
 }
+
+func StartCollection() {
+	_ = reg.Register(collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}))
+	_ = reg.Register(collectors.NewGoCollector())
+
+	registerEventListeners()
+}

metrics/service.go

@@ -5,7 +5,6 @@ import (
 	"github.com/0xERR0R/blocky/service"
 	"github.com/0xERR0R/blocky/util"
 	"github.com/go-chi/chi/v5"
-	"github.com/prometheus/client_golang/prometheus/collectors"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
 )
 
@@ -25,11 +24,6 @@ func NewService(cfg config.MetricsService, metricsCfg config.Metrics) *Service {
 		return new(Service)
 	}
 
-	_ = reg.Register(collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}))
-	_ = reg.Register(collectors.NewGoCollector())
-
-	registerEventListeners()
-
 	s := &Service{
 		HTTPInfo: service.HTTPInfo{
 			Info: service.Info{

server/server.go

@@ -141,6 +141,11 @@ func NewServer(ctx context.Context, cfg *config.Config) (server *Server, err err
 		return nil, err
 	}
 
+	if cfg.Prometheus.Enable {
+		// Must happen before resolver initialization
+		metrics.StartCollection()
+	}
+
 	bootstrap, err := resolver.NewBootstrap(ctx, cfg)
 	if err != nil {
 		return nil, err