forked from kbudde/rabbitmq_exporter
-
Notifications
You must be signed in to change notification settings - Fork 0
/
exporter.go
100 lines (85 loc) · 2.7 KB
/
exporter.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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package main
import (
"sync"
log "github.com/Sirupsen/logrus"
"github.com/prometheus/client_golang/prometheus"
)
type exporter struct {
mutex sync.RWMutex
queueMetricsGauge map[string]*prometheus.GaugeVec
queueMetricsCounter map[string]*prometheus.CounterVec
overviewMetrics map[string]prometheus.Gauge
}
func newExporter() *exporter {
return &exporter{
queueMetricsGauge: queueGaugeVec,
queueMetricsCounter: queueCounterVec,
overviewMetrics: overviewMetricDescription,
}
}
func (e *exporter) fetchRabbit() {
rabbitMqOverviewData := getOverviewMap(config)
rabbitMqQueueData := getQueueMap(config)
log.WithField("overviewData", rabbitMqOverviewData).Debug("Overview data")
for key, gauge := range e.overviewMetrics {
if value, ok := rabbitMqOverviewData[key]; ok {
log.WithFields(log.Fields{"key": key, "value": value}).Debug("Set overview metric for key")
gauge.Set(value)
} else {
log.WithFields(log.Fields{"key": key}).Warn("Overview data not found")
}
}
log.WithField("queueData", rabbitMqQueueData).Debug("Queue data")
for key, gaugevec := range e.queueMetricsGauge {
for queue, data := range rabbitMqQueueData {
if value, ok := data[key]; ok {
log.WithFields(log.Fields{"queue": queue, "key": key, "value": value}).Debug("Set queue metric for key")
gaugevec.WithLabelValues(queue).Set(value)
} else {
//log.WithFields(log.Fields{"queue": queue, "key": key}).Warn("Queue data not found")
}
}
}
for key, countvec := range e.queueMetricsCounter {
for queue, data := range rabbitMqQueueData {
if value, ok := data[key]; ok {
log.WithFields(log.Fields{"queue": queue, "key": key, "value": value}).Debug("Set queue metric for key")
countvec.WithLabelValues(queue).Set(value)
} else {
//log.WithFields(log.Fields{"queue": queue, "key": key}).Warn("Queue data not found")
}
}
}
log.Info("Metrics updated successfully.")
}
func (e *exporter) Describe(ch chan<- *prometheus.Desc) {
for _, gauge := range e.overviewMetrics {
gauge.Describe(ch)
}
for _, gaugevec := range e.queueMetricsGauge {
gaugevec.Describe(ch)
}
for _, countervec := range e.queueMetricsCounter {
countervec.Describe(ch)
}
}
func (e *exporter) Collect(ch chan<- prometheus.Metric) {
e.mutex.Lock() // To protect metrics from concurrent collects.
defer e.mutex.Unlock()
for _, gaugevec := range e.queueMetricsGauge {
gaugevec.Reset()
}
for _, countvec := range e.queueMetricsCounter {
countvec.Reset()
}
e.fetchRabbit()
for _, gauge := range e.overviewMetrics {
gauge.Collect(ch)
}
for _, gaugevec := range e.queueMetricsGauge {
gaugevec.Collect(ch)
}
for _, countervec := range e.queueMetricsCounter {
countervec.Collect(ch)
}
}