Skip to content

Commit 9e058b9

Browse files
authored
Code review sweep (run 25135365127) (#18421)
Co-authored-by: otelbot <197425009+otelbot@users.noreply.github.com>
1 parent fd0784c commit 9e058b9

9 files changed

Lines changed: 75 additions & 67 deletions

File tree

instrumentation/cassandra/cassandra-4.0/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/cassandra/v4_0/CassandraClientInstrumentationModule.java

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -29,13 +29,9 @@ public List<TypeInstrumentation> typeInstrumentations() {
2929

3030
@Override
3131
public ElementMatcher.Junction<ClassLoader> classLoaderMatcher() {
32-
return hasClassesNamed(
33-
// added in 4.0
34-
"com.datastax.oss.driver.api.core.CqlSession")
35-
.and(
36-
not(
37-
hasClassesNamed(
38-
// added in 4.4
39-
"com.datastax.dse.driver.api.core.cql.reactive.ReactiveSession")));
32+
// added in 4.0
33+
return hasClassesNamed("com.datastax.oss.driver.api.core.CqlSession")
34+
// added in 4.4
35+
.and(not(hasClassesNamed("com.datastax.dse.driver.api.core.cql.reactive.ReactiveSession")));
4036
}
4137
}

instrumentation/cassandra/cassandra-4.4/javaagent/build.gradle.kts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ muzzle {
66
pass {
77
group.set("com.datastax.oss")
88
module.set("java-driver-core")
9-
versions.set("[4.4,]")
9+
versions.set("[4.4,)")
1010
assertInverse.set(true)
1111
}
1212
pass {

instrumentation/cassandra/cassandra-4.4/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/cassandra/v4_4/CompletionStageFunction.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55

66
package io.opentelemetry.javaagent.instrumentation.cassandra.v4_4;
77

8+
import static io.opentelemetry.javaagent.instrumentation.cassandra.v4_4.CassandraSingletons.telemetry;
9+
810
import com.datastax.oss.driver.api.core.CqlSession;
911
import java.util.function.Function;
1012

@@ -19,6 +21,6 @@ public Object apply(Object session) {
1921
if (session.getClass().getName().endsWith("cassandra4.TracingCqlSession")) {
2022
return session;
2123
}
22-
return CassandraSingletons.telemetry().wrap((CqlSession) session);
24+
return telemetry().wrap((CqlSession) session);
2325
}
2426
}

instrumentation/clickhouse/clickhouse-client-v1-0.5/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/clickhouse/clientv1/v0_5/ClickHouseClientV1Instrumentation.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
import io.opentelemetry.javaagent.extension.instrumentation.TypeTransformer;
2323
import io.opentelemetry.javaagent.instrumentation.clickhouse.common.ClickHouseDbRequest;
2424
import io.opentelemetry.javaagent.instrumentation.clickhouse.common.ClickHouseScope;
25+
import javax.annotation.Nullable;
2526
import net.bytebuddy.asm.Advice;
2627
import net.bytebuddy.description.type.TypeDescription;
2728
import net.bytebuddy.matcher.ElementMatcher;
@@ -48,8 +49,9 @@ public void transform(TypeTransformer transformer) {
4849
@SuppressWarnings("unused")
4950
public static class ExecuteAndWaitAdvice {
5051
@Advice.OnMethodEnter(suppress = Throwable.class, inline = false)
52+
@Nullable
5153
public static ClickHouseScope onEnter(
52-
@Advice.Argument(0) ClickHouseRequest<?> clickHouseRequest) {
54+
@Advice.Argument(0) @Nullable ClickHouseRequest<?> clickHouseRequest) {
5355

5456
CallDepth callDepth = CallDepth.forClass(ClickHouseClient.class);
5557
if (callDepth.getAndIncrement() > 0 || clickHouseRequest == null) {
@@ -71,7 +73,8 @@ public static ClickHouseScope onEnter(
7173

7274
@Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class, inline = false)
7375
public static void onExit(
74-
@Advice.Thrown Throwable throwable, @Advice.Enter ClickHouseScope scope) {
76+
@Advice.Thrown @Nullable Throwable throwable,
77+
@Advice.Enter @Nullable ClickHouseScope scope) {
7578

7679
CallDepth callDepth = CallDepth.forClass(ClickHouseClient.class);
7780
if (callDepth.decrementAndGet() > 0 || scope == null) {

instrumentation/clickhouse/clickhouse-client-v2-0.8/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/clickhouse/clientv2/v0_8/ClickHouseClientV2Instrumentation.java

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,7 @@
2121
import io.opentelemetry.javaagent.instrumentation.clickhouse.common.ClickHouseDbRequest;
2222
import io.opentelemetry.javaagent.instrumentation.clickhouse.common.ClickHouseScope;
2323
import java.util.Map;
24+
import javax.annotation.Nullable;
2425
import net.bytebuddy.asm.Advice;
2526
import net.bytebuddy.description.type.TypeDescription;
2627
import net.bytebuddy.matcher.ElementMatcher;
@@ -45,8 +46,9 @@ public void transform(TypeTransformer transformer) {
4546
@SuppressWarnings("unused")
4647
public static class QueryAdvice {
4748
@Advice.OnMethodEnter(suppress = Throwable.class, inline = false)
49+
@Nullable
4850
public static ClickHouseScope onEnter(
49-
@Advice.This Client client, @Advice.Argument(0) String sqlQuery) {
51+
@Advice.This Client client, @Advice.Argument(0) @Nullable String sqlQuery) {
5052
CallDepth callDepth = CallDepth.forClass(Client.class);
5153
if (callDepth.getAndIncrement() > 0 || sqlQuery == null) {
5254
return null;
@@ -72,7 +74,8 @@ public static ClickHouseScope onEnter(
7274

7375
@Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class, inline = false)
7476
public static void onExit(
75-
@Advice.Thrown Throwable throwable, @Advice.Enter ClickHouseScope scope) {
77+
@Advice.Thrown @Nullable Throwable throwable,
78+
@Advice.Enter @Nullable ClickHouseScope scope) {
7679
CallDepth callDepth = CallDepth.forClass(Client.class);
7780
if (callDepth.decrementAndGet() > 0 || scope == null) {
7881
return;

instrumentation/clickhouse/clickhouse-client-v2-0.8/javaagent/src/main/java/io/opentelemetry/javaagent/instrumentation/clickhouse/clientv2/v0_8/ClickHouseClientV2Singletons.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@
1313
import io.opentelemetry.instrumentation.api.util.VirtualField;
1414
import io.opentelemetry.javaagent.instrumentation.clickhouse.common.ClickHouseDbRequest;
1515
import io.opentelemetry.javaagent.instrumentation.clickhouse.common.ClickHouseInstrumenterFactory;
16+
import javax.annotation.Nullable;
1617

1718
public class ClickHouseClientV2Singletons {
1819

@@ -37,11 +38,12 @@ public static Instrumenter<ClickHouseDbRequest, Void> instrumenter() {
3738
return instrumenter;
3839
}
3940

41+
@Nullable
4042
public static AddressAndPort getAddressAndPort(Client client) {
4143
return ADDRESS_AND_PORT.get(client);
4244
}
4345

44-
public static AddressAndPort setAddressAndPort(Client client, String endpoint) {
46+
public static AddressAndPort setAddressAndPort(Client client, @Nullable String endpoint) {
4547
AddressAndPort addressAndPort = new AddressAndPort();
4648

4749
if (endpoint != null) {

0 commit comments

Comments
 (0)