#![deny(warnings, missing_docs, clippy::all, rustdoc::broken_intra_doc_links)]
pub mod macros;
mod noop_stats;
pub mod thread_local_aggregator;
pub mod prelude {
pub use stats_traits::dynamic_stat_types::DynamicCounter;
pub use stats_traits::dynamic_stat_types::DynamicHistogram;
pub use stats_traits::dynamic_stat_types::DynamicSingletonCounter;
pub use stats_traits::dynamic_stat_types::DynamicTimeseries;
pub use stats_traits::stat_types::Counter;
pub use stats_traits::stat_types::CounterStatic;
pub use stats_traits::stat_types::Histogram;
pub use stats_traits::stat_types::HistogramStatic;
pub use stats_traits::stat_types::Timeseries;
pub use stats_traits::stat_types::TimeseriesStatic;
pub use crate::define_stats;
pub use crate::define_stats_struct;
}
use std::sync::RwLock;
use stats_traits::stat_types::BoxSingletonCounter;
use stats_traits::stats_manager::BoxStatsManager;
use stats_traits::stats_manager::StatsManagerFactory;
pub use self::thread_local_aggregator::schedule_stats_aggregation_preview;
static STATS_MANAGER_FACTORY: RwLock<Option<Box<dyn StatsManagerFactory + Send + Sync>>> =
RwLock::new(None);
pub fn register_stats_manager_factory(factory: impl StatsManagerFactory + Send + Sync + 'static) {
let mut global_factory = STATS_MANAGER_FACTORY.write().expect("poisoned lock");
assert!(
global_factory.is_none(),
"Called stats::stats_manager::register_stats_manager_factory more than once"
);
global_factory.replace(Box::new(factory));
}
#[doc(hidden)]
pub fn create_stats_manager() -> BoxStatsManager {
if let Some(factory) = STATS_MANAGER_FACTORY
.read()
.expect("poisoned lock")
.as_ref()
{
return factory.create();
}
let mut write_lock = STATS_MANAGER_FACTORY.write().expect("poisoned lock");
let factory = write_lock.get_or_insert_with(get_default_stats_manager_factory);
factory.create()
}
fn get_default_stats_manager_factory() -> Box<dyn StatsManagerFactory + Send + Sync> {
#[cfg(fbcode_build)]
{
Box::new(::stats_facebook::ThreadLocalStatsFactory)
}
#[cfg(not(fbcode_build))]
{
Box::new(crate::noop_stats::NoopStatsFactory)
}
}
#[doc(hidden)]
pub fn create_singleton_counter(name: String) -> BoxSingletonCounter {
#[cfg(fbcode_build)]
{
Box::new(::stats_facebook::singleton_counter::ServiceDataSingletonCounter::new(name))
}
#[cfg(not(fbcode_build))]
{
let _ = name;
Box::new(crate::noop_stats::Noop)
}
}