use std::sync::atomic::{AtomicUsize, Ordering};
#[derive(Debug, Default)]
pub struct ActiveStats {
clean: AtomicUsize,
squash_ancient: AtomicUsize,
shrink: AtomicUsize,
hash: AtomicUsize,
flush: AtomicUsize,
}
#[derive(Debug, Copy, Clone)]
pub enum ActiveStatItem {
Clean,
Shrink,
SquashAncient,
Hash,
Flush,
}
pub struct ActiveStatGuard<'a> {
stats: &'a ActiveStats,
item: ActiveStatItem,
}
impl<'a> Drop for ActiveStatGuard<'a> {
fn drop(&mut self) {
self.stats.update_and_log(self.item, |stat| {
stat.fetch_sub(1, Ordering::Relaxed).wrapping_sub(1)
});
}
}
impl ActiveStats {
#[must_use]
pub fn activate(&self, stat: ActiveStatItem) -> ActiveStatGuard<'_> {
self.update_and_log(stat, |stat| {
stat.fetch_add(1, Ordering::Relaxed).wrapping_add(1)
});
ActiveStatGuard {
stats: self,
item: stat,
}
}
fn update_and_log(&self, item: ActiveStatItem, modify_stat: impl Fn(&AtomicUsize) -> usize) {
let stat = match item {
ActiveStatItem::Clean => &self.clean,
ActiveStatItem::Shrink => &self.shrink,
ActiveStatItem::SquashAncient => &self.squash_ancient,
ActiveStatItem::Hash => &self.hash,
ActiveStatItem::Flush => &self.flush,
};
let value = modify_stat(stat);
match item {
ActiveStatItem::Clean => datapoint_info!("accounts_db_active", ("clean", value, i64)),
ActiveStatItem::SquashAncient => {
datapoint_info!("accounts_db_active", ("squash_ancient", value, i64))
}
ActiveStatItem::Shrink => {
datapoint_info!("accounts_db_active", ("shrink", value, i64))
}
ActiveStatItem::Hash => datapoint_info!("accounts_db_active", ("hash", value, i64)),
ActiveStatItem::Flush => datapoint_info!("accounts_db_active", ("flush", value, i64)),
};
}
}