Update AbstractSparkInstrumentation.java

Author: aboitreaud

dd-java-agent/instrumentation/spark/src/main/java/datadog/trace/instrumentation/spark/AbstractSparkInstrumentation.java

@@ -5,6 +5,7 @@
 import static net.bytebuddy.matcher.ElementMatchers.isMethod;
 import static net.bytebuddy.matcher.ElementMatchers.nameEndsWith;
 import static net.bytebuddy.matcher.ElementMatchers.takesArgument;
+import static net.bytebuddy.matcher.ElementMatchers.takesArguments;
 
 import datadog.trace.agent.tooling.Instrumenter;
 import datadog.trace.agent.tooling.InstrumenterModule;
@@ -68,8 +69,8 @@ public void methodAdvice(MethodTransformer transformer) {
     // LiveListenerBus class is used to manage spark listeners
     transformer.applyAdvice(
         isMethod()
-            .and(named("addToSharedQueue"))
-            .and(takesArgument(0, named("org.apache.spark.scheduler.SparkListenerInterface")))
+            .and(named("addToSharedQueue").or(named("addToQueue")))
+            .and(takesArguments(1))
             .and(isDeclaredBy(named("org.apache.spark.scheduler.LiveListenerBus"))),
         AbstractSparkInstrumentation.class.getName() + "$LiveListenerBusAdvice");
   }