tracing-setup 1.0.4

crate::ix!();

pub trait BufferedSubscriber: Subscriber + Flushable {}

#[derive(Debug)]
pub struct BufferedSubscriberLayer<S> {
    inner:          tracing_subscriber::layer::Layered<BufferedLayer, S>,
    buffered_layer: Arc<BufferedLayer>,
}

impl<S: Subscriber> Flushable for BufferedSubscriberLayer<S> {

    fn flush(&self) {
        self.buffered_layer.flush();
    }
}

impl<S: Subscriber> Subscriber for BufferedSubscriberLayer<S> {
    fn enabled(&self, metadata: &tracing::Metadata<'_>) -> bool {
        self.inner.enabled(metadata)
    }

    fn new_span(&self, span: &tracing::span::Attributes<'_>) -> tracing::Id {
        self.inner.new_span(span)
    }

    fn record(&self, span: &tracing::Id, values: &tracing::span::Record<'_>) {
        self.inner.record(span, values);
    }

    fn record_follows_from(&self, span: &tracing::Id, follows: &tracing::Id) {
        self.inner.record_follows_from(span, follows);
    }

    fn event(&self, event: &tracing::Event<'_>) {
        self.inner.event(event);
    }

    fn enter(&self, span: &tracing::Id) {
        self.inner.enter(span);
    }

    fn exit(&self, span: &tracing::Id) {
        self.inner.exit(span);
    }
}

pub fn setup_default_buffered_tracing() -> Arc<BufferedSubscriberLayer<Registry>> 
{
    setup_buffered_tracing(None, EventPrinter::default())
}

pub fn setup_buffered_tracing(
    tag: Option<&str>,
    printer: EventPrinter,
) -> Arc<BufferedSubscriberLayer<Registry>>
{
    let mut layer = match tag {
        Some(t) => BufferedLayer::new_with_tag(t),
        None    => BufferedLayer::default(),
    };

    // Let the layer know how to print
    layer.set_event_printer(printer);

    let subscriber = Registry::default().with(layer.clone());

    Arc::new(BufferedSubscriberLayer {
        inner: subscriber,
        buffered_layer: layer.into(),
    })
}