Implemented mem ctrl for tree-model plan node

Author: Caideyipi

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/planner/LogicalPlanBuilder.java

@@ -86,8 +86,8 @@
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.DeviceRegionScanNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.LastQueryScanNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.SeriesScanNode;
-import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.SeriesSourceNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.ShowQueriesNode;
+import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.SourceNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.source.TimeseriesRegionScanNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.parameter.AggregationDescriptor;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.parameter.AggregationStep;
@@ -110,6 +110,7 @@
 import org.apache.commons.lang3.Validate;
 import org.apache.tsfile.enums.TSDataType;
 import org.apache.tsfile.file.metadata.IDeviceID;
+import org.apache.tsfile.utils.Accountable;
 
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -1114,15 +1115,16 @@ public LogicalPlanBuilder planSeriesSchemaFetchSource(
           overlappedPatternTree.appendFullPath(pathPattern);
         }
         this.root.addChild(
-            new SeriesSchemaFetchScanNode(
-                context.getQueryId().genPlanNodeId(),
-                storageGroupPath,
-                overlappedPatternTree,
-                templateMap,
-                withTags,
-                withAttributes,
-                withTemplate,
-                withAliasForce));
+            reserveMemoryForAccountableNode(
+                new SeriesSchemaFetchScanNode(
+                    context.getQueryId().genPlanNodeId(),
+                    storageGroupPath,
+                    overlappedPatternTree,
+                    templateMap,
+                    withTags,
+                    withAttributes,
+                    withTemplate,
+                    withAliasForce)));
       } catch (IllegalPathException e) {
         // definitely won't happen
         throw new RuntimeException(e);
@@ -1146,11 +1148,12 @@ public LogicalPlanBuilder planDeviceSchemaFetchSource(
           overlappedPatternTree.appendFullPath(pathPattern);
         }
         this.root.addChild(
-            new DeviceSchemaFetchScanNode(
-                context.getQueryId().genPlanNodeId(),
-                databasePath,
-                overlappedPatternTree,
-                authorityScope));
+            reserveMemoryForAccountableNode(
+                new DeviceSchemaFetchScanNode(
+                    context.getQueryId().genPlanNodeId(),
+                    databasePath,
+                    overlappedPatternTree,
+                    authorityScope)));
       } catch (IllegalPathException e) {
         // definitely won't happen
         throw new RuntimeException(e);
@@ -1425,7 +1428,8 @@ public LogicalPlanBuilder planTimeseriesRegionScan(
    * We need to check the memory used by SeriesSourceNodes.(Number of other PlanNodes are rather
    * small compared to SourceNodes and could be safely ignored for now.)
    */
-  private PlanNode reserveMemoryForAccountableNode(final SeriesSourceNode sourceNode) {
+  private <T extends SourceNode & Accountable> T reserveMemoryForAccountableNode(
+      final T sourceNode) {
     this.context.reserveMemoryForFrontEnd(
         MemoryEstimationHelper.getEstimatedSizeOfAccountableObject(sourceNode));
     return sourceNode;