From 86d148265d51b3a11756a08457abbe7dbb9c5590 Mon Sep 17 00:00:00 2001 From: jerbly Date: Fri, 7 Feb 2025 14:35:18 -0500 Subject: [PATCH 1/2] simplify the span exporter --- crates/weaver_emit/src/spans.rs | 30 ++++++------------------------ 1 file changed, 6 insertions(+), 24 deletions(-) diff --git a/crates/weaver_emit/src/spans.rs b/crates/weaver_emit/src/spans.rs index 74fe752e..1a0cc359 100644 --- a/crates/weaver_emit/src/spans.rs +++ b/crates/weaver_emit/src/spans.rs @@ -157,10 +157,10 @@ mod tests { use opentelemetry_sdk::{trace as sdktrace, Resource}; use futures_util::future::BoxFuture; - use opentelemetry::trace::{SpanKind, TraceError}; + use opentelemetry::trace::SpanKind; use opentelemetry_sdk::export::{self, trace::ExportResult}; use ordered_float::OrderedFloat; - use std::sync::{atomic, Arc, Mutex}; + use std::sync::{Arc, Mutex}; use weaver_forge::registry::{ResolvedGroup, ResolvedRegistry}; use weaver_resolved_schema::attribute::Attribute; use weaver_semconv::attribute::{ @@ -174,8 +174,8 @@ mod tests { #[derive(Debug)] pub struct SpanExporter { - resource: Resource, - is_shutdown: atomic::AtomicBool, + // resource: Resource, + // is_shutdown: atomic::AtomicBool, spans: Arc>>, } @@ -184,22 +184,8 @@ mod tests { &mut self, batch: Vec, ) -> BoxFuture<'static, ExportResult> { - if self.is_shutdown.load(atomic::Ordering::SeqCst) { - Box::pin(std::future::ready(Err(TraceError::from( - "exporter is shut down", - )))) - } else { - self.spans.lock().unwrap().extend(batch); - Box::pin(std::future::ready(Ok(()))) - } - } - - fn shutdown(&mut self) { - self.is_shutdown.store(true, atomic::Ordering::SeqCst); - } - - fn set_resource(&mut self, res: &Resource) { - self.resource = res.clone(); + self.spans.lock().unwrap().extend(batch); + Box::pin(std::future::ready(Ok(()))) } } @@ -207,8 +193,6 @@ mod tests { fn test_emit_trace_for_registry() { let spans = Arc::new(Mutex::new(Vec::new())); let span_exporter = SpanExporter { - resource: Resource::empty(), - is_shutdown: atomic::AtomicBool::new(false), spans: spans.clone(), }; let tracer_provider = sdktrace::TracerProvider::builder() @@ -843,9 +827,7 @@ mod tests { }, ], }; - super::emit_trace_for_registry(®istry, "TEST"); - global::shutdown_tracer_provider(); // Now check the spans stored in the span exporter From 179597933437457fccec9c4a95296a9f43559868 Mon Sep 17 00:00:00 2001 From: jerbly Date: Fri, 7 Feb 2025 14:37:13 -0500 Subject: [PATCH 2/2] remove unused fields from SpanExporter struct --- crates/weaver_emit/src/spans.rs | 2 -- 1 file changed, 2 deletions(-) diff --git a/crates/weaver_emit/src/spans.rs b/crates/weaver_emit/src/spans.rs index 1a0cc359..bb41c40b 100644 --- a/crates/weaver_emit/src/spans.rs +++ b/crates/weaver_emit/src/spans.rs @@ -174,8 +174,6 @@ mod tests { #[derive(Debug)] pub struct SpanExporter { - // resource: Resource, - // is_shutdown: atomic::AtomicBool, spans: Arc>>, }