From 058c4ec567647de94b8f0657d1190f239a92f83d Mon Sep 17 00:00:00 2001 From: Nikolay Martynov Date: Mon, 8 Jul 2019 10:35:57 -0400 Subject: [PATCH] Trace 'records(TopicPartitions)` in kafka consumer --- .../kafka_clients/KafkaConsumerInstrumentation.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java b/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java index 9653a61b0b3..d79e8a084b6 100644 --- a/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java +++ b/dd-java-agent/instrumentation/kafka-clients-0.11/src/main/java/datadog/trace/instrumentation/kafka_clients/KafkaConsumerInstrumentation.java @@ -52,7 +52,9 @@ public Map, String> transfor isMethod() .and(isPublic()) .and(named("records")) - .and(takesArgument(0, String.class)) + .and( + takesArgument(0, String.class) + .or(takesArgument(0, named("org.apache.kafka.common.TopicPartition")))) .and(returns(Iterable.class)), IterableAdvice.class.getName()); transformers.put(