Added partial schema for tree device view

integration-test/src/test/java/org/apache/iotdb/relational/it/schema/IoTDBTableIT.java

@@ -19,25 +19,36 @@
 
 package org.apache.iotdb.relational.it.schema;
 
+import org.apache.iotdb.commons.client.sync.SyncConfigNodeIServiceClient;
+import org.apache.iotdb.commons.schema.table.column.FieldColumnSchema;
+import org.apache.iotdb.commons.schema.table.column.TagColumnSchema;
+import org.apache.iotdb.commons.schema.table.column.TsTableColumnSchemaUtil;
+import org.apache.iotdb.confignode.rpc.thrift.TConstructTreeDeviceViewReq;
 import org.apache.iotdb.db.it.utils.TestUtils;
 import org.apache.iotdb.it.env.EnvFactory;
 import org.apache.iotdb.it.framework.IoTDBTestRunner;
 import org.apache.iotdb.itbase.category.TableClusterIT;
 import org.apache.iotdb.itbase.category.TableLocalStandaloneIT;
 import org.apache.iotdb.itbase.env.BaseEnv;
+import org.apache.iotdb.rpc.TSStatusCode;
 
+import org.apache.tsfile.enums.TSDataType;
 import org.junit.AfterClass;
+import org.junit.Assert;
 import org.junit.BeforeClass;
 import org.junit.Test;
 import org.junit.experimental.categories.Category;
 import org.junit.runner.RunWith;
 
+import java.nio.ByteBuffer;
 import java.sql.Connection;
 import java.sql.ResultSet;
 import java.sql.ResultSetMetaData;
 import java.sql.SQLException;
 import java.sql.Statement;
+import java.util.Arrays;
 import java.util.Collections;
+import java.util.HashSet;
 
 import static org.apache.iotdb.commons.schema.column.ColumnHeaderConstant.describeTableColumnHeaders;
 import static org.apache.iotdb.commons.schema.column.ColumnHeaderConstant.describeTableDetailsColumnHeaders;
@@ -566,4 +577,66 @@ public void testManageTable() {
       fail(e.getMessage());
     }
   }
+
+  @Test
+  public void testTreeViewTable() throws Exception {
+    try (final Connection connection = EnvFactory.getEnv().getConnection();
+        final Statement statement = connection.createStatement()) {
+      statement.execute("create database root.a.b");
+      statement.execute("create timeSeries root.a.b.c.s1 int32");
+      statement.execute("create timeSeries root.a.b.c.s2 string");
+      statement.execute("create timeSeries root.a.b.s1 int32");
+      statement.execute("create timeSeries root.a.b.c.f.g.h.s1 int32");
+
+      // Put schema cache
+      statement.execute("select s1, s2 from root.a.b.c");
+    } catch (SQLException e) {
+      fail(e.getMessage());
+    }
+
+    try (final SyncConfigNodeIServiceClient client =
+            (SyncConfigNodeIServiceClient) EnvFactory.getEnv().getLeaderConfigNodeConnection();
+        final Connection connection = EnvFactory.getEnv().getConnection(BaseEnv.TABLE_SQL_DIALECT);
+        final Statement statement = connection.createStatement()) {
+      Assert.assertEquals(
+          TSStatusCode.SUCCESS_STATUS.getStatusCode(),
+          client
+              .constructTreeView(
+                  new TConstructTreeDeviceViewReq(
+                      "tree_view_db",
+                      "tree_table",
+                      "root.a.**",
+                      ByteBuffer.wrap(
+                          TsTableColumnSchemaUtil.serialize(
+                              Arrays.asList(
+                                  new TagColumnSchema("tag1", TSDataType.STRING),
+                                  new TagColumnSchema("tag2", TSDataType.STRING),
+                                  new FieldColumnSchema("s1", TSDataType.INT32),
+                                  new FieldColumnSchema("s2", TSDataType.STRING))))))
+              .getCode());
+      statement.execute("use tree_view_db");
+      TestUtils.assertResultSetEqual(
+          statement.executeQuery("desc tree_table"),
+          "ColumnName,DataType,Category,",
+          new HashSet<>(
+              Arrays.asList(
+                  "time,TIMESTAMP,TIME,",
+                  "tag1,STRING,TAG,",
+                  "tag2,STRING,TAG,",
+                  "s1,INT32,FIELD,",
+                  "s2,STRING,FIELD,")));
+      TestUtils.assertResultSetEqual(
+          statement.executeQuery("show devices from tree_table where tag1 = 'b'"),
+          "tag1,tag2,",
+          new HashSet<>(Arrays.asList("b,c,", "b,null,")));
+      TestUtils.assertResultSetEqual(
+          statement.executeQuery("show devices from tree_table where tag1 = 'b' and tag2 is null"),
+          "tag1,tag2,",
+          Collections.singleton("b,null,"));
+      TestUtils.assertResultSetEqual(
+          statement.executeQuery("count devices from tree_table"),
+          "count(devices),",
+          Collections.singleton("2,"));
+    }
+  }
 }

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/client/async/handlers/rpc/DataNodeAsyncRequestRPCHandler.java

