set_global_default tracer

Author: KSDaemon

rust/cubestore/cubestore/src/telemetry/otel_tracing.rs

@@ -7,6 +7,8 @@ use opentelemetry::trace::{SpanKind, Tracer, TracerProvider};
 use opentelemetry::{KeyValue, StringValue};
 use opentelemetry_sdk::logs::{LogRecord as LogRecordSDK, Logger as LoggerSDK};
 use opentelemetry_sdk::trace::Tracer as TracerSDK;
+use tracing_subscriber::layer::SubscriberExt;
+use tracing_subscriber::Registry;
 
 const OTEL_SERVICE_NAME: &str = "cubestore";
 
@@ -15,6 +17,14 @@ lazy_static! {
     pub static ref OT_LOGGER: LoggerSDK = init_logging().unwrap();
 }
 
+pub fn init_tracing_telemetry() {
+    let telemetry = tracing_opentelemetry::layer().with_tracer(OT_TRACER.clone());
+    let subscriber = Registry::default().with(telemetry);
+
+    tracing::subscriber::set_global_default(subscriber)
+        .expect("setting default tracing subscriber failed");
+}
+
 pub fn init_tracing() -> Result<TracerSDK, Box<dyn std::error::Error>> {
     let otlp_exporter = opentelemetry_otlp::new_exporter()
         .http()

rust/cubestore/cubestore/src/util/logger.rs

@@ -1,3 +1,4 @@
+use crate::telemetry::otel_tracing::init_tracing_telemetry;
 use crate::telemetry::{OpenTelemetryLogger, ReportingLogger};
 use log::{Level, Log, Metadata, Record};
 use simple_logger::SimpleLogger;
@@ -33,6 +34,7 @@ pub fn init_cube_logger(enable_telemetry: bool) {
         // No need to introduce special env vars, let's use the de facto standard ones
         if let Ok(_) = env::var("OTEL_EXPORTER_OTLP_ENDPOINT") {
             logger = Box::new(OpenTelemetryLogger::new(logger));
+            init_tracing_telemetry();
         } else {
             logger = Box::new(ReportingLogger::new(logger));
         }