From 7edcde3d5e8a3086cd11d6cc8f74896120fb7003 Mon Sep 17 00:00:00 2001 From: Simen Bekkhus Date: Tue, 10 Sep 2024 14:56:35 +0200 Subject: [PATCH] feat: Support OTel 0.25 --- reqwest-tracing/Cargo.toml | 5 ++++- reqwest-tracing/src/lib.rs | 1 + reqwest-tracing/src/middleware.rs | 1 + reqwest-tracing/src/otel.rs | 30 ++++++++++++++++++++++++++++++ 4 files changed, 36 insertions(+), 1 deletion(-) diff --git a/reqwest-tracing/Cargo.toml b/reqwest-tracing/Cargo.toml index dd4f27c..d99c032 100644 --- a/reqwest-tracing/Cargo.toml +++ b/reqwest-tracing/Cargo.toml @@ -15,6 +15,7 @@ opentelemetry_0_21 = ["opentelemetry_0_21_pkg", "tracing-opentelemetry_0_22_pkg" opentelemetry_0_22 = ["opentelemetry_0_22_pkg", "tracing-opentelemetry_0_23_pkg"] opentelemetry_0_23 = ["opentelemetry_0_23_pkg", "tracing-opentelemetry_0_24_pkg"] opentelemetry_0_24 = ["opentelemetry_0_24_pkg", "tracing-opentelemetry_0_25_pkg"] +opentelemetry_0_25 = ["opentelemetry_0_25_pkg", "tracing-opentelemetry_0_26_pkg"] # This feature ensures that both the old (deprecated) and new attributes are published simultaneously. # By doing so, we maintain backward compatibility, allowing existing code that relies on the old attributes # to continue functioning while encouraging the transition to the new attributes. @@ -35,11 +36,13 @@ opentelemetry_0_21_pkg = { package = "opentelemetry", version = "0.21.0", option opentelemetry_0_22_pkg = { package = "opentelemetry", version = "0.22.0", optional = true } opentelemetry_0_23_pkg = { package = "opentelemetry", version = "0.23.0", optional = true } opentelemetry_0_24_pkg = { package = "opentelemetry", version = "0.24.0", optional = true } +opentelemetry_0_25_pkg = { package = "opentelemetry", version = "0.25.0", optional = true } tracing-opentelemetry_0_21_pkg = { package = "tracing-opentelemetry", version = "0.21.0", optional = true } tracing-opentelemetry_0_22_pkg = { package = "tracing-opentelemetry", version = "0.22.0", optional = true } tracing-opentelemetry_0_23_pkg = { package = "tracing-opentelemetry", version = "0.23.0", optional = true } tracing-opentelemetry_0_24_pkg = { package = "tracing-opentelemetry", version = "0.24.0", optional = true } tracing-opentelemetry_0_25_pkg = { package = "tracing-opentelemetry", version = "0.25.0", optional = true } +tracing-opentelemetry_0_26_pkg = { package = "tracing-opentelemetry", version = "0.26.0", optional = true } [target.'cfg(target_arch = "wasm32")'.dependencies] getrandom = { version = "0.2.0", features = ["js"] } @@ -54,9 +57,9 @@ opentelemetry_sdk_0_21 = { package = "opentelemetry_sdk", version = "0.21.0", fe opentelemetry_sdk_0_22 = { package = "opentelemetry_sdk", version = "0.22.0", features = ["trace"] } opentelemetry_sdk_0_23 = { package = "opentelemetry_sdk", version = "0.23.0", features = ["trace"] } opentelemetry_sdk_0_24 = { package = "opentelemetry_sdk", version = "0.24.1", features = ["trace"] } +opentelemetry_sdk_0_25 = { package = "opentelemetry_sdk", version = "0.25.0", features = ["trace"] } opentelemetry_stdout_0_1 = { package = "opentelemetry-stdout", version = "0.1.0", features = ["trace"] } opentelemetry_stdout_0_2 = { package = "opentelemetry-stdout", version = "0.2.0", features = ["trace"] } opentelemetry_stdout_0_3 = { package = "opentelemetry-stdout", version = "0.3.0", features = ["trace"] } opentelemetry_stdout_0_4 = { package = "opentelemetry-stdout", version = "0.4.0", features = ["trace"] } opentelemetry_stdout_0_5 = { package = "opentelemetry-stdout", version = "0.5.0", features = ["trace"] } - diff --git a/reqwest-tracing/src/lib.rs b/reqwest-tracing/src/lib.rs index 44e3b89..6d3d2dd 100644 --- a/reqwest-tracing/src/lib.rs +++ b/reqwest-tracing/src/lib.rs @@ -89,6 +89,7 @@ mod middleware; feature = "opentelemetry_0_22", feature = "opentelemetry_0_23", feature = "opentelemetry_0_24", + feature = "opentelemetry_0_25", ))] mod otel; mod reqwest_otel_span_builder; diff --git a/reqwest-tracing/src/middleware.rs b/reqwest-tracing/src/middleware.rs index 43b305d..8afb54a 100644 --- a/reqwest-tracing/src/middleware.rs +++ b/reqwest-tracing/src/middleware.rs @@ -51,6 +51,7 @@ where feature = "opentelemetry_0_22", feature = "opentelemetry_0_23", feature = "opentelemetry_0_24", + feature = "opentelemetry_0_25", ))] let req = if extensions.get::().is_none() { // Adds tracing headers to the given request to propagate the OpenTelemetry context to downstream revivers of the request. diff --git a/reqwest-tracing/src/otel.rs b/reqwest-tracing/src/otel.rs index fceb636..5162d85 100644 --- a/reqwest-tracing/src/otel.rs +++ b/reqwest-tracing/src/otel.rs @@ -40,6 +40,13 @@ pub fn inject_opentelemetry_context_into_request(mut request: Request) -> Reques injector.inject_context(&context, &mut RequestCarrier::new(&mut request)) }); + #[cfg(feature = "opentelemetry_0_25")] + opentelemetry_0_25_pkg::global::get_text_map_propagator(|injector| { + use tracing_opentelemetry_0_26_pkg::OpenTelemetrySpanExt; + let context = Span::current().context(); + injector.inject_context(&context, &mut RequestCarrier::new(&mut request)) + }); + request } @@ -102,6 +109,13 @@ impl<'a> opentelemetry_0_24_pkg::propagation::Injector for RequestCarrier<'a> { } } +#[cfg(feature = "opentelemetry_0_25")] +impl<'a> opentelemetry_0_25_pkg::propagation::Injector for RequestCarrier<'a> { + fn set(&mut self, key: &str, value: String) { + self.set_inner(key, value) + } +} + #[cfg(test)] mod test { use std::sync::OnceLock; @@ -237,6 +251,22 @@ mod test { subscriber.with(telemetry) }; + #[cfg(feature = "opentelemetry_0_25")] + let subscriber = { + use opentelemetry_0_25_pkg::trace::TracerProvider; + + let provider = opentelemetry_sdk_0_25::trace::TracerProvider::builder().build(); + + let tracer = provider.tracer_builder("reqwest").build(); + let _ = opentelemetry_0_25_pkg::global::set_tracer_provider(provider); + opentelemetry_0_25_pkg::global::set_text_map_propagator( + opentelemetry_sdk_0_25::propagation::TraceContextPropagator::new(), + ); + + let telemetry = tracing_opentelemetry_0_26_pkg::layer().with_tracer(tracer); + subscriber.with(telemetry) + }; + tracing::subscriber::set_global_default(subscriber).unwrap(); });