use std::sync::Arc;
use fxhash::FxHashMap;
use metrics::Key;
use metrics_util::Summary;
use elfo_core::{ActorMeta, Local};
use elfo_macros::message;
#[message(ret = Local<Arc<Snapshot>>, elfo = elfo_core)]
pub struct GetSnapshot;
#[derive(Default, Clone)]
pub struct Snapshot {
pub per_group: FxHashMap<String, Metrics>,
pub per_actor: FxHashMap<Arc<ActorMeta>, Metrics>,
}
#[derive(Default, Clone)]
pub struct Metrics {
pub counters: FxHashMap<Key, u64>,
pub gauges: FxHashMap<Key, f64>,
pub distributions: FxHashMap<Key, Distribution>,
}
#[derive(Clone)]
pub struct Distribution {
summary: Arc<Summary>,
sum: f64,
}
impl Default for Distribution {
fn default() -> Self {
let summary = Summary::with_defaults();
Self {
summary: Arc::new(summary),
sum: 0.0,
}
}
}
impl Distribution {
#[inline]
pub fn quantile(&self, q: f64) -> Option<f64> {
self.summary.quantile(q)
}
#[inline]
pub fn min(&self) -> f64 {
self.summary.min()
}
#[inline]
pub fn max(&self) -> f64 {
self.summary.max()
}
#[inline]
pub fn count(&self) -> usize {
self.summary.count()
}
#[inline]
pub fn sum(&self) -> f64 {
self.sum
}
pub(crate) fn record_samples(&mut self, samples: &[f64]) {
let summary = Arc::make_mut(&mut self.summary);
for sample in samples {
summary.add(*sample);
self.sum += *sample;
}
}
pub(crate) fn estimated_size(&self) -> usize {
self.summary.estimated_size() + std::mem::size_of::<Self>()
}
}