use class from influxdb client for call depth counting

open-telemetry · May 6, 2024 · 4b5d136 · 4b5d136
1 parent 5f50fce
commit 4b5d136
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...io/opentelemetry/javaagent/instrumentation/influxdb/v2_4/InfluxDbImplInstrumentation.java b/...io/opentelemetry/javaagent/instrumentation/influxdb/v2_4/InfluxDbImplInstrumentation.java
@@ -79,7 +79,7 @@ public static void onEnter(
         @Advice.Local("otelRequest") InfluxDbRequest influxDbRequest,
         @Advice.Local("otelContext") Context context,
         @Advice.Local("otelScope") Scope scope) {
-      callDepth = CallDepth.forClass(Retrofit.class);
+      callDepth = CallDepth.forClass(InfluxDBImpl.class);
       if (callDepth.getAndIncrement() > 0) {
         return;
       }
@@ -137,7 +137,7 @@ public static void onEnter(
         @Advice.Local("otelRequest") InfluxDbRequest influxDbRequest,
         @Advice.Local("otelContext") Context context,
         @Advice.Local("otelScope") Scope scope) {
-      callDepth = CallDepth.forClass(Retrofit.class);
+      callDepth = CallDepth.forClass(InfluxDBImpl.class);
       if (callDepth.getAndIncrement() > 0) {
         return;
       }