open-telemetry
/
opentelemetry-java-instrumentation
mirror of https://github.com/open-telemetry/opentelemetry-java-instrumentation.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Kafka instrumentation: make return type more general in instrumentation matching

This commit is contained in:
Nikolay Martynov 2019-07-10 13:47:20 -04:00
parent 2e62ca8497
commit f484295460
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java
View File

@ -1,5 +1,6 @@
package datadog.trace.instrumentation.kafka_clients; package datadog.trace.instrumentation.kafka_clients;
import static datadog.trace.agent.tooling.ByteBuddyElementMatchers.safeHasSuperType;
import static datadog.trace.instrumentation.kafka_clients.KafkaDecorator.CONSUMER_DECORATE; import static datadog.trace.instrumentation.kafka_clients.KafkaDecorator.CONSUMER_DECORATE;
import static net.bytebuddy.matcher.ElementMatchers.isMethod; import static net.bytebuddy.matcher.ElementMatchers.isMethod;
import static net.bytebuddy.matcher.ElementMatchers.isPublic; import static net.bytebuddy.matcher.ElementMatchers.isPublic;
@ -55,7 +56,7 @@ public final class KafkaConsumerInstrumentation extends Instrumenter.Default {
.and( .and(
takesArgument(0, String.class) takesArgument(0, String.class)
.or(takesArgument(0, named("org.apache.kafka.common.TopicPartition")))) .or(takesArgument(0, named("org.apache.kafka.common.TopicPartition"))))
.and(returns(Iterable.class)), .and(returns(safeHasSuperType(named(Iterable.class.getName())))),
IterableAdvice.class.getName()); IterableAdvice.class.getName());
transformers.put( transformers.put(
isMethod() isMethod()