diff --git a/backend/pkg/metrics/server.go b/backend/pkg/metrics/server.go index 4800425cd..b889c4dbf 100644 --- a/backend/pkg/metrics/server.go +++ b/backend/pkg/metrics/server.go @@ -1,10 +1,38 @@ package metrics import ( + "context" + "net/http" + "github.com/prometheus/client_golang/prometheus" + "github.com/prometheus/client_golang/prometheus/collectors" + "github.com/prometheus/client_golang/prometheus/promhttp" + "openreplay/backend/pkg/logger" ) -type MetricServer struct{} +type MetricServer struct { + registry *prometheus.Registry +} -func New(log logger.Logger, cs []prometheus.Collector) {} +func New(log logger.Logger, cs []prometheus.Collector) { + registry := prometheus.NewRegistry() + // Add go runtime metrics and process collectors. + registry.MustRegister( + collectors.NewGoCollector(), + collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}), + ) + // Add extra metrics + registry.MustRegister(cs...) + // Expose /metrics HTTP endpoint using the created custom registry. + http.Handle( + "/metrics", promhttp.HandlerFor( + registry, + promhttp.HandlerOpts{ + EnableOpenMetrics: true, + }), + ) + go func() { + log.Error(context.Background(), "%v", http.ListenAndServe(":8888", nil)) + }() +} diff --git a/ee/backend/pkg/metrics/server.go b/ee/backend/pkg/metrics/server.go deleted file mode 100644 index b889c4dbf..000000000 --- a/ee/backend/pkg/metrics/server.go +++ /dev/null @@ -1,38 +0,0 @@ -package metrics - -import ( - "context" - "net/http" - - "github.com/prometheus/client_golang/prometheus" - "github.com/prometheus/client_golang/prometheus/collectors" - "github.com/prometheus/client_golang/prometheus/promhttp" - - "openreplay/backend/pkg/logger" -) - -type MetricServer struct { - registry *prometheus.Registry -} - -func New(log logger.Logger, cs []prometheus.Collector) { - registry := prometheus.NewRegistry() - // Add go runtime metrics and process collectors. - registry.MustRegister( - collectors.NewGoCollector(), - collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}), - ) - // Add extra metrics - registry.MustRegister(cs...) - // Expose /metrics HTTP endpoint using the created custom registry. - http.Handle( - "/metrics", promhttp.HandlerFor( - registry, - promhttp.HandlerOpts{ - EnableOpenMetrics: true, - }), - ) - go func() { - log.Error(context.Background(), "%v", http.ListenAndServe(":8888", nil)) - }() -}