-
Notifications
You must be signed in to change notification settings - Fork 2
/
prometheus.go
75 lines (68 loc) · 1.87 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
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
package mysql
import (
"fmt"
"time"
"github.com/prometheus/client_golang/prometheus"
"github.com/rs/zerolog/log"
)
type metricSet struct {
ConnPool *prometheus.GaugeVec
Request *prometheus.CounterVec
Latency *prometheus.HistogramVec
}
var (
labels = []string{"sql", "name"}
latencyBucket = []float64{
4, 8, 16, 32, 64, 128, 256, 512, 1024, 2048, 4096, 8192}
connPoolUpdateInterval = 1 * time.Second
)
func newMetricSet(appName string) *metricSet {
return &metricSet{
ConnPool: prometheus.NewGaugeVec(
prometheus.GaugeOpts{
Name: fmt.Sprintf("%s_mysql_conn_pool", appName),
Help: "",
}, []string{"name"}),
Request: prometheus.NewCounterVec(
prometheus.CounterOpts{
Name: fmt.Sprintf("%s_mysql_request_total", appName),
Help: "how many CRUD operations sent to DB.",
}, labels),
Latency: prometheus.NewHistogramVec(
prometheus.HistogramOpts{
Name: fmt.Sprintf("%s_mysql_latency_milliseconds", appName),
Help: "CRUD latency in milliseconds",
Buckets: latencyBucket,
}, labels),
}
}
func (m *metricSet) Register() {
err := prometheus.Register(m.ConnPool)
if err != nil {
log.Err(err).Msgf("failed to register Prometheus ConnPool gauges")
}
err = prometheus.Register(m.Request)
if err != nil {
log.Err(err).Msgf("failed to register prometheus Request counters")
}
err = prometheus.Register(m.Latency)
if err != nil {
log.Err(err).Msgf("failed to register Prometheus Latency histogram")
}
}
func (m *metricSet) Unregister() {
prometheus.Unregister(m.ConnPool)
prometheus.Unregister(m.Request)
prometheus.Unregister(m.Latency)
}
func (s *metricSet) Observe(name string, startedAt time.Time) func() {
return func() {
if s.Request != nil {
s.Request.WithLabelValues(name).Inc()
}
if s.Latency != nil {
s.Latency.WithLabelValues(name).Observe(
float64(time.Since(startedAt).Milliseconds()))
}
}
}