forked from TrueCloudLab/frostfs-s3-gw
66 lines
1.6 KiB
Go
66 lines
1.6 KiB
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()
|
|
}
|