metrics/reporter.go at main · openziti/metrics · GitHub

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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
package metrics

import (
	"fmt"
	"sync/atomic"
	"time"
)

type MetricSink interface {
	Filter(name string) bool
	StartReport(registry Registry)
	EndReport(registry Registry)
	AcceptIntMetric(name string, value int64)
	AcceptFloatMetric(name string, value float64)
	AcceptPercentileMetric(name string, value PercentileSource)
}

type PercentileSource interface {
	Percentile(float64) float64
}

func NewDelegatingReporter(registry Registry, sink MetricSink, closeNotify <-chan struct{}) *DelegatingReporter {
	return &DelegatingReporter{
		registry:    registry,
		closeNotify: closeNotify,
		sink:        sink,
	}
}

type DelegatingReporter struct {
	registry    Registry
	closeNotify <-chan struct{}
	sink        MetricSink
	started     atomic.Bool
}

func (self *DelegatingReporter) Start(interval time.Duration) {
	if !self.started.CompareAndSwap(false, true) {
		return
	}

	timer := time.NewTicker(interval)
	defer timer.Stop()

	for {
		select {
		case <-timer.C:
			self.sink.StartReport(self.registry)
			self.registry.AcceptVisitor(self)
			self.sink.EndReport(self.registry)
		case <-self.closeNotify:
			return
		}
	}
}

func (self *DelegatingReporter) VisitIntMetric(name string, val int64, extra string) {
	if len(extra) > 0 {
		name = fmt.Sprintf("%s.%s", name, extra)
	}
	if self.sink.Filter(name) {
		self.sink.AcceptIntMetric(name, val)
	}
}

func (self *DelegatingReporter) VisitFloatMetric(name string, val float64, extra string) {
	if len(extra) > 0 {
		name = fmt.Sprintf("%s.%s", name, extra)
	}
	if self.sink.Filter(name) {
		self.sink.AcceptFloatMetric(name, val)
	}
}

func (self *DelegatingReporter) VisitPercentileMetric(name string, val PercentileSource, extra string) {
	if len(extra) > 0 {
		name = fmt.Sprintf("%s.%s", name, extra)
	}
	if self.sink.Filter(name) {
		self.sink.AcceptPercentileMetric(name, val)
	}
}

const (
	MetricNameCount      = "count"
	MetricNameMean       = "mean"
	MetricNameRateM1     = "rate_m1"
	MetricNameRateM5     = "rate_m5"
	MetricNameRateM15    = "rate_m15"
	MetricNameMin        = "min"
	MetricNameMax        = "max"
	MetricNamePercentile = "percentile"
)

func (self *DelegatingReporter) VisitGauge(name string, gauge Gauge) {
	self.VisitIntMetric(name, gauge.Value(), "")
}

func (self *DelegatingReporter) VisitGaugeFloat64(name string, gauge GaugeFloat64) {
	self.VisitFloatMetric(name, gauge.Value(), "")
}

func (self *DelegatingReporter) VisitMeter(name string, metric Meter) {
	self.VisitIntMetric(name, metric.Count(), MetricNameCount)
	self.VisitFloatMetric(name, metric.Rate1(), MetricNameRateM1)
	self.VisitFloatMetric(name, metric.Rate5(), MetricNameRateM5)
	self.VisitFloatMetric(name, metric.Rate15(), MetricNameRateM15)
	self.VisitFloatMetric(name, metric.RateMean(), MetricNameMean)
}

func (self *DelegatingReporter) VisitHistogram(name string, metric Histogram) {
	self.VisitIntMetric(name, metric.Count(), MetricNameCount)
	self.VisitFloatMetric(name, metric.Mean(), MetricNameMean)
	self.VisitIntMetric(name, metric.Min(), MetricNameMin)
	self.VisitIntMetric(name, metric.Max(), MetricNameMax)
	self.VisitPercentileMetric(name, metric, MetricNamePercentile)
}

func (self *DelegatingReporter) VisitTimer(name string, metric Timer) {
	self.VisitIntMetric(name, metric.Count(), MetricNameCount)

	self.VisitFloatMetric(name, metric.Rate1(), MetricNameRateM1)
	self.VisitFloatMetric(name, metric.Rate5(), MetricNameRateM5)
	self.VisitFloatMetric(name, metric.Rate15(), MetricNameRateM15)

	self.VisitFloatMetric(name, metric.Mean(), MetricNameMean)
	self.VisitIntMetric(name, metric.Min(), MetricNameMin)
	self.VisitIntMetric(name, metric.Max(), MetricNameMax)
	self.VisitPercentileMetric(name, metric, MetricNamePercentile)
}