Remove streams from B3Propagator

They are expensive in this context
open-telemetry · Mar 24, 2023 · fba3b67 · fba3b67
1 parent c769d53
commit fba3b67
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