cleanup cte in main query

shizy818 · shizy818 · commit 25529fa79eb2 · 2025-12-30T10:19:27.000+08:00
diff --git a/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/common/MPPQueryContext.java b/iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/common/MPPQueryContext.java
@@ -210,10 +210,19 @@ public void releaseMemoryForSchemaTree() {
   }
 
   public void prepareForRetry() {
+    if (!isSubquery()) {
+      cleanUpCte();
+    }
     this.initResultNodeContext();
     this.releaseAllMemoryReservedForFrontEnd();
   }
 
+  private void cleanUpCte() {
+    cteQueries.clear();
+    cteExplainResults.clear();
+    cteMaterializationCosts.clear();
+    subQueryTables.clear();
+  }
   private void initResultNodeContext() {
     this.resultNodeContext = new ResultNodeContext(queryId);
   }

Original file line number	Diff line number	Diff line change
`@@ -210,10 +210,19 @@ public void releaseMemoryForSchemaTree() {`
`210`	`210`	`}`
`211`	`211`
`212`	`212`	`public void prepareForRetry() {`
	`213`	`+ if (!isSubquery()) {`
	`214`	`+ cleanUpCte();`
	`215`	`+ }`
`213`	`216`	`this.initResultNodeContext();`
`214`	`217`	`this.releaseAllMemoryReservedForFrontEnd();`
`215`	`218`	`}`
`216`	`219`
	`220`	`+ private void cleanUpCte() {`
	`221`	`+ cteQueries.clear();`
	`222`	`+ cteExplainResults.clear();`
	`223`	`+ cteMaterializationCosts.clear();`
	`224`	`+ subQueryTables.clear();`
	`225`	`+ }`
`217`	`226`	`private void initResultNodeContext() {`
`218`	`227`	`this.resultNodeContext = new ResultNodeContext(queryId);`
`219`	`228`	`}`