remove ExecutionContextWrapper

Author: pjfanning

instrumentation/pekko/pekko-actor-1.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/pekkoactor/v1_0/ExecutionContextWrapper.java

@@ -15,7 +15,6 @@
 import net.bytebuddy.asm.Advice;
 import net.bytebuddy.description.type.TypeDescription;
 import net.bytebuddy.matcher.ElementMatcher;
-import scala.concurrent.ExecutionContext;
 
 public class PekkoScheduleInstrumentation implements TypeInstrumentation {
 
@@ -40,9 +39,9 @@ public static class ScheduleAdvice {
 
     @Advice.OnMethodEnter(suppress = Throwable.class)
     public static void enterSchedule(
-        @Advice.Argument(value = 3, readOnly = false) ExecutionContext executionContext) {
+        @Advice.Argument(value = 2, readOnly = false) Runnable runnable) {
       Context context = Java8BytecodeBridge.currentContext();
-      executionContext = new ExecutionContextWrapper(executionContext, context);
+      runnable = context.wrap(runnable);
     }
   }
 }

instrumentation/pekko/pekko-actor-1.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/pekkoactor/v1_0/PekkoScheduleInstrumentation.java

@@ -15,7 +15,6 @@
 import net.bytebuddy.asm.Advice;
 import net.bytebuddy.description.type.TypeDescription;
 import net.bytebuddy.matcher.ElementMatcher;
-import scala.concurrent.ExecutionContext;
 
 public class PekkoScheduleOnceInstrumentation implements TypeInstrumentation {
 
@@ -39,9 +38,9 @@ public static class ScheduleOnceAdvice {
 
     @Advice.OnMethodEnter(suppress = Throwable.class)
     public static void enterScheduleOnce(
-        @Advice.Argument(value = 2, readOnly = false) ExecutionContext executionContext) {
+        @Advice.Argument(value = 1, readOnly = false) Runnable runnable) {
       Context context = Java8BytecodeBridge.currentContext();
-      executionContext = new ExecutionContextWrapper(executionContext, context);
+      runnable = context.wrap(runnable);
     }
   }
 }