Early release AcquireReleaseColumnsSegmentOperator during explain plan (

apache#11945) Co-authored-by: Saurabh Dubey <[email protected]>
soumitra-st · Nov 3, 2023 · 18639b9 · 18639b9
1 parent bbeb1da
commit 18639b9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pinot-core/src/main/java/org/apache/pinot/core/common/Operator.java b/pinot-core/src/main/java/org/apache/pinot/core/common/Operator.java
@@ -51,6 +51,9 @@ public interface Operator<T extends Block> {
   default void prepareForExplainPlan(ExplainPlanRows explainPlanRows) {
   }
 
+  default void postExplainPlan(ExplainPlanRows explainPlanRows) {
+  }
+
   default void explainPlan(ExplainPlanRows explainPlanRows, int[] globalId, int parentId) {
     prepareForExplainPlan(explainPlanRows);
     String explainPlanString = toExplainString();
@@ -66,6 +69,7 @@ default void explainPlan(ExplainPlanRows explainPlanRows, int[] globalId, int pa
         child.explainPlan(explainPlanRows, globalId, parentId);
       }
     }
+    postExplainPlan(explainPlanRows);
   }
 
   /**

diff --git a/...re/src/main/java/org/apache/pinot/core/operator/AcquireReleaseColumnsSegmentOperator.java b/...re/src/main/java/org/apache/pinot/core/operator/AcquireReleaseColumnsSegmentOperator.java
@@ -92,6 +92,11 @@ public void prepareForExplainPlan(ExplainPlanRows explainPlanRows) {
     materializeChildOperator();
   }
 
+  @Override
+  public void postExplainPlan(ExplainPlanRows explainPlanRows) {
+    release();
+  }
+
   @Override
   public String toExplainString() {
     return EXPLAIN_NAME;