use std::{cell::RefCell, collections::HashMap};
pub use crate::domain::metrics::{
LifecycleMetricOutcome, LifecycleMetricPhase, LifecycleMetricRole, LifecycleMetricStage,
};
thread_local! {
static LIFECYCLE_METRICS: RefCell<HashMap<LifecycleMetricKey, u64>> =
RefCell::new(HashMap::new());
}
#[derive(Clone, Copy, Eq, Hash, PartialEq)]
pub struct LifecycleMetricKey {
pub phase: LifecycleMetricPhase,
pub role: LifecycleMetricRole,
pub stage: LifecycleMetricStage,
pub outcome: LifecycleMetricOutcome,
}
pub struct LifecycleMetrics;
impl LifecycleMetrics {
pub fn record(
phase: LifecycleMetricPhase,
role: LifecycleMetricRole,
stage: LifecycleMetricStage,
outcome: LifecycleMetricOutcome,
) {
LIFECYCLE_METRICS.with_borrow_mut(|counts| {
let key = LifecycleMetricKey {
phase,
role,
stage,
outcome,
};
let entry = counts.entry(key).or_insert(0);
*entry = entry.saturating_add(1);
});
}
#[must_use]
pub fn snapshot() -> Vec<(LifecycleMetricKey, u64)> {
LIFECYCLE_METRICS
.with_borrow(std::clone::Clone::clone)
.into_iter()
.collect()
}
#[cfg(test)]
pub fn reset() {
LIFECYCLE_METRICS.with_borrow_mut(HashMap::clear);
}
}