57 lines
1.7 KiB
Go
57 lines
1.7 KiB
Go
package testhelpers
|
|
|
|
import "github.com/go-kit/kit/metrics"
|
|
|
|
// CollectingCounter is a metrics.Counter implementation that enables access to the CounterValue and LastLabelValues.
|
|
type CollectingCounter struct {
|
|
CounterValue float64
|
|
LastLabelValues []string
|
|
}
|
|
|
|
// With is there to satisfy the metrics.Counter interface.
|
|
func (c *CollectingCounter) With(labelValues ...string) metrics.Counter {
|
|
c.LastLabelValues = labelValues
|
|
return c
|
|
}
|
|
|
|
// Add is there to satisfy the metrics.Counter interface.
|
|
func (c *CollectingCounter) Add(delta float64) {
|
|
c.CounterValue += delta
|
|
}
|
|
|
|
// CollectingGauge is a metrics.Gauge implementation that enables access to the GaugeValue and LastLabelValues.
|
|
type CollectingGauge struct {
|
|
GaugeValue float64
|
|
LastLabelValues []string
|
|
}
|
|
|
|
// With is there to satisfy the metrics.Gauge interface.
|
|
func (g *CollectingGauge) With(labelValues ...string) metrics.Gauge {
|
|
g.LastLabelValues = labelValues
|
|
return g
|
|
}
|
|
|
|
// Set is there to satisfy the metrics.Gauge interface.
|
|
func (g *CollectingGauge) Set(value float64) {
|
|
g.GaugeValue = value
|
|
}
|
|
|
|
// Add is there to satisfy the metrics.Gauge interface.
|
|
func (g *CollectingGauge) Add(delta float64) {
|
|
g.GaugeValue = delta
|
|
}
|
|
|
|
// CollectingHealthCheckMetrics can be used for testing the Metrics instrumentation of the HealthCheck package.
|
|
type CollectingHealthCheckMetrics struct {
|
|
Gauge *CollectingGauge
|
|
}
|
|
|
|
// NewCollectingHealthCheckMetrics creates a new CollectingHealthCheckMetrics instance.
|
|
func NewCollectingHealthCheckMetrics() *CollectingHealthCheckMetrics {
|
|
return &CollectingHealthCheckMetrics{&CollectingGauge{}}
|
|
}
|
|
|
|
// BackendServerUpGauge is there to satisfy the healthcheck.metricsRegistry interface.
|
|
func (m *CollectingHealthCheckMetrics) BackendServerUpGauge() metrics.Gauge {
|
|
return m.Gauge
|
|
}
|