cc-metric-collector/internal/ccUnits/ccUnitScale.go

package ccunits

import "regexp"

type Scale float64

const (
	Base  Scale = iota
	Peta        = 1e15
	Tera        = 1e12
	Giga        = 1e9
	Mega        = 1e6
	Kilo        = 1e3
	Milli       = 1e-3
	Micro       = 1e-6
	Nano        = 1e-9
	Kibi        = 1024
	Mebi        = 1024 * 1024
	Gibi        = 1024 * 1024 * 1024
	Tebi        = 1024 * 1024 * 1024 * 1024
)
const prefixRegexStr = `^([kKmMgGtTpP]?[i]?)(.*)`

var prefixRegex = regexp.MustCompile(prefixRegexStr)

func (s *Scale) String() string {
	switch *s {
	case Base:
		return ""
	case Kilo:
		return "Kilo"
	case Mega:
		return "Mega"
	case Giga:
		return "Giga"
	case Tera:
		return "Tera"
	case Peta:
		return "Peta"
	case Milli:
		return "Milli"
	case Micro:
		return "Micro"
	case Nano:
		return "Nano"
	case Kibi:
		return "Kibi"
	case Mebi:
		return "Mebi"
	case Gibi:
		return "Gibi"
	case Tebi:
		return "Tebi"
	default:
		return "Unkn"
	}
}

func (s *Scale) Prefix() string {
	switch *s {
	case Base:
		return ""
	case Kilo:
		return "K"
	case Mega:
		return "M"
	case Giga:
		return "G"
	case Tera:
		return "T"
	case Peta:
		return "P"
	case Milli:
		return "m"
	case Micro:
		return "u"
	case Nano:
		return "n"
	case Kibi:
		return "Ki"
	case Mebi:
		return "Mi"
	case Gibi:
		return "Gi"
	case Tebi:
		return "Ti"
	default:
		return "<unkn>"
	}
}

func NewScale(prefix string) Scale {
	switch prefix {
	case "k":
		return Kilo
	case "K":
		return Kilo
	case "m":
		return Milli
	case "M":
		return Mega
	case "g":
		return Giga
	case "G":
		return Giga
	case "t":
		return Tera
	case "T":
		return Tera
	case "u":
		return Micro
	case "n":
		return Nano
	case "ki":
		return Kibi
	case "Ki":
		return Kibi
	case "Mi":
		return Mebi
	case "gi":
		return Gibi
	case "Gi":
		return Gibi
	case "Ti":
		return Tebi
	case "":
		return Base
	default:
		return Base
	}
}