mirror of
https://github.com/hanzoai/traces.git
synced 2026-10-30 18:20:41 +00:00
aeda5144b1
* fix: correctly publish the hedgedmetrics counter with delta * fix and update CHANGELOG.md * fix datarace in the test * replace locking with atomic * remove locking and unnecessary code Signed-off-by: Joe Elliott <[email protected]> * i don't care Signed-off-by: Joe Elliott <[email protected]> --------- Signed-off-by: Joe Elliott <[email protected]> Co-authored-by: Joe Elliott <[email protected]>
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package hedgedmetrics
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/cristalhq/hedgedhttp"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
)
|
|
|
|
const (
|
|
hedgedMetricsPublishDuration = 10 * time.Second
|
|
)
|
|
|
|
type diffCounter struct {
|
|
previous uint64
|
|
counter prometheus.Counter
|
|
}
|
|
|
|
func (d *diffCounter) addAbsoluteToCounter(value uint64) {
|
|
diff := float64(value - d.previous)
|
|
if value < d.previous {
|
|
diff = float64(value)
|
|
}
|
|
d.counter.Add(diff)
|
|
d.previous = value
|
|
}
|
|
|
|
// statsProvider defines the interface that wraps hedgedhttp.Stats for ease of testing
|
|
type statsProvider interface {
|
|
Snapshot() hedgedhttp.StatsSnapshot
|
|
}
|
|
|
|
// Publish flushes metrics from hedged requests every tickerDur
|
|
func Publish(s *hedgedhttp.Stats, counter prometheus.Counter) {
|
|
publishWithDuration(s, counter, hedgedMetricsPublishDuration)
|
|
}
|
|
|
|
func publishWithDuration(s statsProvider, counter prometheus.Counter, duration time.Duration) {
|
|
ticker := time.NewTicker(duration)
|
|
diff := &diffCounter{previous: 0, counter: counter}
|
|
|
|
go func() {
|
|
for range ticker.C {
|
|
snap := s.Snapshot()
|
|
|
|
hedgedRequests := snap.ActualRoundTrips - snap.RequestedRoundTrips
|
|
diff.addAbsoluteToCounter(hedgedRequests)
|
|
}
|
|
}()
|
|
}
|