@@ -2,10 +2,11 @@ package metrics
2
2
3
3
import (
4
4
"fmt"
5
+ "sync"
6
+
5
7
"github.com/prometheus/client_golang/prometheus"
6
- "github.com/prometheus/client_model/go"
8
+ io_prometheus_client "github.com/prometheus/client_model/go"
7
9
"github.com/sirupsen/logrus"
8
- "sync"
9
10
)
10
11
11
12
var (
@@ -20,7 +21,7 @@ const (
20
21
grabbitPrefix = "grabbit"
21
22
)
22
23
23
- type HandlerMetrics struct {
24
+ type handlerMetrics struct {
24
25
result * prometheus.CounterVec
25
26
latency prometheus.Summary
26
27
}
@@ -62,17 +63,17 @@ func RunHandlerWithMetric(handleMessage func() error, handlerName string, logger
62
63
return err
63
64
}
64
65
65
- func GetHandlerMetrics (handlerName string ) * HandlerMetrics {
66
+ func GetHandlerMetrics (handlerName string ) * handlerMetrics {
66
67
entry , ok := handlerMetricsByHandlerName .Load (handlerName )
67
68
if ok {
68
- return entry .(* HandlerMetrics )
69
+ return entry .(* handlerMetrics )
69
70
}
70
71
71
72
return nil
72
73
}
73
74
74
- func newHandlerMetrics (handlerName string ) * HandlerMetrics {
75
- return & HandlerMetrics {
75
+ func newHandlerMetrics (handlerName string ) * handlerMetrics {
76
+ return & handlerMetrics {
76
77
result : prometheus .NewCounterVec (
77
78
prometheus.CounterOpts {
78
79
Namespace : grabbitPrefix ,
@@ -98,15 +99,15 @@ func trackTime(functionToTrack func() error, observer prometheus.Observer) error
98
99
return functionToTrack ()
99
100
}
100
101
101
- func (hm * HandlerMetrics ) GetSuccessCount () (float64 , error ) {
102
+ func (hm * handlerMetrics ) GetSuccessCount () (float64 , error ) {
102
103
return hm .getLabeledCounterValue (success )
103
104
}
104
105
105
- func (hm * HandlerMetrics ) GetFailureCount () (float64 , error ) {
106
+ func (hm * handlerMetrics ) GetFailureCount () (float64 , error ) {
106
107
return hm .getLabeledCounterValue (failure )
107
108
}
108
109
109
- func (hm * HandlerMetrics ) GetLatencySampleCount () (* uint64 , error ) {
110
+ func (hm * handlerMetrics ) GetLatencySampleCount () (* uint64 , error ) {
110
111
m := & io_prometheus_client.Metric {}
111
112
err := hm .latency .Write (m )
112
113
if err != nil {
@@ -116,7 +117,7 @@ func (hm *HandlerMetrics) GetLatencySampleCount() (*uint64, error) {
116
117
return m .GetSummary ().SampleCount , nil
117
118
}
118
119
119
- func (hm * HandlerMetrics ) getLabeledCounterValue (label string ) (float64 , error ) {
120
+ func (hm * handlerMetrics ) getLabeledCounterValue (label string ) (float64 , error ) {
120
121
m := & io_prometheus_client.Metric {}
121
122
err := hm .result .WithLabelValues (label ).Write (m )
122
123
0 commit comments