tracing setup

src/setup_tracing.rs

@@ -0,0 +1,62 @@
+use crate::config::TracingConfig;
+use opentelemetry::global;
+use opentelemetry::trace::TracerProvider;
+use opentelemetry::KeyValue;
+use opentelemetry_otlp::WithExportConfig;
+use opentelemetry_sdk::propagation::TraceContextPropagator;
+use opentelemetry_sdk::trace::{self, BatchConfig, Tracer};
+use opentelemetry_sdk::{runtime, Resource};
+use opentelemetry_semantic_conventions::resource::{SERVICE_NAME, SERVICE_VERSION};
+use opentelemetry_semantic_conventions::SCHEMA_URL;
+use std::time::Duration;
+use tracing::level_filters::LevelFilter;
+use tracing_opentelemetry::OpenTelemetryLayer;
+use tracing_subscriber::layer::SubscriberExt;
+use tracing_subscriber::util::SubscriberInitExt;
+use tracing_subscriber::EnvFilter;
+
+pub fn init_tracer() -> Tracer {
+    let otel_exporter = opentelemetry_otlp::new_exporter()
+        .tonic()
+        .with_timeout(Duration::from_secs(1));
+
+    let tracer_provider = opentelemetry_otlp::new_pipeline()
+        .tracing()
+        .with_exporter(otel_exporter)
+        .with_trace_config(
+            trace::Config::default().with_resource(Resource::from_schema_url(
+                [
+                    KeyValue::new(SERVICE_NAME, env!("CARGO_PKG_NAME")),
+                    KeyValue::new(SERVICE_VERSION, env!("CARGO_PKG_VERSION")),
+                ],
+                SCHEMA_URL,
+            )),
+        )
+        .with_batch_config(BatchConfig::default())
+        .install_batch(runtime::Tokio)
+        .expect("Failed to install tracer");
+
+    global::set_tracer_provider(tracer_provider.clone());
+    tracer_provider.tracer("rustical")
+}
+
+pub fn setup_tracing(config: &TracingConfig) {
+    let fmt_layer = tracing_subscriber::fmt::layer();
+    let filter_layer = EnvFilter::builder()
+        .with_default_directive(LevelFilter::WARN.into())
+        .from_env_lossy()
+        .add_directive("h2=warn".parse().unwrap())
+        .add_directive("hyper_util=warn".parse().unwrap())
+        .add_directive("tower=warn".parse().unwrap());
+
+    let registry = tracing_subscriber::registry()
+        .with(filter_layer)
+        .with(fmt_layer);
+
+    if config.opentelemetry {
+        global::set_text_map_propagator(TraceContextPropagator::new());
+        registry.with(OpenTelemetryLayer::new(init_tracer())).init();
+    } else {
+        registry.init();
+    }
+}