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
|
package telemetry
import (
"context"
"net/http"
"sync"
"time"
"go.opentelemetry.io/otel/metric"
)
type Metrics struct {
Meter metric.Meter
countersLock sync.Mutex
Counters map[string]metric.Int64Counter
histogramsLock sync.Mutex
Histograms map[string]metric.Float64Histogram
Configuration *MetricsConfiguration
}
type MetricsInterface interface {
GetCounter(name string, description string) (metric.Int64Counter, error)
GetHistogram(name string, description string, unit string) (metric.Float64Histogram, error)
CredentialsRequest(value int64, attrs map[*Attribute]string) (metric.Int64Counter, error)
RequestDuration(value float64, attrs map[*Attribute]string) (metric.Float64Histogram, error)
QueryDuration(value float64, attrs map[*Attribute]string) (metric.Float64Histogram, error)
BuildTelemetryAttributes(requestMethod string, methodParameters map[string]interface{}, req *http.Request, res *http.Response, requestStarted time.Time, resendCount int) (map[*Attribute]string, float64, float64, error)
}
func (m *Metrics) GetCounter(name string, description string) (metric.Int64Counter, error) {
m.countersLock.Lock()
defer m.countersLock.Unlock()
if counter, exists := m.Counters[name]; exists {
return counter, nil
}
counter, _ := m.Meter.Int64Counter(name, metric.WithDescription(description))
m.Counters[name] = counter
return counter, nil
}
func (m *Metrics) GetHistogram(name string, description string, unit string) (metric.Float64Histogram, error) {
m.histogramsLock.Lock()
defer m.histogramsLock.Unlock()
if histogram, exists := m.Histograms[name]; exists {
return histogram, nil
}
histogram, _ := m.Meter.Float64Histogram(name, metric.WithDescription(description), metric.WithUnit(unit))
m.Histograms[name] = histogram
return histogram, nil
}
func (m *Metrics) CredentialsRequest(value int64, attrs map[*Attribute]string) (metric.Int64Counter, error) {
var counter, err = m.GetCounter(CredentialsRequest.Name, CredentialsRequest.Description)
if err == nil {
attrs, err := m.PrepareAttributes(CredentialsRequest, attrs, m.Configuration)
if err == nil {
counter.Add(context.Background(), value, metric.WithAttributeSet(attrs))
}
}
return counter, err
}
func (m *Metrics) RequestDuration(value float64, attrs map[*Attribute]string) (metric.Float64Histogram, error) {
var histogram, err = m.GetHistogram(RequestDuration.Name, RequestDuration.Description, RequestDuration.Unit)
if err == nil {
attrs, err := m.PrepareAttributes(RequestDuration, attrs, m.Configuration)
if err == nil {
histogram.Record(context.Background(), value, metric.WithAttributeSet(attrs))
}
}
return histogram, err
}
func (m *Metrics) QueryDuration(value float64, attrs map[*Attribute]string) (metric.Float64Histogram, error) {
var histogram, err = m.GetHistogram(QueryDuration.Name, QueryDuration.Description, QueryDuration.Unit)
if err == nil {
attrs, err := m.PrepareAttributes(QueryDuration, attrs, m.Configuration)
if err == nil {
histogram.Record(context.Background(), value, metric.WithAttributeSet(attrs))
}
}
return histogram, err
}
|
