Move Akka and Armeria virtual fields to helpers (#13604)

JonasKunz · laurit · web-flow · commit 18155581b216 · 2025-04-01T15:35:08.000+03:00
Co-authored-by: Lauri Tulmin &lt;ltulmin@splunk.com&gt;
diff --git a/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaActorCellInstrumentation.java b/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaActorCellInstrumentation.java
@@ -11,8 +11,6 @@
 import akka.dispatch.Envelope;
 import akka.dispatch.sysmsg.SystemMessage;
 import io.opentelemetry.context.Scope;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
-import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
 import io.opentelemetry.javaagent.bootstrap.executors.TaskAdviceHelper;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
@@ -42,9 +40,8 @@ public static class InvokeAdvice {
 
     @Advice.OnMethodEnter(suppress = Throwable.class)
     public static Scope enter(@Advice.Argument(0) Envelope envelope) {
-      VirtualField<Envelope, PropagatedContext> virtualField =
-          VirtualField.find(Envelope.class, PropagatedContext.class);
-      return TaskAdviceHelper.makePropagatedContextCurrent(virtualField, envelope);
+      return TaskAdviceHelper.makePropagatedContextCurrent(
+          VirtualFields.ENVELOPE_PROPAGATED_CONTEXT, envelope);
     }
 
     @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class)
@@ -60,9 +57,8 @@ public static class SystemInvokeAdvice {
 
     @Advice.OnMethodEnter(suppress = Throwable.class)
     public static Scope enter(@Advice.Argument(0) SystemMessage systemMessage) {
-      VirtualField<SystemMessage, PropagatedContext> virtualField =
-          VirtualField.find(SystemMessage.class, PropagatedContext.class);
-      return TaskAdviceHelper.makePropagatedContextCurrent(virtualField, systemMessage);
+      return TaskAdviceHelper.makePropagatedContextCurrent(
+          VirtualFields.SYSTEM_MESSAGE_PROPAGATED_CONTEXT, systemMessage);
     }
 
     @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class)
diff --git a/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDefaultSystemMessageQueueInstrumentation.java b/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDefaultSystemMessageQueueInstrumentation.java
@@ -12,7 +12,6 @@
 
 import akka.dispatch.sysmsg.SystemMessage;
 import io.opentelemetry.context.Context;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
 import io.opentelemetry.javaagent.bootstrap.Java8BytecodeBridge;
 import io.opentelemetry.javaagent.bootstrap.executors.ExecutorAdviceHelper;
 import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
@@ -49,9 +48,8 @@ public static class DispatchSystemAdvice {
     public static PropagatedContext enter(@Advice.Argument(1) SystemMessage systemMessage) {
       Context context = Java8BytecodeBridge.currentContext();
       if (ExecutorAdviceHelper.shouldPropagateContext(context, systemMessage)) {
-        VirtualField<SystemMessage, PropagatedContext> virtualField =
-            VirtualField.find(SystemMessage.class, PropagatedContext.class);
-        return ExecutorAdviceHelper.attachContextToTask(context, virtualField, systemMessage);
+        return ExecutorAdviceHelper.attachContextToTask(
+            context, VirtualFields.SYSTEM_MESSAGE_PROPAGATED_CONTEXT, systemMessage);
       }
       return null;
     }
@@ -61,10 +59,11 @@ public static void exit(
         @Advice.Argument(1) SystemMessage systemMessage,
         @Advice.Enter PropagatedContext propagatedContext,
         @Advice.Thrown Throwable throwable) {
-      VirtualField<SystemMessage, PropagatedContext> virtualField =
-          VirtualField.find(SystemMessage.class, PropagatedContext.class);
       ExecutorAdviceHelper.cleanUpAfterSubmit(
-          propagatedContext, throwable, virtualField, systemMessage);
+          propagatedContext,
+          throwable,
+          VirtualFields.SYSTEM_MESSAGE_PROPAGATED_CONTEXT,
+          systemMessage);
     }
   }
 }
