Skip to content

Commit b6d9c13

Browse files
author
Andrew Kent
committed
Make cassandra matching stricter
1 parent 3ea7ad7 commit b6d9c13

File tree

1 file changed

+18
-1
lines changed

1 file changed

+18
-1
lines changed

dd-java-agent/integrations/datastax-cassandra-3.2/src/main/java/dd/inst/datastax/cassandra/CassandraClientInstrumentation.java

Lines changed: 18 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
package dd.inst.datastax.cassandra;
22

3+
import static dd.trace.ClassLoaderMatcher.classLoaderHasClasses;
34
import static dd.trace.ExceptionHandlers.defaultExceptionHandler;
45
import static net.bytebuddy.matcher.ElementMatchers.*;
56

@@ -17,7 +18,23 @@ public class CassandraClientInstrumentation implements Instrumenter {
1718
@Override
1819
public AgentBuilder instrument(AgentBuilder agentBuilder) {
1920
return agentBuilder
20-
.type(named("com.datastax.driver.core.Cluster$Manager"))
21+
.type(
22+
named("com.datastax.driver.core.Cluster$Manager"),
23+
classLoaderHasClasses(
24+
"com.datastax.driver.core.BoundStatement",
25+
"com.datastax.driver.core.BoundStatement",
26+
"com.datastax.driver.core.CloseFuture",
27+
"com.datastax.driver.core.Cluster",
28+
"com.datastax.driver.core.Host",
29+
"com.datastax.driver.core.PreparedStatement",
30+
"com.datastax.driver.core.RegularStatement",
31+
"com.datastax.driver.core.ResultSet",
32+
"com.datastax.driver.core.ResultSetFuture",
33+
"com.datastax.driver.core.Session",
34+
"com.datastax.driver.core.Statement",
35+
"com.google.common.base.Function",
36+
"com.google.common.util.concurrent.Futures",
37+
"com.google.common.util.concurrent.ListenableFuture"))
2138
.transform(
2239
new AgentBuilder.Transformer.ForAdvice()
2340
.advice(

0 commit comments

Comments
 (0)