metry 0.1.1

All-in-one telemetry framework, based on tracing crate.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use opentelemetry_appender_tracing::layer::OpenTelemetryTracingBridge;
use opentelemetry_sdk::logs::{Config, LoggerProvider};
use opentelemetry_stdout::LogExporter;
use tracing_core::Subscriber;
use tracing_subscriber::{registry::LookupSpan, Layer};

use crate::layers::TelemetryLayerBuilder;

/// Configuration options for a logs [LayerBuilder].
#[derive(Debug, Default)]
pub struct LogOptions {
    config: Option<Config>,
}

impl LogOptions {
    pub fn with_log_config(config: Config) -> Self {
        LogOptions {
            config: Some(config),
        }
    }
}

#[derive(Debug)]
pub enum LayerBuilder {
    /// Logs layer emitting to stdout.
    Stdout(LogOptions),
    /// Logs layer emitting to stderr.
    Stderr(LogOptions),
}

impl TelemetryLayerBuilder for LayerBuilder {
    async fn layer<S>(self) -> Result<Box<dyn Layer<S> + Send + Sync + 'static>, String>
    where
        S: Subscriber,
        for<'a> S: LookupSpan<'a>,
    {
        // Configure the writer based on the desired log target:
        let (logger_exporter, config) = match self {
            LayerBuilder::Stdout(LogOptions { config }) => (LogExporter::builder().build(), config),
            LayerBuilder::Stderr(LogOptions { config }) => (
                LogExporter::builder()
                    .with_writer(std::io::stderr())
                    .build(),
                config,
            ),
        };
        let logger_provider = LoggerProvider::builder()
            .with_config(config.unwrap_or_default())
            .with_simple_exporter(logger_exporter)
            .build();
        let logs_layer = OpenTelemetryTracingBridge::new(&logger_provider);
        Ok(Box::new(logs_layer))
    }
}