diff --git a/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDispatcherInstrumentation.java b/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaDispatcherInstrumentation.java
@@ -10,7 +10,6 @@
 
 import akka.dispatch.Envelope;
 import io.opentelemetry.context.Context;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
 import io.opentelemetry.javaagent.bootstrap.Java8BytecodeBridge;
 import io.opentelemetry.javaagent.bootstrap.executors.ExecutorAdviceHelper;
 import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
@@ -43,9 +42,8 @@ public static class DispatchEnvelopeAdvice {
     public static PropagatedContext enterDispatch(@Advice.Argument(1) Envelope envelope) {
       Context context = Java8BytecodeBridge.currentContext();
       if (ExecutorAdviceHelper.shouldPropagateContext(context, envelope.message())) {
-        VirtualField<Envelope, PropagatedContext> virtualField =
-            VirtualField.find(Envelope.class, PropagatedContext.class);
-        return ExecutorAdviceHelper.attachContextToTask(context, virtualField, envelope);
+        return ExecutorAdviceHelper.attachContextToTask(
+            context, VirtualFields.ENVELOPE_PROPAGATED_CONTEXT, envelope);
       }
       return null;
     }
@@ -55,9 +53,8 @@ public static void exitDispatch(
         @Advice.Argument(1) Envelope envelope,
         @Advice.Enter PropagatedContext propagatedContext,
         @Advice.Thrown Throwable throwable) {
-      VirtualField<Envelope, PropagatedContext> virtualField =
-          VirtualField.find(Envelope.class, PropagatedContext.class);
-      ExecutorAdviceHelper.cleanUpAfterSubmit(propagatedContext, throwable, virtualField, envelope);
+      ExecutorAdviceHelper.cleanUpAfterSubmit(
+          propagatedContext, throwable, VirtualFields.ENVELOPE_PROPAGATED_CONTEXT, envelope);
     }
   }
 }
diff --git a/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/VirtualFields.java b/instrumentation/akka/akka-actor-2.3/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/VirtualFields.java
@@ -0,0 +1,22 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.akkaactor;
+
+import akka.dispatch.Envelope;
+import akka.dispatch.sysmsg.SystemMessage;
+import io.opentelemetry.instrumentation.api.util.VirtualField;
+import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
+
+public class VirtualFields {
+
+  private VirtualFields() {}
+
+  public static final VirtualField<Envelope, PropagatedContext> ENVELOPE_PROPAGATED_CONTEXT =
+      VirtualField.find(Envelope.class, PropagatedContext.class);
+  public static final VirtualField<SystemMessage, PropagatedContext>
+      SYSTEM_MESSAGE_PROPAGATED_CONTEXT =
+          VirtualField.find(SystemMessage.class, PropagatedContext.class);
+}
diff --git a/instrumentation/akka/akka-actor-fork-join-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaForkJoinPoolInstrumentation.java b/instrumentation/akka/akka-actor-fork-join-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaForkJoinPoolInstrumentation.java
@@ -10,7 +10,6 @@
 
 import akka.dispatch.forkjoin.ForkJoinTask;
 import io.opentelemetry.context.Context;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
 import io.opentelemetry.javaagent.bootstrap.Java8BytecodeBridge;
 import io.opentelemetry.javaagent.bootstrap.executors.ExecutorAdviceHelper;
 import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
