Fix concurrency with callback test on okhttp3 latest (#3676)

open-telemetry · Jul 26, 2021 · b483d7d · b483d7d
1 parent 6a2ee32
commit b483d7d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...opentelemetry/javaagent/instrumentation/okhttp/v3_0/OkHttp3DispatcherInstrumentation.java b/...opentelemetry/javaagent/instrumentation/okhttp/v3_0/OkHttp3DispatcherInstrumentation.java
@@ -5,6 +5,7 @@
 
 package io.opentelemetry.javaagent.instrumentation.okhttp.v3_0;
 
+import static io.opentelemetry.javaagent.extension.matcher.AgentElementMatchers.implementsInterface;
 import static net.bytebuddy.matcher.ElementMatchers.named;
 import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
 
@@ -28,7 +29,9 @@ public ElementMatcher<TypeDescription> typeMatcher() {
   @Override
   public void transform(TypeTransformer transformer) {
     transformer.applyAdviceToMethod(
-        named("enqueue").and(takesArgument(0, named("okhttp3.RealCall$AsyncCall"))),
+        named("enqueue")
+            .or(named("enqueue$okhttp"))
+            .and(takesArgument(0, implementsInterface(named(Runnable.class.getName())))),
         OkHttp3DispatcherInstrumentation.class.getName() + "$AttachStateAdvice");
   }