Fix XrayPropagator when no header is present (#867)

open-telemetry · Aug 27, 2022 · 043e4b7 · 043e4b7
1 parent c92a1ad
commit 043e4b7
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/opentelemetry-aws/src/lib.rs b/opentelemetry-aws/src/lib.rs
@@ -203,11 +203,9 @@ pub mod trace {
         }
 
         fn extract_with_context(&self, cx: &Context, extractor: &dyn Extractor) -> Context {
-            let extracted = self
-                .extract_span_context(extractor)
-                .unwrap_or_else(|_| SpanContext::empty_context());
-
-            cx.with_remote_span_context(extracted)
+            self.extract_span_context(extractor)
+                .map(|sc| cx.with_remote_span_context(sc))
+                .unwrap_or_else(|_| cx.clone())
         }
 
         fn fields(&self) -> FieldIter<'_> {