Muzzle Kafka Client & Streams Instrumentation

Add muzzle blocks and remove class loader matcher methods.
This commit is contained in:
Gary Huang 2018-09-07 20:11:01 -04:00
parent 0f3e119ae5
commit ea0600c0c3
No known key found for this signature in database
GPG Key ID: 225B4CE0235DF2D1
6 changed files with 13 additions and 44 deletions

View File

@ -1,13 +1,10 @@
apply plugin: 'version-scan'
versionScan {
muzzle {
pass {
group = "org.apache.kafka"
module = "kafka-clients"
versions = "[0.11.0.0,)"
verifyPresent = [
'org.apache.kafka.common.header.Header' : null,
'org.apache.kafka.common.header.Headers': null,
]
assertInverse = true
}
}
apply from: "${rootDir}/gradle/java.gradle"

View File

@ -1,6 +1,5 @@
package datadog.trace.instrumentation.kafka_clients;
import static datadog.trace.agent.tooling.ClassLoaderMatcher.classLoaderHasClasses;
import static net.bytebuddy.matcher.ElementMatchers.isMethod;
import static net.bytebuddy.matcher.ElementMatchers.isPublic;
import static net.bytebuddy.matcher.ElementMatchers.named;
@ -45,12 +44,6 @@ public final class KafkaConsumerInstrumentation extends Instrumenter.Default {
return named("org.apache.kafka.clients.consumer.ConsumerRecords");
}
@Override
public ElementMatcher<ClassLoader> classLoaderMatcher() {
return classLoaderHasClasses(
"org.apache.kafka.common.header.Header", "org.apache.kafka.common.header.Headers");
}
@Override
public String[] helperClassNames() {
return HELPER_CLASS_NAMES;

View File

@ -1,6 +1,5 @@
package datadog.trace.instrumentation.kafka_clients;
import static datadog.trace.agent.tooling.ClassLoaderMatcher.classLoaderHasClasses;
import static io.opentracing.log.Fields.ERROR_OBJECT;
import static net.bytebuddy.matcher.ElementMatchers.isMethod;
import static net.bytebuddy.matcher.ElementMatchers.isPublic;
@ -46,12 +45,6 @@ public final class KafkaProducerInstrumentation extends Instrumenter.Default {
return named("org.apache.kafka.clients.producer.KafkaProducer");
}
@Override
public ElementMatcher<ClassLoader> classLoaderMatcher() {
return classLoaderHasClasses(
"org.apache.kafka.common.header.Header", "org.apache.kafka.common.header.Headers");
}
@Override
public String[] helperClassNames() {
return HELPER_CLASS_NAMES;

View File

@ -1,12 +1,9 @@
apply plugin: 'version-scan'
versionScan {
muzzle {
pass {
group = "org.apache.kafka"
module = "kafka-streams"
versions = "[0.11.0.0,)"
verifyPresent = [
'org.apache.kafka.streams.state.internals.OrderedBytes': null
]
}
}
apply from: "${rootDir}/gradle/java.gradle"

View File

@ -46,11 +46,6 @@ public class KafkaStreamsProcessorInstrumentation {
return named("org.apache.kafka.streams.processor.internals.PartitionGroup");
}
@Override
public ElementMatcher<ClassLoader> classLoaderMatcher() {
return classLoaderHasClasses("org.apache.kafka.streams.state.internals.OrderedBytes");
}
@Override
public String[] helperClassNames() {
return HELPER_CLASS_NAMES;

View File

@ -1,6 +1,5 @@
package datadog.trace.instrumentation.kafka_streams;
import static datadog.trace.agent.tooling.ClassLoaderMatcher.classLoaderHasClasses;
import static net.bytebuddy.matcher.ElementMatchers.isMethod;
import static net.bytebuddy.matcher.ElementMatchers.isPublic;
import static net.bytebuddy.matcher.ElementMatchers.named;
@ -30,11 +29,6 @@ public class KafkaStreamsSourceNodeRecordDeserializerInstrumentation extends Ins
return named("org.apache.kafka.streams.processor.internals.SourceNodeRecordDeserializer");
}
@Override
public ElementMatcher<ClassLoader> classLoaderMatcher() {
return classLoaderHasClasses("org.apache.kafka.streams.state.internals.OrderedBytes");
}
@Override
public Map<ElementMatcher, String> transformers() {
final Map<ElementMatcher, String> transformers = new HashMap<>();