remove OutputNode from CopyToNode

Author: shuwenwei

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/common/header/DatasetHeader.java

@@ -20,7 +20,6 @@
 package org.apache.iotdb.db.queryengine.common.header;
 
 import org.apache.iotdb.commons.schema.column.ColumnHeader;
-import org.apache.iotdb.db.queryengine.plan.planner.plan.node.PlanNode;
 import org.apache.iotdb.db.queryengine.plan.relational.planner.Symbol;
 import org.apache.iotdb.db.queryengine.plan.relational.planner.node.OutputNode;
 
@@ -91,11 +90,7 @@ public void setTreeColumnToTsBlockIndexMap(List<String> outputColumnNames) {
   }
 
   public void setTableColumnToTsBlockIndexMap(OutputNode outputNode) {
-    setTableColumnToTsBlockIndexMap(outputNode, outputNode.getChild());
-  }
-
-  public void setTableColumnToTsBlockIndexMap(OutputNode outputNode, PlanNode outputNodeChild) {
-    List<Symbol> childOutputSymbols = outputNodeChild.getOutputSymbols();
+    List<Symbol> childOutputSymbols = outputNode.getChild().getOutputSymbols();
     Map<Symbol, Integer> outputSymbolsIndexMap = new HashMap<>(childOutputSymbols.size());
     for (int i = 0; i < childOutputSymbols.size(); i++) {
       outputSymbolsIndexMap.put(childOutputSymbols.get(i), i);

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/copyto/TableCopyToOperator.java

@@ -47,7 +47,7 @@ public class TableCopyToOperator implements ProcessOperator {
   private final String targetFilePath;
   private final CopyToOptions options;
   private final List<ColumnHeader> innerQueryColumnHeaders;
-  private final List<Integer> columnIndex2TsBlockColumnIndexList;
+  private final int[] columnIndex2TsBlockColumnIndex;
 
   private IFormatCopyToWriter writer;
   private File targetFile;
@@ -60,13 +60,13 @@ public TableCopyToOperator(
       String targetFilePath,
       CopyToOptions options,
       List<ColumnHeader> innerQueryColumnHeaders,
-      List<Integer> columnIndex2TsBlockColumnIndexList) {
+      int[] columnIndex2TsBlockColumnIndex) {
     this.operatorContext = operatorContext;
     this.childOperator = child;
     this.targetFilePath = targetFilePath;
     this.options = options;
     this.innerQueryColumnHeaders = innerQueryColumnHeaders;
-    this.columnIndex2TsBlockColumnIndexList = columnIndex2TsBlockColumnIndexList;
+    this.columnIndex2TsBlockColumnIndex = columnIndex2TsBlockColumnIndex;
   }
 
   @Override
@@ -104,7 +104,7 @@ private IFormatCopyToWriter getWriter() throws Exception {
                 this.targetFile,
                 (CopyToTsFileOptions) options,
                 innerQueryColumnHeaders,
-                columnIndex2TsBlockColumnIndexList);
+                columnIndex2TsBlockColumnIndex);
     }
     return writer;
   }
@@ -180,8 +180,7 @@ public long ramBytesUsed() {
         + RamUsageEstimator.sizeOf(targetFilePath)
         + RamUsageEstimator.sizeOfObject(innerQueryColumnHeaders)
         + MemoryEstimationHelper.getEstimatedSizeOfAccountableObject(operatorContext)
-        + MemoryEstimationHelper.getEstimatedSizeOfIntegerArrayList(
-            columnIndex2TsBlockColumnIndexList)
+        + RamUsageEstimator.sizeOf(columnIndex2TsBlockColumnIndex)
         + options.estimatedMaxRamBytesInWrite()
         + childOperator.ramBytesUsed();
   }

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/execution/operator/process/copyto/tsfile/TsFileFormatCopyToWriter.java

