fix depth count

DataDog · Nov 16, 2023 · 46e8651 · 46e8651
1 parent 619cb0f
commit 46e8651
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/...c/main/java/datadog/trace/instrumentation/kafka_clients/KafkaProducerInstrumentation.java b/...c/main/java/datadog/trace/instrumentation/kafka_clients/KafkaProducerInstrumentation.java
@@ -131,6 +131,9 @@ record =
     @Advice.OnMethodExit(onThrowable = Throwable.class, suppress = Throwable.class)
     public static void stopSpan(
         @Advice.Enter final AgentScope scope, @Advice.Thrown final Throwable throwable) {
+      // reset depth count for EstimateSizeAdvice
+      CallDepthThreadLocalMap.reset(AbstractRecords.class);
+
       PRODUCER_DECORATE.onError(scope, throwable);
       PRODUCER_DECORATE.beforeFinish(scope);
       scope.close();