Remove streams from B3Propagator (#5326)

They are expensive in this context
open-telemetry · Mar 29, 2023 · aad95f8 · aad95f8
1 parent ba06aa7
commit aad95f8
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/...-propagators/src/main/java/io/opentelemetry/extension/trace/propagation/B3Propagator.java b/...-propagators/src/main/java/io/opentelemetry/extension/trace/propagation/B3Propagator.java
@@ -12,8 +12,6 @@
 import io.opentelemetry.context.propagation.TextMapSetter;
 import java.util.Collection;
 import java.util.Optional;
-import java.util.function.Supplier;
-import java.util.stream.Stream;
 import javax.annotation.Nullable;
 import javax.annotation.concurrent.Immutable;
 
@@ -113,15 +111,11 @@ public <C> void inject(Context context, @Nullable C carrier, TextMapSetter<C> se
 
   @Override
   public <C> Context extract(Context context, @Nullable C carrier, TextMapGetter<C> getter) {
-    return Stream.<Supplier<Optional<Context>>>of(
-            () -> singleHeaderExtractor.extract(context, carrier, getter),
-            () -> multipleHeadersExtractor.extract(context, carrier, getter),
-            () -> Optional.of(context))
-        .map(Supplier::get)
-        .filter(Optional::isPresent)
-        .map(Optional::get)
-        .findFirst()
-        .get();
+    Optional<Context> ctx = singleHeaderExtractor.extract(context, carrier, getter);
+    if (ctx.isPresent()) {
+      return ctx.get();
+    }
+    return multipleHeadersExtractor.extract(context, carrier, getter).orElse(context);
   }
 
   @Override