@@ -60,15 +60,15 @@ public TsFileFormatCopyToWriter(
       File file,
       CopyToTsFileOptions copyToOptions,
       List<ColumnHeader> innerQueryDatasetHeader,
-      List<Integer> columnIndex2TsBlockColumnIndexList) {
+      int[] columnIndex2TsBlockColumnIndex) {
     this.targetFile = file;
     this.targetTableName = copyToOptions.getTargetTableName();
     this.targetTimeColumn = copyToOptions.getTargetTimeColumn();
     this.generateNewTimeColumn = copyToOptions.isGenerateNewTimeColumn();
     targetTagColumns = copyToOptions.getTargetTagColumns();
 
     ColumnHeader[] columnHeadersMatchChildTsBlock =
-        getColumnHeadersMatchTsBlock(innerQueryDatasetHeader, columnIndex2TsBlockColumnIndexList);
+        getColumnHeadersMatchTsBlock(innerQueryDatasetHeader, columnIndex2TsBlockColumnIndex);
     List<ColumnSchema> columnSchemas =
         new ArrayList<>(columnHeadersMatchChildTsBlock.length + (generateNewTimeColumn ? 1 : 0));
     Map<String, Integer> columnNameIndexMapInDatasetHeader = new HashMap<>();
@@ -133,11 +133,10 @@ public TsFileFormatCopyToWriter(
   }
 
   private ColumnHeader[] getColumnHeadersMatchTsBlock(
-      List<ColumnHeader> queryOutputColumnHeaders,
-      List<Integer> columnIndex2TsBlockColumnIndexList) {
+      List<ColumnHeader> queryOutputColumnHeaders, int[] columnIndex2TsBlockColumnIndexList) {
     ColumnHeader[] columnHeadersMatchTsBlock = new ColumnHeader[queryOutputColumnHeaders.size()];
-    for (int i = 0; i < columnIndex2TsBlockColumnIndexList.size(); i++) {
-      int tsBlockIndex = columnIndex2TsBlockColumnIndexList.get(i);
+    for (int i = 0; i < columnIndex2TsBlockColumnIndexList.length; i++) {
+      int tsBlockIndex = columnIndex2TsBlockColumnIndexList[i];
       columnHeadersMatchTsBlock[tsBlockIndex] = queryOutputColumnHeaders.get(i);
     }
     return columnHeadersMatchTsBlock;

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

@@ -31,7 +31,6 @@
 import org.apache.iotdb.db.conf.IoTDBDescriptor;
 import org.apache.iotdb.db.exception.sql.SemanticException;
 import org.apache.iotdb.db.queryengine.common.FragmentInstanceId;
-import org.apache.iotdb.db.queryengine.common.header.DatasetHeader;
 import org.apache.iotdb.db.queryengine.execution.aggregation.timerangeiterator.ITableTimeRangeIterator;
 import org.apache.iotdb.db.queryengine.execution.aggregation.timerangeiterator.TableDateBinTimeRangeIterator;
 import org.apache.iotdb.db.queryengine.execution.aggregation.timerangeiterator.TableSingleTimeWindowIterator;
@@ -3480,8 +3479,17 @@ public Operator visitExplainAnalyze(ExplainAnalyzeNode node, LocalExecutionPlanC
   public Operator visitCopyTo(CopyToNode node, LocalExecutionPlanContext context) {
     PlanNode childNode = node.getChild();
 
-    DatasetHeader datasetHeader = node.getInnerQueryDatasetHeader();
-    datasetHeader.setTableColumnToTsBlockIndexMap(node.getInnerQueryOutputNode(), childNode);
+    List<Symbol> innerQueryOutputSymbols = node.getInnerQueryOutputSymbols();
+    List<Symbol> childOutputSymbols = childNode.getOutputSymbols();
+    Map<Symbol, Integer> childOutputSymbolsIndexMap = new HashMap<>(childOutputSymbols.size());
+    for (int i = 0; i < childOutputSymbols.size(); i++) {
+      childOutputSymbolsIndexMap.put(childOutputSymbols.get(i), i);
+    }
+    int[] columnIndex2TsBlockColumnIndexList = new int[innerQueryOutputSymbols.size()];
+    for (int i = 0; i < innerQueryOutputSymbols.size(); i++) {
+      int index = childOutputSymbolsIndexMap.get(innerQueryOutputSymbols.get(i));
+      columnIndex2TsBlockColumnIndexList[i] = index;
+    }
 
     Operator operator = childNode.accept(this, context);
 
@@ -3497,8 +3505,8 @@ public Operator visitCopyTo(CopyToNode node, LocalExecutionPlanContext context)
         operator,
         node.getTargetFilePath(),
         node.getCopyToOptions(),
-        datasetHeader.getColumnHeaders(),
-        datasetHeader.getColumnIndex2TsBlockColumnIndexList());
+        node.getInnerQueryDatasetHeader().getColumnHeaders(),
+        columnIndex2TsBlockColumnIndexList);
   }
 
   @Override

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

@@ -655,7 +655,7 @@ private RelationPlan createRelationPlan(final CopyTo statement, final Analysis a
             getChildPermittedOutputs(
                 analysis, statement.getQueryStatement(), innerQueryRelationPlan),
             innerQueryRespDatasetHeader,
-            outputNode);
+            outputNode.getOutputSymbols());
     return new RelationPlan(
         newRoot,
         innerQueryRelationPlan.getScope(),

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/node/CopyToNode.java

@@ -40,7 +40,7 @@ public class CopyToNode extends SingleChildProcessNode {
   private final String targetFilePath;
   private final CopyToOptions copyToOptions;
   private final List<Symbol> childPermittedOutputs;
-  private final OutputNode innerQueryOutputNode;
+  private final List<Symbol> innerQueryOutputSymbols;
   private final DatasetHeader innerQueryDatasetHeader;
 
   public CopyToNode(
@@ -50,22 +50,22 @@ public CopyToNode(
       CopyToOptions copyToOptions,
       List<Symbol> childPermittedOutputs,
       DatasetHeader innerQueryDatasetHeader,
-      OutputNode innerQueryOutputNode) {
+      List<Symbol> innerQueryOutputSymbols) {
     super(id);
     this.child = child;
     this.targetFilePath = targetFilePath;
     this.copyToOptions = copyToOptions;
     this.childPermittedOutputs = childPermittedOutputs;
     this.innerQueryDatasetHeader = innerQueryDatasetHeader;
-    this.innerQueryOutputNode = innerQueryOutputNode;
+    this.innerQueryOutputSymbols = innerQueryOutputSymbols;
   }
 
   public DatasetHeader getInnerQueryDatasetHeader() {
     return innerQueryDatasetHeader;
   }
 
-  public OutputNode getInnerQueryOutputNode() {
-    return innerQueryOutputNode;
+  public List<Symbol> getInnerQueryOutputSymbols() {
+    return innerQueryOutputSymbols;
   }
 
   public String getTargetFilePath() {
@@ -89,7 +89,7 @@ public PlanNode clone() {
         copyToOptions,
         childPermittedOutputs,
         innerQueryDatasetHeader,
-        innerQueryOutputNode);
+        innerQueryOutputSymbols);
   }
 
   @Override
@@ -120,7 +120,7 @@ public PlanNode replaceChildren(List<PlanNode> newChildren) {
         copyToOptions,
         childPermittedOutputs,
         innerQueryDatasetHeader,
-        innerQueryOutputNode);
+        innerQueryOutputSymbols);
   }
 
   @Override

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/optimizations/UnaliasSymbolReferences.java

@@ -403,7 +403,7 @@ public PlanAndMappings visitCopyTo(CopyToNode node, UnaliasContext context) {
               node.getCopyToOptions(),
               newChildPermittedOutputs,
               node.getInnerQueryDatasetHeader(),
-              node.getInnerQueryOutputNode()),
+              node.getInnerQueryOutputSymbols()),
           mapping);
     }