mirrors/statsd_exporter
1
0
Fork 1
mirror of https://github.com/prometheus/statsd_exporter.git synced 2024-06-03 13:59:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add option to allow using another Registerer

Browse source 
Signed-off-by: bakins <brian@akins.org>
This commit is contained in:
bakins 2020-08-26 10:34:36 -04:00
parent bac6cbe8c5
commit dd0a8cfdc8
1 changed files with 30 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

38
pkg/registry/registry.go
View file

@ -47,13 +47,23 @@ type Registry struct {
// hash.
ValueBuf, NameBuf bytes.Buffer
Hasher hash.Hash64
registerer prometheus.Registerer
}
func NewRegistry(mapper *mapper.MetricMapper) *Registry {
func NewRegistry(mapper *mapper.MetricMapper, options ...RegistryOption) *Registry {
o := registryOptions{
registerer: prometheus.DefaultRegisterer,
}
for _, f := range options {
f(&o)
}
return &Registry{
Metrics: make(map[string]metrics.Metric),
Mapper: mapper,
Hasher: fnv.New64a(),
Metrics: make(map[string]metrics.Metric),
Mapper: mapper,
Hasher: fnv.New64a(),
registerer: o.registerer,
}
}
@ -168,7 +178,7 @@ func (r *Registry) GetCounter(metricName string, labels prometheus.Labels, help
Help: help,
}, labelNames)
if err := prometheus.Register(uncheckedCollector{counterVec}); err != nil {
if err := r.registerer.Register(uncheckedCollector{counterVec}); err != nil {
return nil, err
}
} else {
@ -204,7 +214,7 @@ func (r *Registry) GetGauge(metricName string, labels prometheus.Labels, help st
Help: help,
}, labelNames)
if err := prometheus.Register(uncheckedCollector{gaugeVec}); err != nil {
if err := r.registerer.Register(uncheckedCollector{gaugeVec}); err != nil {
return nil, err
}
} else {
@ -254,7 +264,7 @@ func (r *Registry) GetHistogram(metricName string, labels prometheus.Labels, hel
Buckets: buckets,
}, labelNames)
if err := prometheus.Register(uncheckedCollector{histogramVec}); err != nil {
if err := r.registerer.Register(uncheckedCollector{histogramVec}); err != nil {
return nil, err
}
} else {
@ -316,7 +326,7 @@ func (r *Registry) GetSummary(metricName string, labels prometheus.Labels, help
BufCap: summaryOptions.BufCap,
}, labelNames)
if err := prometheus.Register(uncheckedCollector{summaryVec}); err != nil {
if err := r.registerer.Register(uncheckedCollector{summaryVec}); err != nil {
return nil, err
}
} else {
@ -381,3 +391,15 @@ func (r *Registry) HashLabels(labels prometheus.Labels) (metrics.LabelHash, []st
return lh, labelNames
}
type registryOptions struct {
registerer prometheus.Registerer
}
type RegistryOption func(*registryOptions)
func WithRegisterer(registerer prometheus.Registerer) RegistryOption {
return func(o *registryOptions) {
o.registerer = registerer
}
}