ratio.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
}