metrics.go

package gateway

import (
	"net/http"

	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/collectors"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	"github.com/rancher/opni/pkg/config/v1beta1"
)

type MetricsEndpointHandler struct {
	reg *prometheus.Registry
	cfg v1beta1.MetricsSpec
}

func NewMetricsEndpointHandler(cfg v1beta1.MetricsSpec) *MetricsEndpointHandler {
	reg := prometheus.NewRegistry()
	reg.MustRegister(collectors.NewProcessCollector(collectors.ProcessCollectorOpts{}))
	reg.MustRegister(collectors.NewBuildInfoCollector())
	reg.MustRegister(collectors.NewGoCollector())
	return &MetricsEndpointHandler{
		reg: reg,
		cfg: cfg,
	}
}

func (h *MetricsEndpointHandler) Handler() http.Handler {
	return promhttp.HandlerFor(h.reg, promhttp.HandlerOpts{
		Registry: h.reg,
	})
}