73 lines
2.4 KiB
Go
73 lines
2.4 KiB
Go
package rocsp
|
|
|
|
import (
|
|
"github.com/go-redis/redis/v8"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
// An interface satisfied by *redis.ClusterClient and also by a mock in our tests.
|
|
type poolStatGetter interface {
|
|
PoolStats() *redis.PoolStats
|
|
}
|
|
|
|
var _ poolStatGetter = (*redis.ClusterClient)(nil)
|
|
|
|
type metricsCollector struct {
|
|
statGetter poolStatGetter
|
|
|
|
// Stats accessible from the go-redis connector:
|
|
// https://pkg.go.dev/github.com/go-redis/redis@v6.15.9+incompatible/internal/pool#Stats
|
|
lookups *prometheus.Desc
|
|
totalConns *prometheus.Desc
|
|
idleConns *prometheus.Desc
|
|
staleConns *prometheus.Desc
|
|
}
|
|
|
|
// Describe is implemented with DescribeByCollect. That's possible because the
|
|
// Collect method will always return the same metrics with the same descriptors.
|
|
func (dbc metricsCollector) Describe(ch chan<- *prometheus.Desc) {
|
|
prometheus.DescribeByCollect(dbc, ch)
|
|
}
|
|
|
|
// Collect first triggers the Redis ClusterClient's PoolStats function.
|
|
// Then it creates constant metrics for each Stats value on the fly based
|
|
// on the returned data.
|
|
//
|
|
// Note that Collect could be called concurrently, so we depend on PoolStats()
|
|
// to be concurrency-safe.
|
|
func (dbc metricsCollector) Collect(ch chan<- prometheus.Metric) {
|
|
writeGauge := func(stat *prometheus.Desc, val uint32, labelValues ...string) {
|
|
ch <- prometheus.MustNewConstMetric(stat, prometheus.GaugeValue, float64(val), labelValues...)
|
|
}
|
|
|
|
stats := dbc.statGetter.PoolStats()
|
|
writeGauge(dbc.lookups, stats.Hits, "hit")
|
|
writeGauge(dbc.lookups, stats.Misses, "miss")
|
|
writeGauge(dbc.lookups, stats.Timeouts, "timeout")
|
|
writeGauge(dbc.totalConns, stats.TotalConns)
|
|
writeGauge(dbc.idleConns, stats.IdleConns)
|
|
writeGauge(dbc.staleConns, stats.StaleConns)
|
|
}
|
|
|
|
func newMetricsCollector(statGetter poolStatGetter, labels prometheus.Labels) metricsCollector {
|
|
return metricsCollector{
|
|
statGetter: statGetter,
|
|
lookups: prometheus.NewDesc(
|
|
"redis_connection_pool_lookups",
|
|
"Number of lookups for a connection in the pool, labeled by hit/miss",
|
|
[]string{"result"}, labels),
|
|
totalConns: prometheus.NewDesc(
|
|
"redis_connection_pool_total_conns",
|
|
"Number of total connections in the pool.",
|
|
nil, labels),
|
|
idleConns: prometheus.NewDesc(
|
|
"redis_connection_pool_idle_conns",
|
|
"Number of idle connections in the pool.",
|
|
nil, labels),
|
|
staleConns: prometheus.NewDesc(
|
|
"redis_connection_pool_stale_conns",
|
|
"Number of stale connections removed from the pool.",
|
|
nil, labels),
|
|
}
|
|
}
|