[pinpoint-apm#9693] CallStack OverFlow check is missing in AsyncChild…

…Trace
emeroad · Feb 13, 2023 · 84eacea · 84eacea
1 parent f40b797
commit 84eacea
Show file tree

Hide file tree

Showing 5 changed files with 22 additions and 14 deletions.
diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/AsyncChildTrace.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/AsyncChildTrace.java
@@ -258,20 +258,20 @@ public SpanEventRecorder currentSpanEventRecorder() {
     }
 
     private SpanEvent newSpanEvent(int stackId) {
-        final SpanEvent spanEvent = callStack.getFactory().newInstance();
+        final SpanEvent spanEvent = callStack.newInstance();
         spanEvent.markStartTime();
         spanEvent.setStackId(stackId);
         return spanEvent;
     }
 
     @VisibleForTesting
     SpanEvent dummySpanEvent() {
-        return callStack.getFactory().disableInstance();
+        return callStack.disableInstance();
     }
 
     @VisibleForTesting
     boolean isDummySpanEvent(final SpanEvent spanEvent) {
-        return callStack.getFactory().isDisable(spanEvent);
+        return callStack.isDisable(spanEvent);
     }
 
     @Override

diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/CallStack.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/CallStack.java
@@ -38,7 +38,9 @@ public interface CallStack<T> {
 
     T newInstance();
 
-    Factory<T> getFactory();
+    T disableInstance();
+
+    boolean isDisable(T element);
 
     void setOverflowListener(CallStackOverflowListener overflowListener);
 

diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultCallStack.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultCallStack.java
@@ -190,8 +190,14 @@ public T newInstance() {
     }
 
     @Override
-    public Factory<T> getFactory() {
-        return factory;
+    public T disableInstance() {
+        return this.factory.disableInstance();
+    }
+
+    @Override
+    public boolean isDisable(T element) {
+        return this.factory.isDisable(element);
+
     }
 
     @Override

diff --git a/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultTrace.java b/profiler/src/main/java/com/navercorp/pinpoint/profiler/context/DefaultTrace.java
@@ -282,12 +282,12 @@ private SpanEvent newSpanEvent(int stackId) {
 
     @VisibleForTesting
     SpanEvent dummySpanEvent() {
-        return callStack.getFactory().disableInstance();
+        return callStack.disableInstance();
     }
 
     @VisibleForTesting
     boolean isDummySpanEvent(final SpanEvent spanEvent) {
-        return callStack.getFactory().isDisable(spanEvent);
+        return callStack.isDisable(spanEvent);
     }
 
     @Override

diff --git a/profiler/src/test/java/com/navercorp/pinpoint/profiler/context/CallStackTest.java b/profiler/src/test/java/com/navercorp/pinpoint/profiler/context/CallStackTest.java
@@ -115,7 +115,7 @@ public void newInstance() {
         callStack.push(spanEvent1);
 
         SpanEvent spanEvent2 = callStack.newInstance();
-        assertTrue(callStack.getFactory().isDisable(spanEvent2));
+        assertTrue(callStack.isDisable(spanEvent2));
     }
 
     @Test
@@ -177,19 +177,19 @@ public void overflow2() {
         // overflow by sequence
         assertEquals(maxDepth - 1, callStack.getIndex());
         assertTrue(callStack.isOverflow());
-        assertFalse(callStack.getFactory().isDisable(callStack.peek()));
+        assertFalse(callStack.isDisable(callStack.peek()));
 
         callStack.push(getSpanEvent());
         assertEquals(maxDepth, callStack.getIndex());
-        assertTrue(callStack.getFactory().isDisable(callStack.peek()));
-        assertTrue(callStack.getFactory().isDisable(callStack.pop()));
+        assertTrue(callStack.isDisable(callStack.peek()));
+        assertTrue(callStack.isDisable(callStack.pop()));
 
         for (int i = maxDepth - 1; i > 0; i--) {
             assertNotNull(callStack.peek());
-            assertFalse(callStack.getFactory().isDisable(callStack.peek()));
+            assertFalse(callStack.isDisable(callStack.peek()));
             SpanEvent element = callStack.pop();
             assertNotNull(element);
-            assertFalse(callStack.getFactory().isDisable(element));
+            assertFalse(callStack.isDisable(element));
             assertTrue(callStack.isOverflow());
         }
     }