44 lines
1.2 KiB
Go
44 lines
1.2 KiB
Go
|
package provider
|
||
|
|
||
|
import (
|
||
|
"github.com/go-kit/kit/metrics"
|
||
|
"github.com/go-kit/kit/metrics/statsd"
|
||
|
)
|
||
|
|
||
|
type statsdProvider struct {
|
||
|
s *statsd.Statsd
|
||
|
stop func()
|
||
|
}
|
||
|
|
||
|
// NewStatsdProvider wraps the given Statsd object and stop func and returns a
|
||
|
// Provider that produces Statsd metrics. A typical stop function would be
|
||
|
// ticker.Stop from the ticker passed to the SendLoop helper method.
|
||
|
func NewStatsdProvider(s *statsd.Statsd, stop func()) Provider {
|
||
|
return &statsdProvider{
|
||
|
s: s,
|
||
|
stop: stop,
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// NewCounter implements Provider.
|
||
|
func (p *statsdProvider) NewCounter(name string) metrics.Counter {
|
||
|
return p.s.NewCounter(name, 1.0)
|
||
|
}
|
||
|
|
||
|
// NewGauge implements Provider.
|
||
|
func (p *statsdProvider) NewGauge(name string) metrics.Gauge {
|
||
|
return p.s.NewGauge(name)
|
||
|
}
|
||
|
|
||
|
// NewHistogram implements Provider, returning a StatsD Timing that accepts
|
||
|
// observations in milliseconds. The sample rate is fixed at 1.0. The bucket
|
||
|
// parameter is ignored.
|
||
|
func (p *statsdProvider) NewHistogram(name string, _ int) metrics.Histogram {
|
||
|
return p.s.NewTiming(name, 1.0)
|
||
|
}
|
||
|
|
||
|
// Stop implements Provider, invoking the stop function passed at construction.
|
||
|
func (p *statsdProvider) Stop() {
|
||
|
p.stop()
|
||
|
}
|