Support Execute Immediate statement in TrinoQueryProperties

Author: JeongDaeKim

gateway-ha/src/main/java/io/trino/gateway/ha/router/TrinoQueryProperties.java

@@ -39,6 +39,7 @@
 import io.trino.sql.tree.DropSchema;
 import io.trino.sql.tree.DropTable;
 import io.trino.sql.tree.Execute;
+import io.trino.sql.tree.ExecuteImmediate;
 import io.trino.sql.tree.Identifier;
 import io.trino.sql.tree.Node;
 import io.trino.sql.tree.NodeLocation;
@@ -186,6 +187,10 @@ private void processRequestBody(HttpServletRequest request, RequestAnalyzerConfi
                 body = preparedStatements.get(statementName);
                 statement = parser.createStatement(body);
             }
+            else if (statement instanceof ExecuteImmediate executeImmediate) {
+                body = executeImmediate.getStatement().getValue();
+                statement = parser.createStatement(body);
+            }
 
             queryType = statement.getClass().getSimpleName();
             resourceGroupQueryType = StatementUtils.getQueryType(statement).toString();

gateway-ha/src/test/java/io/trino/gateway/ha/router/TestRoutingGroupSelector.java

@@ -379,7 +379,11 @@ private Stream<Arguments> provideTableExtractionQueries()
                 Arguments.of("SELECT * FROM TABLE(kat.system.funk(arg => 'expr'))",
                         ImmutableSet.of("kat"),
                         ImmutableSet.of("system"),
-                        ImmutableSet.of(QualifiedName.of("kat", "system", "funk"))));
+                        ImmutableSet.of(QualifiedName.of("kat", "system", "funk"))),
+                Arguments.of("EXECUTE IMMEDIATE 'SELECT * FROM cat.schem.tbl'",
+                        ImmutableSet.of("cat"),
+                        ImmutableSet.of("schem"),
+                        ImmutableSet.of(QualifiedName.of("cat", "schem", "tbl"))));
     }
 
     @ParameterizedTest