use std::sync::{
atomic::{AtomicUsize, Ordering},
Arc,
};
macro_rules! metrics {
(
$pub_container:ty {
$field:ident: $internal_container:ident {
$(
$(#[$meta:meta])*
$metric:ident: $ty:ident -> $pub_ty:ident,
)*
}
}
) => {
#[derive(Default)]
pub(crate) struct $internal_container {
$(
$(#[$meta])*
pub(crate) $metric: $ty,
)*
}
impl $internal_container {
pub fn sample(&self) -> impl Iterator<Item = emit::metric::Metric<'static, emit::empty::Empty>> + 'static {
let $internal_container { $($metric),* } = self;
[$(
emit::metric::Metric::new(
emit::pkg!(),
stringify!($metric),
<$ty>::AGG,
emit::empty::Empty,
$metric.sample(),
emit::empty::Empty,
),
)*]
.into_iter()
}
}
impl $pub_container {
$(
$(#[$meta])*
pub fn $metric(&self) -> $pub_ty {
self.$field.$metric.sample()
}
)*
}
};
}
#[derive(Default)]
pub(crate) struct Counter(AtomicUsize);
impl Counter {
const AGG: &'static str = emit::well_known::METRIC_AGG_COUNT;
pub fn increment(&self) {
self.increment_by(1);
}
pub fn increment_by(&self, by: usize) {
self.0.fetch_add(by, Ordering::Relaxed);
}
pub fn sample(&self) -> usize {
self.0.load(Ordering::Relaxed)
}
}
metrics!(
FileSetMetrics {
metrics: InternalMetrics {
file_set_read_failed: Counter -> usize,
file_open_failed: Counter -> usize,
file_create: Counter -> usize,
file_create_failed: Counter -> usize,
file_write_failed: Counter -> usize,
file_delete: Counter -> usize,
file_delete_failed: Counter -> usize,
event_format_failed: Counter -> usize,
configuration_failed: Counter -> usize,
}
}
);
pub struct FileSetMetrics {
pub(crate) channel_metrics: Option<emit_batcher::ChannelMetrics<crate::EventBatch>>,
pub(crate) metrics: Arc<InternalMetrics>,
}
impl emit::metric::Source for FileSetMetrics {
fn sample_metrics<S: emit::metric::sampler::Sampler>(&self, sampler: S) {
self.channel_metrics
.sample_metrics(emit::metric::sampler::from_fn(|metric| {
sampler.metric(
metric
.by_ref()
.with_name(&format!("file_{}", metric.name()))
.with_mdl(emit::pkg!()),
);
}));
for metric in self.metrics.sample() {
sampler.metric(metric);
}
}
}