Fix highestPriority of SHOW QUERIES are not set correctly (#17505)

Author: Wei-hao-Li

iotdb-core/datanode/src/main/java/org/apache/iotdb/db/queryengine/plan/relational/analyzer/Analysis.java

@@ -62,7 +62,6 @@
 import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.RangeQuantifier;
 import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.Relation;
 import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.RowPattern;
-import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.ShowStatement;
 import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.Statement;
 import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.SubqueryExpression;
 import org.apache.iotdb.db.queryengine.plan.relational.sql.ast.SubsetDefinition;
@@ -118,6 +117,7 @@ public class Analysis implements IAnalysis {
   private List<TEndPoint> redirectNodeList;
 
   @Nullable private Statement root;
+  private boolean needSetHighestPriority;
 
   private final Map<NodeRef<Parameter>, Expression> parameters;
 
@@ -268,6 +268,14 @@ public Analysis(@Nullable Statement root, Map<NodeRef<Parameter>, Expression> pa
     this.parameters = ImmutableMap.copyOf(requireNonNull(parameters, "parameters is null"));
   }
 
+  public void updateNeedSetHighestPriority(QualifiedObjectName tableName) {
+    if (needSetHighestPriority) {
+      return;
+    }
+    //  the database of table has been judged to be 'information_schema' in outer
+    needSetHighestPriority = InformationSchema.QUERIES.equals(tableName.getObjectName());
+  }
+
   public Map<NodeRef<Parameter>, Expression> getParameters() {
     return parameters;
   }
@@ -960,8 +968,7 @@ public void setQuery(boolean query) {
 
   @Override
   public boolean needSetHighestPriority() {
-    return root instanceof ShowStatement
-        && ((ShowStatement) root).getTableName().equals(InformationSchema.QUERIES);
+    return needSetHighestPriority;
   }
 
   @Override

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

@@ -390,6 +390,7 @@ private RelationPlan processPhysicalTable(Table table, Scope scope) {
       tableScanNode =
           new InformationSchemaTableScanNode(
               idAllocator.genPlanNodeId(), qualifiedObjectName, outputSymbols, tableColumnSchema);
+      analysis.updateNeedSetHighestPriority(qualifiedObjectName);
     } else {
       tableScanNode =
           new DeviceTableScanNode(

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/plan/relational/analyzer/NeedSetHighestPriorityTest.java

@@ -0,0 +1,56 @@
+/*
+ * 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.db.queryengine.plan.relational.analyzer;
+
+import org.apache.iotdb.db.queryengine.plan.relational.planner.PlanTester;
+
+import org.junit.Test;
+
+import static org.junit.Assert.assertTrue;
+
+public class NeedSetHighestPriorityTest {
+
+  @Test
+  public void testShowQueriesNeedSetHighestPriority() {
+    final Analysis analysis = planAndGetAnalysis("show queries");
+    assertTrue(analysis.needSetHighestPriority());
+  }
+
+  @Test
+  public void testInformationSchemaQueriesNeedSetHighestPriority() {
+    final Analysis analysis = planAndGetAnalysis("select query_id from information_schema.queries");
+    assertTrue(analysis.needSetHighestPriority());
+  }
+
+  @Test
+  public void testNestedInformationSchemaQueriesNeedSetHighestPriority() {
+    final Analysis analysis =
+        planAndGetAnalysis(
+            "select * from (select query_id from information_schema.queries) t "
+                + "where t.query_id is not null");
+    assertTrue(analysis.needSetHighestPriority());
+  }
+
+  private Analysis planAndGetAnalysis(final String sql) {
+    PlanTester planTester = new PlanTester();
+    planTester.createPlan(sql);
+    return planTester.getAnalysis();
+  }
+}

iotdb-core/datanode/src/test/java/org/apache/iotdb/db/queryengine/plan/relational/planner/PlanTester.java

@@ -161,6 +161,10 @@ public LogicalQueryPlan createPlan(
     return plan;
   }
 
+  public Analysis getAnalysis() {
+    return analysis;
+  }
+
   public LogicalQueryPlan createPlan(
       SessionInfo sessionInfo,
       String sql,