ignore temporary tables

Author: Chaho12

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

@@ -345,7 +345,12 @@ private void getNames(Node node, ImmutableSet.Builder<QualifiedName> tableBuilde
             case SetSchemaAuthorization s -> setCatalogAndSchemaNameFromSchemaQualifiedName(Optional.of(s.getSource()), catalogBuilder, schemaBuilder, catalogSchemaBuilder);
             case SetTableAuthorization s -> tableBuilder.add(qualifyName(s.getSource()));
             case SetViewAuthorization s -> tableBuilder.add(qualifyName(s.getSource()));
-            case Table s -> tableBuilder.add(qualifyName(s.getName()));
+            case Table s -> {
+                // ignore temporary tables as they can have various table parts
+                if (!temporaryTables.contains(s.getName())) {
+                    tableBuilder.add(qualifyName(s.getName()));
+                }
+            }
             case TableFunctionInvocation s -> tableBuilder.add(qualifyName(s.getName()));
             case WithQuery withQuery -> temporaryTables.add(QualifiedName.of(withQuery.getName().getValue()));
             default -> {}
@@ -396,11 +401,6 @@ private QualifiedName qualifyName(QualifiedName table)
     {
         List<String> tableParts = table.getParts();
 
-        // ignore temporary tables created by WITH clause as they can have various table parts
-        if (temporaryTables.contains(table)) {
-            return table;
-        }
-
         return switch (tableParts.size()) {
             case 1 -> QualifiedName.of(defaultCatalog.orElseThrow(this::unsetDefaultExceptionSupplier), defaultSchema.orElseThrow(this::unsetDefaultExceptionSupplier), tableParts.getFirst());
             case 2 -> QualifiedName.of(defaultCatalog.orElseThrow(this::unsetDefaultExceptionSupplier), tableParts.getFirst(), tableParts.get(1));

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

@@ -422,8 +422,6 @@ uno as (SELECT c1 FROM dos)
 
         Set<QualifiedName> tablesWithDefaults = trinoQueryPropertiesWithDefaults.getTables();
         assertThat(tablesWithDefaults).containsExactlyInAnyOrder(
-                QualifiedName.of("uno"),
-                QualifiedName.of("dos"),
                 QualifiedName.of("cat", "schem", "tbl1")
         );
 
@@ -433,8 +431,6 @@ uno as (SELECT c1 FROM dos)
         TrinoQueryProperties trinoQueryPropertiesNoDefaults = new TrinoQueryProperties(mockRequestNoDefaults, requestAnalyzerConfig);
         Set<QualifiedName> tablesNoDefaults = trinoQueryPropertiesNoDefaults.getTables();
         assertThat(tablesNoDefaults).containsExactlyInAnyOrder(
-                QualifiedName.of("uno"),
-                QualifiedName.of("dos"),
                 QualifiedName.of("cat", "schem", "tbl1")
         );
     }