arti_testing/
traces.rs

Help
1//! Code to count the traces at different severities.
2
3use std::sync::{
4    atomic::{AtomicUsize, Ordering},
5    Arc,
6};
7use tracing::Subscriber;
8use tracing_subscriber::Layer;
9
10/// Count of the number of tracing events of each severity.
11#[derive(Default)]
12pub(crate) struct TraceCount {
13    /// number of trace events
14    trace: AtomicUsize,
15    /// number of debug events
16    debug: AtomicUsize,
17    /// number of info events
18    info: AtomicUsize,
19    /// number of warn events
20    warn: AtomicUsize,
21    /// number of error events
22    error: AtomicUsize,
23}
24
25impl std::fmt::Display for TraceCount {
26    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
27        std::sync::atomic::fence(Ordering::SeqCst);
28        let trace = self.trace.load(Ordering::SeqCst);
29        let debug = self.debug.load(Ordering::SeqCst);
30        let info = self.info.load(Ordering::SeqCst);
31        let warn = self.warn.load(Ordering::SeqCst);
32        let error = self.error.load(Ordering::SeqCst);
33
34        write!(
35            f,
36            "Trace: {}, Debug: {}, Info: {}, Warn: {}, Error: {}",
37            trace, debug, info, warn, error
38        )
39    }
40}
41
42/// A log subscriber to count the number of events of each severity.
43pub(crate) struct TraceCounter(pub(crate) Arc<TraceCount>);
44
45impl<S: Subscriber> Layer<S> for TraceCounter {
46    fn on_event(
47        &self,
48        event: &tracing::Event<'_>,
49        _ctx: tracing_subscriber::layer::Context<'_, S>,
50    ) {
51        use tracing::Level;
52        let var = match *event.metadata().level() {
53            Level::TRACE => &self.0.trace,
54            Level::DEBUG => &self.0.debug,
55            Level::INFO => &self.0.info,
56            Level::WARN => &self.0.warn,
57            Level::ERROR => &self.0.error,
58        };
59
60        var.fetch_add(1, Ordering::Relaxed);
61    }
62}
arti_testing/traces.rs

arti_testing/
traces.rs
