diff options
Diffstat (limited to 'ma/ma.go')
| -rw-r--r-- | ma/ma.go | 25 |
1 files changed, 21 insertions, 4 deletions
@@ -1,6 +1,8 @@ package ma import ( + "math" + "github.com/hawkinsw/goresponsiveness/saturating" ) @@ -8,25 +10,40 @@ import ( // in Go (1.18). type MovingAverage struct { intervals int - instants []uint64 + instants []float64 index int divisor *saturating.SaturatingInt } func NewMovingAverage(intervals int) *MovingAverage { - return &MovingAverage{instants: make([]uint64, intervals), intervals: intervals, divisor: saturating.NewSaturatingInt(intervals)} + return &MovingAverage{instants: make([]float64, intervals), intervals: intervals, divisor: saturating.NewSaturatingInt(intervals)} } -func (ma *MovingAverage) AddMeasurement(measurement uint64) { +func (ma *MovingAverage) AddMeasurement(measurement float64) { ma.instants[ma.index] = measurement ma.divisor.Add(1) ma.index = (ma.index + 1) % ma.intervals } func (ma *MovingAverage) CalculateAverage() float64 { - total := uint64(0) + total := float64(0) for i := 0; i < ma.intervals; i++ { total += ma.instants[i] } return float64(total) / float64(ma.divisor.Value()) } + +func (ma *MovingAverage) ConsistentWithin(limit float64) bool { + previous := ma.instants[0] + for i := 1; i < ma.intervals; i++ { + current := ma.instants[i] + delta := math.Abs(current - previous) + percentChange := (float64(delta) / (float64(current+previous) / 2.0)) * float64(100) + + previous = current + if math.Abs(percentChange) > limit { + return false + } + } + return true +} |