@@ -210,6 +210,16 @@ public static DataNodeAsyncRequestRPCHandler<?> buildHandler(
       case KILL_QUERY_INSTANCE:
       case RESET_PEER_LIST:
       case TEST_CONNECTION:
+      case INVALIDATE_TABLE_CACHE:
+      case DELETE_DATA_FOR_DROP_TABLE:
+      case DELETE_DEVICES_FOR_DROP_TABLE:
+      case INVALIDATE_COLUMN_CACHE:
+      case DELETE_COLUMN_DATA:
+      case CONSTRUCT_TABLE_DEVICE_BLACK_LIST:
+      case ROLLBACK_TABLE_DEVICE_BLACK_LIST:
+      case INVALIDATE_MATCHED_TABLE_DEVICE_CACHE:
+      case DELETE_DATA_FOR_TABLE_DEVICE:
+      case DELETE_TABLE_DEVICE_IN_BLACK_LIST:
       default:
         return new DataNodeTSStatusRPCHandler(
             requestType,

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/consensus/request/ConfigPhysicalPlan.java

@@ -104,6 +104,7 @@
 import org.apache.iotdb.confignode.consensus.request.write.table.PreDeleteColumnPlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.PreDeleteTablePlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.RenameTableColumnPlan;
+import org.apache.iotdb.confignode.consensus.request.write.table.RenameTablePlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.RollbackCreateTablePlan;
 import org.apache.iotdb.confignode.consensus.request.write.table.SetTablePropertiesPlan;
 import org.apache.iotdb.confignode.consensus.request.write.template.CommitSetSchemaTemplatePlan;
@@ -371,6 +372,9 @@ public static ConfigPhysicalPlan create(final ByteBuffer buffer) throws IOExcept
         case CommitDeleteColumn:
           plan = new CommitDeleteColumnPlan();
           break;
+        case RenameTable:
+          plan = new RenameTablePlan();
+          break;
         case CreatePipeSinkV1:
           plan = new CreatePipeSinkPlanV1();
           break;

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/consensus/request/ConfigPhysicalPlanType.java

@@ -171,6 +171,7 @@ public enum ConfigPhysicalPlanType {
   PreDeleteColumn((short) 860),
   CommitDeleteColumn((short) 861),
   DescTable((short) 862),
+  RenameTable((short) 863),
 
   /** Deprecated types for sync, restored them for upgrade. */
   @Deprecated

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/consensus/request/read/database/GetDatabasePlan.java

@@ -25,11 +25,18 @@
 import java.util.List;
 
 public class GetDatabasePlan extends CountDatabasePlan {
+  private final boolean isShowDatabasePlan;
 
   public GetDatabasePlan(
       final List<String> storageGroupPathPattern,
       final PathPatternTree scope,
-      final boolean isTableModel) {
+      final boolean isTableModel,
+      final boolean isShowDatabasePlan) {
     super(ConfigPhysicalPlanType.GetDatabase, storageGroupPathPattern, scope, isTableModel);
+    this.isShowDatabasePlan = isShowDatabasePlan;
+  }
+
+  public boolean isShowDatabasePlan() {
+    return isShowDatabasePlan;
   }
 }

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/consensus/request/write/table/RenameTablePlan.java

@@ -0,0 +1,57 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package org.apache.iotdb.confignode.consensus.request.write.table;
+
+import org.apache.iotdb.confignode.consensus.request.ConfigPhysicalPlanType;
+
+import org.apache.tsfile.utils.ReadWriteIOUtils;
+
+import java.io.DataOutputStream;
+import java.io.IOException;
+import java.nio.ByteBuffer;
+
+public class RenameTablePlan extends AbstractTablePlan {
+  private String newName;
+
+  public RenameTablePlan() {
+    super(ConfigPhysicalPlanType.RenameTable);
+  }
+
+  public RenameTablePlan(final String database, final String tableName, final String newName) {
+    super(ConfigPhysicalPlanType.RenameTable, database, tableName);
+    this.newName = newName;
+  }
+
+  public String getNewName() {
+    return newName;
+  }
+
+  @Override
+  protected void serializeImpl(final DataOutputStream stream) throws IOException {
+    super.serializeImpl(stream);
+    ReadWriteIOUtils.write(newName, stream);
+  }
+
+  @Override
+  protected void deserializeImpl(final ByteBuffer buffer) throws IOException {
+    super.deserializeImpl(buffer);
+    this.newName = ReadWriteIOUtils.readString(buffer);
+  }
+}

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/manager/ConfigManager.java

@@ -51,8 +51,10 @@
 import org.apache.iotdb.commons.pipe.connector.payload.airgap.AirGapPseudoTPipeTransferRequest;
 import org.apache.iotdb.commons.schema.SchemaConstant;
 import org.apache.iotdb.commons.schema.table.AlterOrDropTableOperationType;
+import org.apache.iotdb.commons.schema.table.TreeViewSchema;
 import org.apache.iotdb.commons.schema.table.TsTable;
 import org.apache.iotdb.commons.schema.table.TsTableInternalRPCUtil;
+import org.apache.iotdb.commons.schema.table.column.TsTableColumnSchema;
 import org.apache.iotdb.commons.schema.ttl.TTLCache;
 import org.apache.iotdb.commons.service.metric.MetricService;
 import org.apache.iotdb.commons.utils.AuthUtils;
@@ -257,6 +259,7 @@
 import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Objects;
 import java.util.Set;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicBoolean;
@@ -1868,7 +1871,10 @@ public TShowDatabaseResp showDatabase(final TGetDatabaseReq req) {
               : PathPatternTree.deserialize(ByteBuffer.wrap(req.getScopePatternTree()));
       final GetDatabasePlan getDatabasePlan =
           new GetDatabasePlan(
-              req.getDatabasePathPattern(), scope, req.isSetIsTableModel() && req.isIsTableModel());
+              req.getDatabasePathPattern(),
+              scope,
+              req.isSetIsTableModel() && req.isIsTableModel(),
+              true);
       return getClusterSchemaManager().showDatabase(getDatabasePlan);
     } else {
       return new TShowDatabaseResp().setStatus(status);
@@ -2599,6 +2605,42 @@ public TThrottleQuotaResp getThrottleQuota() {
         : new TThrottleQuotaResp(status);
   }
 
+  @Override
+  public TSStatus createTreeViewTable(
+      final String databaseName,
+      final String tableName,
+      final PartialPath pathPattern,
+      final List<TsTableColumnSchema> columns,
+      final String ttl) {
+    TSStatus status;
+    // May be concurrently deleted, now just ignore
+    if (!clusterSchemaManager.isDatabaseExist(databaseName)) {
+      final TDatabaseSchema newSchema = new TDatabaseSchema(databaseName).setIsTableModel(true);
+      status = ClusterSchemaManager.enrichDatabaseSchemaWithDefaultProperties(newSchema);
+      if (TSStatusCode.SUCCESS_STATUS.getStatusCode() != status.getCode()) {
+        return status;
+      }
+      status =
+          setDatabase(new DatabaseSchemaPlan(ConfigPhysicalPlanType.CreateDatabase, newSchema));
+      if (TSStatusCode.SUCCESS_STATUS.getStatusCode() != status.getCode()
+          && TSStatusCode.DATABASE_ALREADY_EXISTS.getStatusCode() != status.getCode()) {
+        return status;
+      }
+    }
+
+    final TsTable table = new TsTable(tableName);
+    status = TreeViewSchema.setPathPattern(table, pathPattern);
+    if (TSStatusCode.SUCCESS_STATUS.getStatusCode() != status.getCode()) {
+      return status;
+    }
+
+    if (Objects.nonNull(ttl)) {
+      table.addProp(TsTable.TTL_PROPERTY, ttl);
+    }
+    columns.forEach(table::addColumnSchema);
+    return procedureManager.createTable(databaseName, table);
+  }
+
   @Override
   public TSStatus createTable(final ByteBuffer tableInfo) {
     final TSStatus status = confirmLeader();
@@ -2626,6 +2668,8 @@ public TSStatus alterOrDropTable(final TAlterOrDropTableReq req) {
           return procedureManager.alterTableDropColumn(req);
         case DROP_TABLE:
           return procedureManager.dropTable(req);
+        case RENAME_TABLE:
+          return procedureManager.renameTable(req);
         default:
           throw new IllegalArgumentException();
       }

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/manager/IManager.java

@@ -30,6 +30,7 @@
 import org.apache.iotdb.commons.cluster.NodeStatus;
 import org.apache.iotdb.commons.path.PartialPath;
 import org.apache.iotdb.commons.path.PathPatternTree;
+import org.apache.iotdb.commons.schema.table.column.TsTableColumnSchema;
 import org.apache.iotdb.confignode.consensus.request.read.ainode.GetAINodeConfigurationPlan;
 import org.apache.iotdb.confignode.consensus.request.read.auth.AuthorReadPlan;
 import org.apache.iotdb.confignode.consensus.request.read.database.CountDatabasePlan;
@@ -835,6 +836,13 @@ TDataPartitionTableResp getOrCreateDataPartition(
   /** Set space quota. */
   TSStatus setSpaceQuota(TSetSpaceQuotaReq req);
 
+  TSStatus createTreeViewTable(
+      final String databaseName,
+      final String tableName,
+      final PartialPath nameSpace,
+      final List<TsTableColumnSchema> columns,
+      final String ttl);
+
   TSStatus createTable(final ByteBuffer tableInfo);
 
   TSStatus alterOrDropTable(final TAlterOrDropTableReq req);

iotdb-core/confignode/src/main/java/org/apache/iotdb/confignode/manager/ProcedureManager.java

@@ -92,6 +92,7 @@
 import org.apache.iotdb.confignode.procedure.impl.schema.table.DropTableColumnProcedure;
 import org.apache.iotdb.confignode.procedure.impl.schema.table.DropTableProcedure;
 import org.apache.iotdb.confignode.procedure.impl.schema.table.RenameTableColumnProcedure;
+import org.apache.iotdb.confignode.procedure.impl.schema.table.RenameTableProcedure;
 import org.apache.iotdb.confignode.procedure.impl.schema.table.SetTablePropertiesProcedure;
 import org.apache.iotdb.confignode.procedure.impl.subscription.consumer.CreateConsumerProcedure;
 import org.apache.iotdb.confignode.procedure.impl.subscription.consumer.DropConsumerProcedure;
@@ -167,7 +168,7 @@ public class ProcedureManager {
   private static final CommonConfig COMMON_CONFIG = CommonDescriptor.getInstance().getConfig();
 
   public static final long PROCEDURE_WAIT_TIME_OUT = COMMON_CONFIG.getDnConnectionTimeoutInMS();
-  private static final int PROCEDURE_WAIT_RETRY_TIMEOUT = 10;
+  public static final int PROCEDURE_WAIT_RETRY_TIMEOUT = 10;
   private static final String PROCEDURE_TIMEOUT_MESSAGE =
       "Timed out to wait for procedure return. The procedure is still running.";
 
@@ -1549,6 +1550,17 @@ public TSStatus dropTable(final TAlterOrDropTableReq req) {
         new DropTableProcedure(req.database, req.tableName, req.queryId));
   }
 
+  public TSStatus renameTable(final TAlterOrDropTableReq req) {
+    return executeWithoutDuplicate(
+        req.database,
+        null,
+        req.tableName,
+        req.queryId,
+        ProcedureType.RENAME_TABLE_PROCEDURE,
+        new RenameTableProcedure(
+            req.database, req.tableName, req.queryId, ReadWriteIOUtils.readString(req.updateInfo)));
+  }
+
   public TDeleteTableDeviceResp deleteDevices(final TDeleteTableDeviceReq req) {
     long procedureId;
     DeleteDevicesProcedure procedure = null;
@@ -1650,6 +1662,7 @@ public Pair<Long, Boolean> checkDuplicateTableTask(
         case DROP_TABLE_COLUMN_PROCEDURE:
         case DROP_TABLE_PROCEDURE:
         case DELETE_DEVICES_PROCEDURE:
+        case RENAME_TABLE_PROCEDURE:
           final AbstractAlterOrDropTableProcedure<?> alterTableProcedure =
               (AbstractAlterOrDropTableProcedure<?>) procedure;
           if (type == thisType && queryId.equals(alterTableProcedure.getQueryId())) {