[iotdb] branch master updated: [IOTDB-2803] support AlterTimeseries sql (#5735)

[ha...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

haonan pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/iotdb.git


The following commit(s) were added to refs/heads/master by this push:
     new d4647beaef [IOTDB-2803] support AlterTimeseries sql (#5735)
d4647beaef is described below

commit d4647beaefebc7212234baf2cbe4ff73898323d8
Author: Yifu Zhou <ef...@outlook.com>
AuthorDate: Sat Apr 30 11:21:43 2022 +0800

    [IOTDB-2803] support AlterTimeseries sql (#5735)
---
 .../node/metedata/write/AlterTimeSeriesNode.java   | 26 +++++++++++++++++++++-
 .../write/CreateAlignedTimeSeriesNode.java         |  7 ------
 2 files changed, 25 insertions(+), 8 deletions(-)

diff --git a/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/AlterTimeSeriesNode.java b/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/AlterTimeSeriesNode.java
index 14803278a1..baff24f28a 100644
--- a/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/AlterTimeSeriesNode.java
+++ b/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/AlterTimeSeriesNode.java
@@ -19,23 +19,28 @@
 
 package org.apache.iotdb.db.mpp.sql.planner.plan.node.metedata.write;
 
+import org.apache.iotdb.common.rpc.thrift.TRegionReplicaSet;
 import org.apache.iotdb.db.exception.metadata.IllegalPathException;
 import org.apache.iotdb.db.metadata.path.PartialPath;
+import org.apache.iotdb.db.mpp.sql.analyze.Analysis;
 import org.apache.iotdb.db.mpp.sql.planner.plan.node.PlanNode;
 import org.apache.iotdb.db.mpp.sql.planner.plan.node.PlanNodeId;
 import org.apache.iotdb.db.mpp.sql.planner.plan.node.PlanNodeType;
 import org.apache.iotdb.db.mpp.sql.planner.plan.node.PlanVisitor;
+import org.apache.iotdb.db.mpp.sql.planner.plan.node.WritePlanNode;
 import org.apache.iotdb.db.mpp.sql.statement.metadata.AlterTimeSeriesStatement.AlterType;
 import org.apache.iotdb.tsfile.exception.NotImplementedException;
 import org.apache.iotdb.tsfile.utils.ReadWriteIOUtils;
 
+import com.google.common.collect.ImmutableList;
+
 import java.nio.ByteBuffer;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 
-public class AlterTimeSeriesNode extends PlanNode {
+public class AlterTimeSeriesNode extends WritePlanNode {
   private PartialPath path;
   private AlterType alterType;
 
@@ -53,6 +58,8 @@ public class AlterTimeSeriesNode extends PlanNode {
   private Map<String, String> tagsMap;
   private Map<String, String> attributesMap;
 
+  private TRegionReplicaSet regionReplicaSet;
+
   public AlterTimeSeriesNode(
       PlanNodeId id,
       PartialPath path,
@@ -270,4 +277,21 @@ public class AlterTimeSeriesNode extends PlanNode {
     return Objects.hash(
         this.getPlanNodeId(), path, alias, alterType, alterMap, attributesMap, tagsMap);
   }
+
+  @Override
+  public TRegionReplicaSet getRegionReplicaSet() {
+    return regionReplicaSet;
+  }
+
+  public void setRegionReplicaSet(TRegionReplicaSet regionReplicaSet) {
+    this.regionReplicaSet = regionReplicaSet;
+  }
+
+  @Override
+  public List<WritePlanNode> splitByPartition(Analysis analysis) {
+    TRegionReplicaSet regionReplicaSet =
+        analysis.getSchemaPartitionInfo().getSchemaRegionReplicaSet(path.getDevice());
+    setRegionReplicaSet(regionReplicaSet);
+    return ImmutableList.of(this);
+  }
 }
diff --git a/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/CreateAlignedTimeSeriesNode.java b/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/CreateAlignedTimeSeriesNode.java
index da3ef0afc7..99dd453772 100644
--- a/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/CreateAlignedTimeSeriesNode.java
+++ b/server/src/main/java/org/apache/iotdb/db/mpp/sql/planner/plan/node/metedata/write/CreateAlignedTimeSeriesNode.java
@@ -376,11 +376,4 @@ public class CreateAlignedTimeSeriesNode extends WritePlanNode {
     setRegionReplicaSet(regionReplicaSet);
     return ImmutableList.of(this);
   }
-
-  //  @Override
-  //  public void executeOn(SchemaRegion schemaRegion) throws MetadataException {
-  //    schemaRegion.createAlignedTimeSeries((CreateAlignedTimeSeriesPlan)
-  // transferToPhysicalPlan());
-  //  }
-
 }