traefik/vendor/github.com/vulcand/oxy/memmetrics/ratio.go
2017-03-09 13:13:02 +01:00

120 lines
2.1 KiB
Go

package memmetrics
import (
"time"
"github.com/mailgun/timetools"
)
type ratioOptSetter func(r *RatioCounter) error
func RatioClock(clock timetools.TimeProvider) ratioOptSetter {
return func(r *RatioCounter) error {
r.clock = clock
return nil
}
}
// RatioCounter calculates a ratio of a/a+b over a rolling window of predefined buckets
type RatioCounter struct {
clock timetools.TimeProvider
a *RollingCounter
b *RollingCounter
}
func NewRatioCounter(buckets int, resolution time.Duration, options ...ratioOptSetter) (*RatioCounter, error) {
rc := &RatioCounter{}
for _, o := range options {
if err := o(rc); err != nil {
return nil, err
}
}
if rc.clock == nil {
rc.clock = &timetools.RealTime{}
}
a, err := NewCounter(buckets, resolution, CounterClock(rc.clock))
if err != nil {
return nil, err
}
b, err := NewCounter(buckets, resolution, CounterClock(rc.clock))
if err != nil {
return nil, err
}
rc.a = a
rc.b = b
return rc, nil
}
func (r *RatioCounter) Reset() {
r.a.Reset()
r.b.Reset()
}
func (r *RatioCounter) IsReady() bool {
return r.a.countedBuckets+r.b.countedBuckets >= len(r.a.values)
}
func (r *RatioCounter) CountA() int64 {
return r.a.Count()
}
func (r *RatioCounter) CountB() int64 {
return r.b.Count()
}
func (r *RatioCounter) Resolution() time.Duration {
return r.a.Resolution()
}
func (r *RatioCounter) Buckets() int {
return r.a.Buckets()
}
func (r *RatioCounter) WindowSize() time.Duration {
return r.a.WindowSize()
}
func (r *RatioCounter) ProcessedCount() int64 {
return r.CountA() + r.CountB()
}
func (r *RatioCounter) Ratio() float64 {
a := r.a.Count()
b := r.b.Count()
// No data, return ok
if a+b == 0 {
return 0
}
return float64(a) / float64(a+b)
}
func (r *RatioCounter) IncA(v int) {
r.a.Inc(v)
}
func (r *RatioCounter) IncB(v int) {
r.b.Inc(v)
}
type TestMeter struct {
Rate float64
NotReady bool
WindowSize time.Duration
}
func (tm *TestMeter) GetWindowSize() time.Duration {
return tm.WindowSize
}
func (tm *TestMeter) IsReady() bool {
return !tm.NotReady
}
func (tm *TestMeter) GetRate() float64 {
return tm.Rate
}