frostfs-s3-gw/metrics/gate.go

package metrics

import (
	"net/http"

	"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/pool"
	"github.com/prometheus/client_golang/prometheus"
	"github.com/prometheus/client_golang/prometheus/promhttp"
	dto "github.com/prometheus/client_model/go"
)

const namespace = "frostfs_s3_gw"

type StatisticScraper interface {
	Statistic() pool.Statistic
}

type GateMetrics struct {
	registry *prometheus.Registry
	State    *StateMetrics
	Pool     *poolMetricsCollector
	Billing  *billingMetrics
	Stats    *APIStatMetrics
}

func NewGateMetrics(scraper StatisticScraper) *GateMetrics {
	registry := prometheus.NewRegistry()

	stateMetric := newStateMetrics()
	registry.MustRegister(stateMetric)

	poolMetric := newPoolMetricsCollector(scraper)
	registry.MustRegister(poolMetric)

	billingMetric := newBillingMetrics()
	billingMetric.Register()

	statsMetric := newAPIStatMetrics()
	registry.MustRegister(statsMetric)

	return &GateMetrics{
		registry: registry,
		State:    stateMetric,
		Pool:     poolMetric,
		Billing:  billingMetric,
		Stats:    statsMetric,
	}
}

func (g *GateMetrics) Unregister() {
	g.registry.Unregister(g.State)
	g.registry.Unregister(g.Pool)
	g.Billing.Unregister()
	g.registry.Unregister(g.Stats)
}

func (g *GateMetrics) Handler() http.Handler {
	handler := http.NewServeMux()
	handler.Handle("/", promhttp.Handler())
	handler.Handle("/metrics/billing", promhttp.HandlerFor(g.Billing.Gatherer(), promhttp.HandlerOpts{}))
	return handler
}

func (g *GateMetrics) Gather() ([]*dto.MetricFamily, error) {
	return prometheus.Gatherers([]prometheus.Gatherer{g.Billing.Gatherer()}).Gather()
}