cost-model/core/pkg/stats/stats.go

package stats

import (
	"errors"
	"fmt"
	"math"
)

type StatType string

const (
	Val StatType = ""
	Avg StatType = "avg"
	Max StatType = "max"
	Min StatType = "min"
	P95 StatType = "p95"
	P85 StatType = "p85"
)

type Stats map[StatType]float64

func NewStats(capacity ...int) Stats {
	if len(capacity) == 1 {
		s := make(map[StatType]float64, capacity[0])
		return s
	}

	return map[StatType]float64{}
}

func (s Stats) Value() (float64, bool) {
	if s == nil {
		return 0.0, false
	}

	val, ok := s[Val]

	return val, ok
}

func (s Stats) Avg() (float64, bool) {
	if s == nil {
		return 0.0, false
	}

	val, ok := s[Avg]

	return val, ok
}

func (s Stats) Max() (float64, bool) {
	if s == nil {
		return 0.0, false
	}

	val, ok := s[Max]

	return val, ok
}

func (s Stats) Min() (float64, bool) {
	if s == nil {
		return 0.0, false
	}

	val, ok := s[Min]

	return val, ok
}

func (s Stats) P95() (float64, bool) {
	if s == nil {
		return 0.0, false
	}

	val, ok := s[P95]

	return val, ok
}

func (s Stats) P85() (float64, bool) {
	if s == nil {
		return 0.0, false
	}

	val, ok := s[P85]

	return val, ok
}

func (s Stats) Sanitize() error {
	if s == nil {
		return nil
	}

	var errs []error

	for t := range s {
		if math.IsNaN(s[t]) {
			delete(s, t)
			errs = append(errs, fmt.Errorf("%v is NaN", t))
		}
		if math.IsInf(s[t], 0) {
			delete(s, t)
			errs = append(errs, fmt.Errorf("%v is Inf", t))
		}
	}

	if len(errs) > 0 {
		errStr := fmt.Sprintf("%d errors:", len(errs))
		for _, e := range errs {
			errStr += fmt.Sprintf(" [%s]", e)
		}
		return errors.New(errStr)
	}

	return nil
}