Line data    Source code

       1              : use tracing_opentelemetry::OpenTelemetryLayer;
       2              : use tracing_subscriber::{
       3              :     filter::{EnvFilter, LevelFilter},
       4              :     prelude::*,
       5              : };
       6              : 
       7              : /// Initialize logging and OpenTelemetry tracing and exporter.
       8              : ///
       9              : /// Logging can be configured using `RUST_LOG` environment variable.
      10              : ///
      11              : /// OpenTelemetry is configured with OTLP/HTTP exporter. It picks up
      12              : /// configuration from environment variables. For example, to change the
      13              : /// destination, set `OTEL_EXPORTER_OTLP_ENDPOINT=http://jaeger:4318`.
      14              : /// See <https://opentelemetry.io/docs/reference/specification/sdk-environment-variables>
      15            0 : pub async fn init() -> anyhow::Result<LoggingGuard> {
      16            0 :     let env_filter = EnvFilter::builder()
      17            0 :         .with_default_directive(LevelFilter::INFO.into())
      18            0 :         .from_env_lossy()
      19            0 :         .add_directive("azure_core::policies::transport=off".parse().unwrap());
      20            0 : 
      21            0 :     let fmt_layer = tracing_subscriber::fmt::layer()
      22            0 :         .with_ansi(false)
      23            0 :         .with_writer(std::io::stderr)
      24            0 :         .with_target(false);
      25              : 
      26            0 :     let otlp_layer = tracing_utils::init_tracing("proxy")
      27            0 :         .await
      28            0 :         .map(OpenTelemetryLayer::new);
      29            0 : 
      30            0 :     tracing_subscriber::registry()
      31            0 :         .with(env_filter)
      32            0 :         .with(otlp_layer)
      33            0 :         .with(fmt_layer)
      34            0 :         .try_init()?;
      35              : 
      36            0 :     Ok(LoggingGuard)
      37            0 : }
      38              : 
      39              : pub struct LoggingGuard;
      40              : 
      41              : impl Drop for LoggingGuard {
      42            0 :     fn drop(&mut self) {
      43            0 :         // Shutdown trace pipeline gracefully, so that it has a chance to send any
      44            0 :         // pending traces before we exit.
      45            0 :         tracing::info!("shutting down the tracing machinery");
      46            0 :         tracing_utils::shutdown_tracing();
      47            0 :     }
      48              : }