core/tests/common/tracing.rs
26 lines
use opentelemetry::global;use opentelemetry::trace::TracerProvider as _;use opentelemetry_sdk::propagation::TraceContextPropagator;use opentelemetry_sdk::trace::SdkTracerProvider;use tracing::dispatcher::DefaultGuard;use tracing_subscriber::layer::SubscriberExt;use tracing_subscriber::util::SubscriberInitExt;pub struct TestTracingContext { _provider: SdkTracerProvider, _guard: DefaultGuard,}pub fn install_test_tracing(tracer_name: &str) -> TestTracingContext { global::set_text_map_propagator(TraceContextPropagator::new()); let provider = SdkTracerProvider::builder().build(); let tracer = provider.tracer(tracer_name.to_string()); let subscriber = tracing_subscriber::registry().with(tracing_opentelemetry::layer().with_tracer(tracer)); TestTracingContext { _provider: provider, _guard: subscriber.set_default(), }}