Update LogicalPlanner.java

apache · Aug 9, 2024 · e47d7ab · e47d7ab
1 parent 904e574
commit e47d7ab
Showing 1 changed file with 4 additions and 11 deletions.
diff --git a/...src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/LogicalPlanner.java b/...src/main/java/org/apache/iotdb/db/queryengine/plan/relational/planner/LogicalPlanner.java
@@ -24,7 +24,6 @@
 import org.apache.iotdb.db.queryengine.common.header.ColumnHeader;
 import org.apache.iotdb.db.queryengine.common.header.DatasetHeader;
 import org.apache.iotdb.db.queryengine.execution.warnings.WarningCollector;
-import org.apache.iotdb.db.queryengine.plan.analyze.QueryType;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.LogicalQueryPlan;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.PlanNode;
 import org.apache.iotdb.db.queryengine.plan.planner.plan.node.WritePlanNode;
@@ -215,8 +214,6 @@ private RelationPlanner getRelationPlanner(Analysis analysis) {
   }
 
   private PlanNode planCreateDevice(final CreateDevice statement, final Analysis analysis) {
-    queryContext.setQueryType(QueryType.WRITE);
-
     final CreateTableDeviceNode node =
         new CreateTableDeviceNode(
             queryContext.getQueryId().genPlanNodeId(),
@@ -238,8 +235,6 @@ private PlanNode planCreateDevice(final CreateDevice statement, final Analysis a
   }
 
   private PlanNode planFetchDevice(final FetchDevice statement, final Analysis analysis) {
-    queryContext.setQueryType(QueryType.READ);
-
     final List<ColumnHeader> columnHeaderList =
         getColumnHeaderList(statement.getDatabase(), statement.getTableName());
 
@@ -312,8 +307,6 @@ private PlanNode planCountDevice(final CountDevice statement, final Analysis ana
   }
 
   private String planQueryDevice(final AbstractQueryDevice statement, final Analysis analysis) {
-    queryContext.setQueryType(QueryType.READ);
-
     final String database =
         Objects.isNull(statement.getDatabase())
             ? analysis.getDatabaseName()
@@ -341,12 +334,12 @@ private String planQueryDevice(final AbstractQueryDevice statement, final Analys
     return database;
   }
 
-  private List<ColumnHeader> getColumnHeaderList(String database, String tableName) {
-    List<TsTableColumnSchema> columnSchemaList =
+  private List<ColumnHeader> getColumnHeaderList(final String database, final String tableName) {
+    final List<TsTableColumnSchema> columnSchemaList =
         DataNodeTableCache.getInstance().getTable(database, tableName).getColumnList();
 
-    List<ColumnHeader> columnHeaderList = new ArrayList<>(columnSchemaList.size());
-    for (TsTableColumnSchema columnSchema : columnSchemaList) {
+    final List<ColumnHeader> columnHeaderList = new ArrayList<>(columnSchemaList.size());
+    for (final TsTableColumnSchema columnSchema : columnSchemaList) {
       if (columnSchema.getColumnCategory().equals(TsTableColumnCategory.ID)
           || columnSchema.getColumnCategory().equals(TsTableColumnCategory.ATTRIBUTE)) {
         columnHeaderList.add(