Adds BaggagePropagation benchmarks for decorate (#1425)

Signed-off-by: Adrian Cole <[email protected]>
openzipkin · Mar 13, 2024 · 69e30f3 · 69e30f3
1 parent b7ece7b
commit 69e30f3
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/instrumentation/benchmarks/pom.xml b/instrumentation/benchmarks/pom.xml
@@ -43,6 +43,13 @@
  <version>${jmh.version}</version>
  <scope>test</scope>
  </dependency>
+ <!-- IntelliJ doesn't know to use annotation processing on tests without this -->
+ <dependency>
+ <groupId>org.openjdk.jmh</groupId>
+ <artifactId>jmh-generator-annprocess</artifactId>
+ <version>${jmh.version}</version>
+ <scope>test</scope>
+ </dependency>
 
  <dependency>
  <groupId>com.amazonaws</groupId>

diff --git a/instrumentation/benchmarks/src/test/java/brave/baggage/BaggagePropagationBenchmarks.java b/instrumentation/benchmarks/src/test/java/brave/baggage/BaggagePropagationBenchmarks.java
@@ -14,6 +14,7 @@
 package brave.baggage;
 
 import brave.baggage.BaggagePropagationConfig.SingleBaggageField;
+import brave.internal.InternalPropagation;
 import brave.internal.codec.HexCodec;
 import brave.propagation.B3Propagation;
 import brave.propagation.Propagation;
@@ -65,6 +66,8 @@ public class BaggagePropagationBenchmarks {
  }
  };
 
+ static final TraceContext contextWithBaggage = extractor.extract(incoming).context();
+
  static final Map<String, String> incomingNoBaggage = new LinkedHashMap<String, String>() {
  {
  injector.inject(context, this);
@@ -73,6 +76,14 @@ public class BaggagePropagationBenchmarks {
 
  static final Map<String, String> nothingIncoming = Collections.emptyMap();
 
+ @Benchmark public TraceContext decorate() {
+ return factory.decorate(InternalPropagation.instance.shallowCopy(context));
+ }
+
+ @Benchmark public TraceContext decorate_withBaggage() {
+ return factory.decorate(InternalPropagation.instance.shallowCopy(contextWithBaggage));
+ }
+
  @Benchmark public void inject() {
  Map<String, String> request = new LinkedHashMap<>();
  injector.inject(context, request);