platform_core/
telemetry.rs

1use crate::config::{LogFormat, TelemetryConfig};
2use crate::error::{AppError, AppResult, ErrorCode};
3use opentelemetry::KeyValue;
4use opentelemetry::trace::TracerProvider as _;
5use opentelemetry_otlp::WithExportConfig;
6use opentelemetry_sdk::Resource;
7use opentelemetry_sdk::trace::{RandomIdGenerator, Sampler, SdkTracerProvider};
8use std::sync::OnceLock;
9use tracing_subscriber::{EnvFilter, fmt, layer::SubscriberExt, util::SubscriberInitExt};
10
11static OTEL_PROVIDER: OnceLock<SdkTracerProvider> = OnceLock::new();
12
13pub fn init(config: &TelemetryConfig) -> AppResult<()> {
14    let env_filter = EnvFilter::try_new(&config.log_level)
15        .or_else(|_| EnvFilter::try_new("info"))
16        .expect("fallback tracing filter must be valid");
17    let otel_layer = config.otlp_endpoint.as_deref().and_then(build_otel_layer);
18
19    match (config.log_format, otel_layer) {
20        (LogFormat::Compact, Some(otel_layer)) => {
21            tracing_subscriber::registry()
22                .with(otel_layer)
23                .with(env_filter)
24                .with(fmt::layer().compact().with_target(false))
25                .try_init()
26                .ok();
27        }
28        (LogFormat::Compact, None) => {
29            tracing_subscriber::registry()
30                .with(env_filter)
31                .with(fmt::layer().compact().with_target(false))
32                .try_init()
33                .ok();
34        }
35        (LogFormat::Json, Some(otel_layer)) => {
36            tracing_subscriber::registry()
37                .with(otel_layer)
38                .with(env_filter)
39                .with(fmt::layer().json())
40                .try_init()
41                .ok();
42        }
43        (LogFormat::Json, None) => {
44            tracing_subscriber::registry()
45                .with(env_filter)
46                .with(fmt::layer().json())
47                .try_init()
48                .ok();
49        }
50    }
51
52    Ok(())
53}
54
55pub fn force_flush() -> AppResult<()> {
56    let Some(provider) = OTEL_PROVIDER.get() else {
57        return Ok(());
58    };
59
60    provider.force_flush().map_err(|error| {
61        AppError::new(ErrorCode::ExternalDependency, "OpenTelemetry flush failed")
62            .with_source(error)
63    })
64}
65
66fn build_otel_layer(
67    endpoint: &str,
68) -> Option<
69    tracing_opentelemetry::OpenTelemetryLayer<
70        tracing_subscriber::Registry,
71        opentelemetry_sdk::trace::Tracer,
72    >,
73> {
74    let exporter = match opentelemetry_otlp::SpanExporter::builder()
75        .with_tonic()
76        .with_endpoint(endpoint.to_owned())
77        .build()
78    {
79        Ok(exporter) => exporter,
80        Err(error) => {
81            eprintln!("failed to configure OTLP span exporter: {error}");
82            return None;
83        }
84    };
85
86    let provider = SdkTracerProvider::builder()
87        .with_sampler(Sampler::AlwaysOn)
88        .with_id_generator(RandomIdGenerator::default())
89        .with_resource(
90            Resource::builder()
91                .with_service_name("lenso")
92                .with_attribute(KeyValue::new(
93                    "deployment.environment.name",
94                    std::env::var("APP_ENV").unwrap_or_else(|_| "local".to_owned()),
95                ))
96                .build(),
97        )
98        .with_batch_exporter(exporter)
99        .build();
100    let tracer = provider.tracer("lenso-runtime");
101    opentelemetry::global::set_tracer_provider(provider.clone());
102    OTEL_PROVIDER.set(provider).ok();
103
104    Some(tracing_opentelemetry::layer().with_tracer(tracer))
105}
platform_core/telemetry.rs

platform_core/
telemetry.rs
