Trace 'records(TopicPartitions)` in kafka consumer

Author: mar-kolya

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

@@ -52,7 +52,9 @@ public Map<? extends ElementMatcher<? super MethodDescription>, 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(