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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72

use std::cmp::max;

use axum_tracing_opentelemetry::{otlp, resource::DetectResource};
use http::header::HeaderMap;
use opentelemetry_http::HeaderInjector;
use tracing_opentelemetry::OpenTelemetrySpanExt;
use tracing_subscriber::{
    filter::LevelFilter,
    fmt::{format::FmtSpan, writer::MakeWriterExt},
    layer::SubscriberExt,
};

pub use axum_tracing_opentelemetry::opentelemetry_tracing_layer;
pub mod propagation;

#[cfg(feature = "integration_test")]
pub mod test;

pub fn init_tracing_with_fallbacks(
    log_level: tracing::Level,
    fallback_service_name: &'static str,
    fallback_service_version: &'static str,
) {
    let otel_rsrc = DetectResource::default()
        .with_fallback_service_name(fallback_service_name)
        .with_fallback_service_version(fallback_service_version)
        .build();
    let otel_tracer =
        otlp::init_tracer(otel_rsrc, otlp::identity).expect("setup of Tracer");
    let otel_layer = tracing_opentelemetry::layer().with_tracer(otel_tracer);

    opentelemetry::global::set_text_map_propagator(
        propagation::TextMapSplitPropagator::default(),
    );

    let fmt_layer = tracing_subscriber::fmt::layer()
        .json()
        .with_timer(tracing_subscriber::fmt::time::uptime())
        .with_span_events(FmtSpan::NEW | FmtSpan::CLOSE)
        .with_writer(std::io::stdout.with_max_level(log_level));

    let level_filter: LevelFilter = max(log_level, tracing::Level::INFO).into();
    let subscriber = tracing_subscriber::registry()
        .with(fmt_layer)
        .with(level_filter)
        .with(otel_layer);
    tracing::subscriber::set_global_default(subscriber).unwrap();
}

#[macro_export]
macro_rules! init_tracing {
    ($log_level:expr) => {
        $crate::init_tracing_with_fallbacks(
            $log_level,
            env!("CARGO_PKG_NAME"),
            env!("CARGO_PKG_VERSION"),
        );
    };
}

pub fn inject_context(headers: &mut HeaderMap) {
    let context = tracing::Span::current().context();
    let mut injector = HeaderInjector(headers);

    opentelemetry::global::get_text_map_propagator(|propagator| {
        propagator.inject_context(&context, &mut injector)
    });
}

pub fn shutdown_signal() {
    opentelemetry::global::shutdown_tracer_provider();
}