unistack-org/micro-meter-prometheus
4
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases 29 Wiki Activity

protect meter by mutex
All checks were successful
test / test (push) Successful in 3m22s
Details

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов
2025-03-07 16:53:47 +03:00
parent 304bb99b78
commit 591c427815
1 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
prometheus.go
View File

@@ -26,7 +26,7 @@ type prometheusMeter struct {
gauge map[string]*gauges gauge map[string]*gauges
histogram map[string]*histograms histogram map[string]*histograms
summary map[string]*summaries summary map[string]*summaries
sync.Mutex mu *sync.Mutex
} }
type counters struct { type counters struct {
@@ -79,6 +79,7 @@ func NewMeter(opts ...meter.Option) *prometheusMeter {
gauge: make(map[string]*gauges), gauge: make(map[string]*gauges),
histogram: make(map[string]*histograms), histogram: make(map[string]*histograms),
summary: make(map[string]*summaries), summary: make(map[string]*summaries),
mu: &sync.Mutex{},
} }
} }
@@ -100,8 +101,8 @@ func (m *prometheusMeter) Name() string {
} }
func (m *prometheusMeter) Counter(name string, labels ...string) meter.Counter { func (m *prometheusMeter) Counter(name string, labels ...string) meter.Counter {
m.Lock() m.mu.Lock()
defer m.Unlock() defer m.mu.Unlock()
nm := m.buildMetric(name) nm := m.buildMetric(name)
labels = append(m.opts.Labels, labels...) labels = append(m.opts.Labels, labels...)
cd, ok := m.counter[nm] cd, ok := m.counter[nm]
@@ -123,8 +124,8 @@ func (m *prometheusMeter) Counter(name string, labels ...string) meter.Counter {
} }
func (m *prometheusMeter) FloatCounter(name string, labels ...string) meter.FloatCounter { func (m *prometheusMeter) FloatCounter(name string, labels ...string) meter.FloatCounter {
m.Lock() m.mu.Lock()
defer m.Unlock() defer m.mu.Unlock()
nm := m.buildMetric(name) nm := m.buildMetric(name)
labels = append(m.opts.Labels, labels...) labels = append(m.opts.Labels, labels...)
cd, ok := m.floatCounter[nm] cd, ok := m.floatCounter[nm]
@@ -146,8 +147,8 @@ func (m *prometheusMeter) FloatCounter(name string, labels ...string) meter.Floa
} }
func (m *prometheusMeter) Gauge(name string, fn func() float64, labels ...string) meter.Gauge { func (m *prometheusMeter) Gauge(name string, fn func() float64, labels ...string) meter.Gauge {
m.Lock() m.mu.Lock()
defer m.Unlock() defer m.mu.Unlock()
nm := m.buildMetric(name) nm := m.buildMetric(name)
labels = append(m.opts.Labels, labels...) labels = append(m.opts.Labels, labels...)
cd, ok := m.gauge[nm] cd, ok := m.gauge[nm]
@@ -169,8 +170,8 @@ func (m *prometheusMeter) Gauge(name string, fn func() float64, labels ...string
} }
func (m *prometheusMeter) Histogram(name string, labels ...string) meter.Histogram { func (m *prometheusMeter) Histogram(name string, labels ...string) meter.Histogram {
m.Lock() m.mu.Lock()
defer m.Unlock() defer m.mu.Unlock()
nm := m.buildMetric(name) nm := m.buildMetric(name)
labels = append(m.opts.Labels, labels...) labels = append(m.opts.Labels, labels...)
cd, ok := m.histogram[nm] cd, ok := m.histogram[nm]
@@ -192,8 +193,8 @@ func (m *prometheusMeter) Histogram(name string, labels ...string) meter.Histogr
} }
func (m *prometheusMeter) Summary(name string, labels ...string) meter.Summary { func (m *prometheusMeter) Summary(name string, labels ...string) meter.Summary {
m.Lock() m.mu.Lock()
defer m.Unlock() defer m.mu.Unlock()
nm := m.buildMetric(name) nm := m.buildMetric(name)
labels = append(m.opts.Labels, labels...) labels = append(m.opts.Labels, labels...)
cd, ok := m.summary[nm] cd, ok := m.summary[nm]
@@ -215,8 +216,8 @@ func (m *prometheusMeter) Summary(name string, labels ...string) meter.Summary {
} }
func (m *prometheusMeter) SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) meter.Summary { func (m *prometheusMeter) SummaryExt(name string, window time.Duration, quantiles []float64, labels ...string) meter.Summary {
m.Lock() m.mu.Lock()
defer m.Unlock() defer m.mu.Unlock()
nm := m.buildMetric(name) nm := m.buildMetric(name)
labels = append(m.opts.Labels, labels...) labels = append(m.opts.Labels, labels...)
cd, ok := m.summary[nm] cd, ok := m.summary[nm]
@@ -277,6 +278,8 @@ func (m *prometheusMeter) Write(w io.Writer, opts ...meter.Option) error {
enc := expfmt.NewEncoder(w, expfmt.NewFormat(expfmt.TypeTextPlain)) enc := expfmt.NewEncoder(w, expfmt.NewFormat(expfmt.TypeTextPlain))
m.mu.Lock()
for name, metrics := range m.counter { for name, metrics := range m.counter {
mf := &dto.MetricFamily{ mf := &dto.MetricFamily{
Name: newName(name), Name: newName(name),
@@ -352,6 +355,8 @@ func (m *prometheusMeter) Write(w io.Writer, opts ...meter.Option) error {
mfs = append(mfs, mf) mfs = append(mfs, mf)
} }
m.mu.Unlock()
for _, mf := range mfs { for _, mf := range mfs {
_ = enc.Encode(mf) _ = enc.Encode(mf)
} }
@@ -370,6 +375,7 @@ func (m *prometheusMeter) Clone(opts ...meter.Option) meter.Meter {
} }
return &prometheusMeter{ return &prometheusMeter{
mu: m.mu,
set: m.set, set: m.set,
opts: options, opts: options,
floatCounter: m.floatCounter, floatCounter: m.floatCounter,