forked from lukasjarosch/genki
/
prometheus.go
46 lines (39 loc) · 1.18 KB
/
prometheus.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
package interceptor
import (
"github.com/prometheus/client_golang/prometheus"
"github.com/marcoEgger/genki/broker"
)
var (
InboundGauge = prometheus.NewGaugeVec(prometheus.GaugeOpts{
Name: "amqp_inbound",
Help: "Increased on incoming deliveries, decreased on ack/nack",
ConstLabels: nil,
}, []string{"routing_key"})
TransportErrorCounter = prometheus.NewCounterVec(prometheus.CounterOpts{
Name: "amqp_transport_error",
Help: "Increased when a message could not be decoded",
ConstLabels: nil,
}, []string{"routing_key"})
NackCounter = prometheus.NewCounterVec(prometheus.CounterOpts{
Name: "amqp_not_acknowledged",
Help: "Incremented on every NACK",
}, []string{"routing_key", "requeue"})
)
func PrometheusInterceptor(next broker.Handler) broker.Handler {
return func(event broker.Event) {
InboundGauge.With(prometheus.Labels{
"routing_key": event.RoutingKey(),
}).Inc()
defer func() {
InboundGauge.With(prometheus.Labels{
"routing_key": event.RoutingKey(),
}).Dec()
}()
next(event)
}
}
func init() {
_ = prometheus.Register(InboundGauge)
_ = prometheus.Register(TransportErrorCounter)
_ = prometheus.Register(NackCounter)
}