/
prometheus.go
116 lines (100 loc) · 3.57 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
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
// Package contains the collection of prometheus meters/counters
// and related update methods
package metrics
import (
"strconv"
"github.com/prometheus/client_golang/prometheus"
"github.com/ory/x/logrusx"
"github.com/ory/oathkeeper/driver"
)
var (
// RequestTotal provides the total number of requests
RequestTotal = prometheus.NewCounterVec(
prometheus.CounterOpts{
Name: "ory_oathkeeper_requests_total",
Help: "Total number of requests",
},
[]string{"service", "method", "request", "status_code"},
)
// HistogramRequestDuration provides the duration of requests
HistogramRequestDuration = prometheus.NewHistogramVec(
prometheus.HistogramOpts{
Name: "ory_oathkeeper_requests_duration_seconds",
Help: "Time spent serving requests.",
Buckets: prometheus.DefBuckets,
},
[]string{"service", "method", "request", "status_code"},
)
)
// RequestDurationObserve tracks request durations
type RequestDurationObserve func(histogram *prometheus.HistogramVec, service, request, method string, statusCode int) func(float64)
// UpdateRequest tracks total requests done
type UpdateRequest func(counter *prometheus.CounterVec, service, request, method string, statusCode int)
// PrometheusRepository provides methods to manage prometheus metrics
type PrometheusRepository struct {
logger *logrusx.Logger
requestDurationObserve RequestDurationObserve
updateRequest UpdateRequest
Registry *prometheus.Registry
metrics []prometheus.Collector
}
// NewConfigurablePrometheusRepository creates a new prometheus repository with the given settings
func NewConfigurablePrometheusRepository(d driver.Driver, logger *logrusx.Logger) *PrometheusRepository {
RequestTotal = prometheus.NewCounterVec(
prometheus.CounterOpts{
Name: d.Configuration().PrometheusMetricsNamePrefix() + "requests_total",
Help: "Total number of requests",
},
[]string{"service", "method", "request", "status_code"},
)
HistogramRequestDuration = prometheus.NewHistogramVec(
prometheus.HistogramOpts{
Name: d.Configuration().PrometheusMetricsNamePrefix() + "requests_duration_seconds",
Help: "Time spent serving requests.",
Buckets: prometheus.DefBuckets,
},
[]string{"service", "method", "request", "status_code"},
)
return NewPrometheusRepository(logger)
}
// NewPrometheusRepository creates a new prometheus repository
func NewPrometheusRepository(logger *logrusx.Logger) *PrometheusRepository {
m := []prometheus.Collector{
prometheus.NewGoCollector(),
prometheus.NewProcessCollector(prometheus.ProcessCollectorOpts{}),
RequestTotal,
HistogramRequestDuration,
}
r := prometheus.NewRegistry()
for _, metric := range m {
if err := r.Register(metric); err != nil {
logger.WithError(err).Error("Unable to register prometheus metric.")
}
}
mr := &PrometheusRepository{
logger: logger,
Registry: r,
metrics: m,
}
return mr
}
// RequestDurationObserve tracks request durations
func (r *PrometheusRepository) RequestDurationObserve(service, request, method string, statusCode int) func(float64) {
return func(v float64) {
HistogramRequestDuration.With(prometheus.Labels{
"service": service,
"method": method,
"request": request,
"status_code": strconv.Itoa(statusCode),
}).Observe(v)
}
}
// UpdateRequest tracks total requests done
func (r *PrometheusRepository) UpdateRequest(service, request, method string, statusCode int) {
RequestTotal.With(prometheus.Labels{
"service": service,
"method": method,
"request": request,
"status_code": strconv.Itoa(statusCode),
}).Inc()
}