@@ -51,9 +50,8 @@ public static class SetAkkaForkJoinStateAdvice {
     public static PropagatedContext enterJobSubmit(@Advice.Argument(0) ForkJoinTask<?> task) {
       Context context = Java8BytecodeBridge.currentContext();
       if (ExecutorAdviceHelper.shouldPropagateContext(context, task)) {
-        VirtualField<ForkJoinTask<?>, PropagatedContext> virtualField =
-            VirtualField.find(ForkJoinTask.class, PropagatedContext.class);
-        return ExecutorAdviceHelper.attachContextToTask(context, virtualField, task);
+        return ExecutorAdviceHelper.attachContextToTask(
+            context, VirtualFields.FORK_JOIN_TASK_PROPAGATED_CONTEXT, task);
       }
       return null;
     }
@@ -63,9 +61,8 @@ public static void exitJobSubmit(
         @Advice.Argument(0) ForkJoinTask<?> task,
         @Advice.Enter PropagatedContext propagatedContext,
         @Advice.Thrown Throwable throwable) {
-      VirtualField<ForkJoinTask<?>, PropagatedContext> virtualField =
-          VirtualField.find(ForkJoinTask.class, PropagatedContext.class);
-      ExecutorAdviceHelper.cleanUpAfterSubmit(propagatedContext, throwable, virtualField, task);
+      ExecutorAdviceHelper.cleanUpAfterSubmit(
+          propagatedContext, throwable, VirtualFields.FORK_JOIN_TASK_PROPAGATED_CONTEXT, task);
     }
   }
 }
diff --git a/instrumentation/akka/akka-actor-fork-join-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaForkJoinTaskInstrumentation.java b/instrumentation/akka/akka-actor-fork-join-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/AkkaForkJoinTaskInstrumentation.java
@@ -15,8 +15,6 @@
 import akka.dispatch.forkjoin.ForkJoinPool;
 import akka.dispatch.forkjoin.ForkJoinTask;
 import io.opentelemetry.context.Scope;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
-import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
 import io.opentelemetry.javaagent.bootstrap.executors.TaskAdviceHelper;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
