diff options
-rw-r--r-- | service/metrics/rpc.go | 21 | ||||
-rw-r--r-- | service/metrics/rpc_test.go | 17 |
2 files changed, 19 insertions, 19 deletions
diff --git a/service/metrics/rpc.go b/service/metrics/rpc.go index ca99d4ac..ee8ef984 100644 --- a/service/metrics/rpc.go +++ b/service/metrics/rpc.go @@ -53,25 +53,8 @@ func (rpc *rpcServer) Add(m *Metric, ok *bool) (err error) { c.(*prometheus.CounterVec).WithLabelValues(m.Labels...).Add(m.Value) - case prometheus.Summary: - c.(prometheus.Counter).Add(m.Value) - - case *prometheus.SummaryVec: - if len(m.Labels) == 0 { - return fmt.Errorf("required labels for collector `%s`", m.Name) - } - - c.(*prometheus.SummaryVec).WithLabelValues(m.Labels...).Observe(m.Value) - - case prometheus.Histogram: - c.(prometheus.Histogram).Observe(m.Value) - - case *prometheus.HistogramVec: - if len(m.Labels) == 0 { - return fmt.Errorf("required labels for collector `%s`", m.Name) - } - - c.(*prometheus.HistogramVec).WithLabelValues(m.Labels...).Observe(m.Value) + default: + return fmt.Errorf("collector `%s` does not support method `Add`", m.Name) } *ok = true diff --git a/service/metrics/rpc_test.go b/service/metrics/rpc_test.go index da991b34..74b067c5 100644 --- a/service/metrics/rpc_test.go +++ b/service/metrics/rpc_test.go @@ -616,3 +616,20 @@ func Test_Add_RPC_MetricError_2(t *testing.T) { Value: 100.0, }, &ok)) } + +func Test_Add_RPC_MetricError_3(t *testing.T) { + client, c := setup( + t, + `"user_gauge":{ + "type": "histogram", + "labels": ["type", "section"] + }`, + ) + defer c.Stop() + + var ok bool + assert.Error(t, client.Call("metrics.Add", Metric{ + Name: "user_gauge", + Value: 100.0, + }, &ok)) +} |