logos-storage-nim/codex/utils/statsummary.nim
2025-01-15 10:46:58 +01:00

55 lines
1.3 KiB
Nim

import pkg/metrics
when defined(metrics):
type StatSummary* = ref object of Collector
hasValues: bool
min: float64
max: float64
ravg: float64
method collect(st: StatSummary, output: MetricHandler) =
if not st.hasValues:
return
let timestamp = st.now()
output(
name = st.name & "_min",
value = st.min,
timestamp = timestamp
)
output(
name = st.name & "_max",
value = st.max,
timestamp = timestamp
)
output(
name = st.name & "_ravg",
value = st.ravg,
timestamp = timestamp
)
proc declareStatSummary*(name: string, help: string = ""): StatSummary =
when defined(metrics):
result = StatSummary.newCollector(name, help)
result.hasValues = false
else:
return IgnoredCollector
proc observeStatSummary(st: StatSummary, value: float64) =
if st.hasValues:
if value < st.min:
st.min = value
if value > st.max:
st.max = value
st.ravg = (st.ravg + value) / float64(2)
else:
st.hasValues = true
st.min = value
st.max = value
st.ravg = value
template observe*(statSummary: StatSummary | type IgnoredCollector, amount: int64 | float64 = 1) =
when defined(metrics) and statSummary is not IgnoredCollector:
{.gcsafe.}:
observeStatSummary(statSummary, amount.float64)