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!(
EmitOpenTelemetryMetrics {
metrics: InternalMetrics {
span_unexpected_close: Counter -> usize,
span_unexpected_emit: Counter -> usize,
}
}
);
pub struct EmitOpenTelemetryMetrics {
pub(crate) metrics: Arc<InternalMetrics>,
}
impl emit::metric::Source for EmitOpenTelemetryMetrics {
fn sample_metrics<S: emit::metric::sampler::Sampler>(&self, sampler: S) {
for metric in self.metrics.sample() {
sampler.metric(metric);
}
}
}