Skip to content

Commit 2a05236

Browse files
authored
Fix mongo latest dep test (#14642)
1 parent 42c5fa1 commit 2a05236

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

instrumentation/mongo/mongo-4.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/mongo/v4_0/DefaultConnectionPoolTaskInstrumentation.java

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,10 @@ public void transform(TypeTransformer transformer) {
3434
transformer.applyAdviceToMethod(
3535
isConstructor().and(takesArgument(3, Consumer.class)),
3636
this.getClass().getName() + "$TaskArg3Advice");
37+
// since 5.6.0
38+
transformer.applyAdviceToMethod(
39+
isConstructor().and(takesArgument(4, Consumer.class)),
40+
this.getClass().getName() + "$TaskArg4Advice");
3741
}
3842

3943
@SuppressWarnings("unused")
@@ -55,4 +59,14 @@ public static void wrapCallback(
5559
action = new TaskWrapper(Java8BytecodeBridge.currentContext(), action);
5660
}
5761
}
62+
63+
@SuppressWarnings("unused")
64+
public static class TaskArg4Advice {
65+
66+
@Advice.OnMethodEnter(suppress = Throwable.class)
67+
public static void wrapCallback(
68+
@Advice.Argument(value = 4, readOnly = false) Consumer<Object> action) {
69+
action = new TaskWrapper(Java8BytecodeBridge.currentContext(), action);
70+
}
71+
}
5872
}

0 commit comments

Comments
 (0)