forked from TrueCloudLab/frostfs-s3-gw
80 lines
2.1 KiB
Go
80 lines
2.1 KiB
Go
package metrics
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/pool"
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/pool/tree"
|
|
"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 TreePoolStatistic interface {
|
|
Statistic() tree.Statistic
|
|
}
|
|
|
|
type GateMetrics struct {
|
|
registry prometheus.Registerer
|
|
State *StateMetrics
|
|
Pool *poolMetricsCollector
|
|
Billing *billingMetrics
|
|
Stats *APIStatMetrics
|
|
HTTPServer *httpServerMetrics
|
|
TreePool *treePoolMetricsCollector
|
|
}
|
|
|
|
func NewGateMetrics(scraper StatisticScraper, treeScraper TreePoolStatistic, registry prometheus.Registerer) *GateMetrics {
|
|
stateMetric := newStateMetrics()
|
|
registry.MustRegister(stateMetric)
|
|
|
|
poolMetric := newPoolMetricsCollector(scraper)
|
|
registry.MustRegister(poolMetric)
|
|
|
|
billingMetric := newBillingMetrics()
|
|
billingMetric.Register()
|
|
|
|
statsMetric := newAPIStatMetrics()
|
|
registry.MustRegister(statsMetric)
|
|
|
|
serverMetric := newHTTPServerMetrics()
|
|
registry.MustRegister(serverMetric)
|
|
|
|
treePoolMetric := newTreePoolMetricsCollector(treeScraper)
|
|
registry.MustRegister(treePoolMetric)
|
|
|
|
return &GateMetrics{
|
|
registry: registry,
|
|
State: stateMetric,
|
|
Pool: poolMetric,
|
|
Billing: billingMetric,
|
|
Stats: statsMetric,
|
|
HTTPServer: serverMetric,
|
|
TreePool: treePoolMetric,
|
|
}
|
|
}
|
|
|
|
func (g *GateMetrics) Unregister() {
|
|
g.registry.Unregister(g.State)
|
|
g.registry.Unregister(g.Pool)
|
|
g.Billing.Unregister()
|
|
g.registry.Unregister(g.Stats)
|
|
g.registry.Unregister(g.HTTPServer)
|
|
}
|
|
|
|
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()
|
|
}
|