diff options
Diffstat (limited to 'service/metrics/rpc.go')
| -rw-r--r-- | service/metrics/rpc.go | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/service/metrics/rpc.go b/service/metrics/rpc.go index 285feee1..9a36f14c 100644 --- a/service/metrics/rpc.go +++ b/service/metrics/rpc.go @@ -158,7 +158,7 @@ func (rpc *rpcServer) Register(c *NamedCollector, ok *bool) (err error) { var collector prometheus.Collector switch c.Type { - case "histogram": + case Histogram: opts := prometheus.HistogramOpts{ Name: c.Name, Namespace: c.Namespace, @@ -172,7 +172,7 @@ func (rpc *rpcServer) Register(c *NamedCollector, ok *bool) (err error) { } else { collector = prometheus.NewHistogram(opts) } - case "gauge": + case Gauge: opts := prometheus.GaugeOpts{ Name: c.Name, Namespace: c.Namespace, @@ -185,7 +185,7 @@ func (rpc *rpcServer) Register(c *NamedCollector, ok *bool) (err error) { } else { collector = prometheus.NewGauge(opts) } - case "counter": + case Counter: opts := prometheus.CounterOpts{ Name: c.Name, Namespace: c.Namespace, @@ -198,7 +198,7 @@ func (rpc *rpcServer) Register(c *NamedCollector, ok *bool) (err error) { } else { collector = prometheus.NewCounter(opts) } - case "summary": + case Summary: opts := prometheus.SummaryOpts{ Name: c.Name, Namespace: c.Namespace, @@ -217,6 +217,8 @@ func (rpc *rpcServer) Register(c *NamedCollector, ok *bool) (err error) { } + // add collector to sync.Map + rpc.svc.collectors.Store(c.Name, collector) // that method might panic, we handle it by recover rpc.svc.MustRegister(collector) |