From 043e4b7523f66e79338ada84e7ab2da53251d448 Mon Sep 17 00:00:00 2001 From: Hargo Khalsa Date: Fri, 26 Aug 2022 18:01:25 -0600 Subject: [PATCH] Fix XrayPropagator when no header is present (#867) --- opentelemetry-aws/src/lib.rs | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/opentelemetry-aws/src/lib.rs b/opentelemetry-aws/src/lib.rs index 89a6d2d2d1..61dfc45f31 100644 --- 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<'_> {