Properly handle the depth increase and decrease

lhotari · Feb 3, 2024 · 5f8fef5 · 5f8fef5
1 parent a751294
commit 5f8fef5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bookkeeper-server/src/main/java/org/apache/bookkeeper/util/ByteBufVisitor.java b/bookkeeper-server/src/main/java/org/apache/bookkeeper/util/ByteBufVisitor.java
@@ -124,7 +124,6 @@ public void visitBuffer(InternalContext<T> internalContext, ByteBuf visitBuffer,
                 internalContext.userCallback.visitBuffer(internalContext.userContext, visitBuffer,
                         visitIndex, visitLength);
             } else {
-                internalContext.depth++;
                 // use the doRecursivelyVisitBuffers method to visit the wrapped buffer, possibly recursively
                 doRecursivelyVisitBuffers(visitBuffer, visitIndex, visitLength, this, internalContext);
             }
@@ -152,7 +151,9 @@ private static <T> void doRecursivelyVisitBuffers(ByteBuf buffer, int offset, in
             internalContext.parentBuffer = buffer;
             internalContext.parentOffset = offset;
             internalContext.parentLength = length;
+            internalContext.depth++;
             visitBuffersImpl(buffer, offset, length, callback, internalContext);
+            internalContext.depth--;
         } else {
             // visit the buffer
             internalContext.userCallback.visitBuffer(internalContext.userContext, buffer, offset, length);