@@ -62,14 +60,13 @@ public static class ForkJoinTaskAdvice {
      */
     @Advice.OnMethodEnter(suppress = Throwable.class)
     public static Scope enter(@Advice.This ForkJoinTask<?> thiz) {
-      VirtualField<ForkJoinTask<?>, PropagatedContext> virtualField =
-          VirtualField.find(ForkJoinTask.class, PropagatedContext.class);
-      Scope scope = TaskAdviceHelper.makePropagatedContextCurrent(virtualField, thiz);
+      Scope scope =
+          TaskAdviceHelper.makePropagatedContextCurrent(
+              VirtualFields.FORK_JOIN_TASK_PROPAGATED_CONTEXT, thiz);
       if (thiz instanceof Runnable) {
-        VirtualField<Runnable, PropagatedContext> runnableVirtualField =
-            VirtualField.find(Runnable.class, PropagatedContext.class);
         Scope newScope =
-            TaskAdviceHelper.makePropagatedContextCurrent(runnableVirtualField, (Runnable) thiz);
+            TaskAdviceHelper.makePropagatedContextCurrent(
+                VirtualFields.RUNNABLE_PROPAGATED_CONTEXT, (Runnable) thiz);
         if (null != newScope) {
           if (null != scope) {
             newScope.close();
@@ -79,10 +76,9 @@ public static Scope enter(@Advice.This ForkJoinTask<?> thiz) {
         }
       }
       if (thiz instanceof Callable) {
-        VirtualField<Callable<?>, PropagatedContext> callableVirtualField =
-            VirtualField.find(Callable.class, PropagatedContext.class);
         Scope newScope =
-            TaskAdviceHelper.makePropagatedContextCurrent(callableVirtualField, (Callable<?>) thiz);
+            TaskAdviceHelper.makePropagatedContextCurrent(
+                VirtualFields.CALLABLE_PROPAGATED_CONTEXT, (Callable<?>) thiz);
         if (null != newScope) {
           if (null != scope) {
             newScope.close();
diff --git a/instrumentation/akka/akka-actor-fork-join-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/VirtualFields.java b/instrumentation/akka/akka-actor-fork-join-2.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkaactor/VirtualFields.java
@@ -0,0 +1,24 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.akkaactor;
+
+import akka.dispatch.forkjoin.ForkJoinTask;
+import io.opentelemetry.instrumentation.api.util.VirtualField;
+import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;
+import java.util.concurrent.Callable;
+
+public class VirtualFields {
+
+  private VirtualFields() {}
+
+  public static final VirtualField<ForkJoinTask<?>, PropagatedContext>
+      FORK_JOIN_TASK_PROPAGATED_CONTEXT =
+          VirtualField.find(ForkJoinTask.class, PropagatedContext.class);
+  public static final VirtualField<Runnable, PropagatedContext> RUNNABLE_PROPAGATED_CONTEXT =
+      VirtualField.find(Runnable.class, PropagatedContext.class);
+  public static final VirtualField<Callable<?>, PropagatedContext> CALLABLE_PROPAGATED_CONTEXT =
+      VirtualField.find(Callable.class, PropagatedContext.class);
+}
diff --git a/instrumentation/akka/akka-http-10.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkahttp/server/route/PathMatcherInstrumentation.java b/instrumentation/akka/akka-http-10.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkahttp/server/route/PathMatcherInstrumentation.java
@@ -11,7 +11,6 @@
 
 import akka.http.scaladsl.model.Uri;
 import akka.http.scaladsl.server.PathMatcher;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
 import net.bytebuddy.asm.Advice;
@@ -41,7 +40,7 @@ public static void onEnter(
         @Advice.Argument(0) Uri.Path prefix, @Advice.Return PathMatcher<?> result) {
       // store the path being matched inside a VirtualField on the given matcher, so it can be used
       // for constructing the route
-      VirtualField.find(PathMatcher.class, String.class).set(result, prefix.toString());
+      PathMatcherUtil.setMatched(result, prefix.toString());
     }
   }
 }
diff --git a/instrumentation/akka/akka-http-10.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkahttp/server/route/PathMatcherStaticInstrumentation.java b/instrumentation/akka/akka-http-10.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkahttp/server/route/PathMatcherStaticInstrumentation.java
@@ -12,7 +12,6 @@
 import akka.http.scaladsl.model.Uri;
 import akka.http.scaladsl.server.PathMatcher;
 import akka.http.scaladsl.server.PathMatchers;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
 import net.bytebuddy.asm.Advice;
@@ -48,7 +47,7 @@ public static void onExit(
         }
         // if present use the matched path that was remembered in PathMatcherInstrumentation,
         // otherwise just use a *
-        String prefix = VirtualField.find(PathMatcher.class, String.class).get(pathMatcher);
+        String prefix = PathMatcherUtil.getMatched(pathMatcher);
         if (prefix == null) {
           if (PathMatchers.Slash$.class == pathMatcher.getClass()) {
             prefix = "/";
diff --git a/instrumentation/akka/akka-http-10.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkahttp/server/route/PathMatcherUtil.java b/instrumentation/akka/akka-http-10.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/akkahttp/server/route/PathMatcherUtil.java
@@ -0,0 +1,25 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.javaagent.instrumentation.akkahttp.server.route;
+
+import akka.http.scaladsl.server.PathMatcher;
+import io.opentelemetry.instrumentation.api.util.VirtualField;
+
+public class PathMatcherUtil {
+
+  private static final VirtualField<PathMatcher<?>, String> PATH_MATCHER_ROUTE =
+      VirtualField.find(PathMatcher.class, String.class);
+
+  public static void setMatched(PathMatcher<?> matcher, String route) {
+    PATH_MATCHER_ROUTE.set(matcher, route);
+  }
+
+  public static String getMatched(PathMatcher<?> matcher) {
+    return PATH_MATCHER_ROUTE.get(matcher);
+  }
+
+  private PathMatcherUtil() {}
+}
diff --git a/instrumentation/armeria/armeria-grpc-1.14/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/armeria/grpc/v1_14/ArmeriaServerCallInstrumentation.java b/instrumentation/armeria/armeria-grpc-1.14/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/armeria/grpc/v1_14/ArmeriaServerCallInstrumentation.java
@@ -10,7 +10,7 @@
 
 import com.linecorp.armeria.server.ServiceRequestContext;
 import io.grpc.ServerCall;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
+import io.opentelemetry.instrumentation.grpc.v1_6.GrpcAuthorityStorage;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;
 import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
 import net.bytebuddy.asm.Advice;
@@ -43,7 +43,7 @@ public static void onExit(
         // ArmeriaServerCall does not implement getAuthority. We will store the value for authority
         // header as virtual field, this field is read in grpc instrumentation in
         // TracingServerInterceptor
-        VirtualField.find(ServerCall.class, String.class).set(serverCall, authority);
+        GrpcAuthorityStorage.setAuthority(serverCall, authority);
       }
     }
   }
diff --git a/instrumentation/grpc-1.6/library/src/main/java/io/opentelemetry/instrumentation/grpc/v1_6/GrpcAuthorityStorage.java b/instrumentation/grpc-1.6/library/src/main/java/io/opentelemetry/instrumentation/grpc/v1_6/GrpcAuthorityStorage.java
@@ -0,0 +1,29 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+package io.opentelemetry.instrumentation.grpc.v1_6;
+
+import io.grpc.ServerCall;
+import io.opentelemetry.instrumentation.api.util.VirtualField;
+
+/**
+ * In case a {@link ServerCall} implementation does not implement {@link ServerCall#getAuthority()}
+ * like armeria, this utility class should be used to provide the authority instead
+ */
+public class GrpcAuthorityStorage {
+
+  private static final VirtualField<ServerCall<?, ?>, String> AUTHORITY_FIELD =
+      VirtualField.find(ServerCall.class, String.class);
+
+  private GrpcAuthorityStorage() {}
+
+  public static void setAuthority(ServerCall<?, ?> call, String authority) {
+    AUTHORITY_FIELD.set(call, authority);
+  }
+
+  static String getAuthority(ServerCall<?, ?> call) {
+    return AUTHORITY_FIELD.get(call);
+  }
+}
diff --git a/instrumentation/grpc-1.6/library/src/main/java/io/opentelemetry/instrumentation/grpc/v1_6/TracingServerInterceptor.java b/instrumentation/grpc-1.6/library/src/main/java/io/opentelemetry/instrumentation/grpc/v1_6/TracingServerInterceptor.java
@@ -20,7 +20,6 @@
 import io.opentelemetry.context.Context;
 import io.opentelemetry.context.Scope;
 import io.opentelemetry.instrumentation.api.instrumenter.Instrumenter;
-import io.opentelemetry.instrumentation.api.util.VirtualField;
 import java.util.concurrent.atomic.AtomicLongFieldUpdater;
 
 final class TracingServerInterceptor implements ServerInterceptor {
@@ -46,9 +45,6 @@ final class TracingServerInterceptor implements ServerInterceptor {
   private static final AtomicLongFieldUpdater<TracingServerCall> RECEIVED_MESSAGE_ID_UPDATER =
       AtomicLongFieldUpdater.newUpdater(TracingServerCall.class, "receivedMessageId");
 
-  private static final VirtualField<ServerCall<?, ?>, String> AUTHORITY_FIELD =
-      VirtualField.find(ServerCall.class, String.class);
-
   private final Instrumenter<GrpcRequest, Status> instrumenter;
   private final boolean captureExperimentalSpanAttributes;
   private final boolean emitMessageEvents;
@@ -72,7 +68,7 @@ public <REQUEST, RESPONSE> ServerCall.Listener<REQUEST> interceptCall(
       // Armeria grpc server call does not implement getAuthority(). In
       // ArmeriaServerCallInstrumentation we store the value for the authority header in a virtual
       // field.
-      authority = AUTHORITY_FIELD.get(call);
+      authority = GrpcAuthorityStorage.getAuthority(call);
     }
     GrpcRequest request =
         new GrpcRequest(

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,6 @@`
`10`	`10`
`11`	`11`	`import akka.dispatch.Envelope;`
`12`	`12`	`import io.opentelemetry.context.Context;`
`13`		`-import io.opentelemetry.instrumentation.api.util.VirtualField;`
`14`	`13`	`import io.opentelemetry.javaagent.bootstrap.Java8BytecodeBridge;`
`15`	`14`	`import io.opentelemetry.javaagent.bootstrap.executors.ExecutorAdviceHelper;`
`16`	`15`	`import io.opentelemetry.javaagent.bootstrap.executors.PropagatedContext;`
`@@ -43,9 +42,8 @@ public static class DispatchEnvelopeAdvice {`
`43`	`42`	`public static PropagatedContext enterDispatch(@Advice.Argument(1) Envelope envelope) {`
`44`	`43`	`Context context = Java8BytecodeBridge.currentContext();`
`45`	`44`	`if (ExecutorAdviceHelper.shouldPropagateContext(context, envelope.message())) {`
`46`		`- VirtualField<Envelope, PropagatedContext> virtualField =`
`47`		`- VirtualField.find(Envelope.class, PropagatedContext.class);`
`48`		`- return ExecutorAdviceHelper.attachContextToTask(context, virtualField, envelope);`
	`45`	`+ return ExecutorAdviceHelper.attachContextToTask(`
	`46`	`+ context, VirtualFields.ENVELOPE_PROPAGATED_CONTEXT, envelope);`
`49`	`47`	`}`
`50`	`48`	`return null;`
`51`	`49`	`}`
`@@ -55,9 +53,8 @@ public static void exitDispatch(`
`55`	`53`	`@Advice.Argument(1) Envelope envelope,`
`56`	`54`	`@Advice.Enter PropagatedContext propagatedContext,`
`57`	`55`	`@Advice.Thrown Throwable throwable) {`
`58`		`- VirtualField<Envelope, PropagatedContext> virtualField =`
`59`		`- VirtualField.find(Envelope.class, PropagatedContext.class);`
`60`		`- ExecutorAdviceHelper.cleanUpAfterSubmit(propagatedContext, throwable, virtualField, envelope);`
	`56`	`+ ExecutorAdviceHelper.cleanUpAfterSubmit(`
	`57`	`+ propagatedContext, throwable, VirtualFields.ENVELOPE_PROPAGATED_CONTEXT, envelope);`
`61`	`58`	`}`
`62`	`59`	`}`
`63`	`60`	`}`
Original file line number	Diff line number	Diff line change
`@@ -11,7 +11,6 @@`
`11`	`11`
`12`	`12`	`import akka.http.scaladsl.model.Uri;`
`13`	`13`	`import akka.http.scaladsl.server.PathMatcher;`
`14`		`-import io.opentelemetry.instrumentation.api.util.VirtualField;`
`15`	`14`	`import io.opentelemetry.javaagent.extension.instrumentation.TypeInstrumentation;`
`16`	`15`	`import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;`
`17`	`16`	`import net.bytebuddy.asm.Advice;`
`@@ -41,7 +40,7 @@ public static void onEnter(`
`41`	`40`	`@Advice.Argument(0) Uri.Path prefix, @Advice.Return PathMatcher<?> result) {`
`42`	`41`	`// store the path being matched inside a VirtualField on the given matcher, so it can be used`
`43`	`42`	`// for constructing the route`
`44`		`- VirtualField.find(PathMatcher.class, String.class).set(result, prefix.toString());`
	`43`	`+ PathMatcherUtil.setMatched(result, prefix.toString());`
`45`	`44`	`}`
`46`	`45`	`}`
`47`	`46`	`}`