fix: TableNameFinder for Alter Table

- fixes #2062

Signed-off-by: Andreas Reichel <[email protected]>
Signed-off-by: manticore-projects <[email protected]>

Author: manticore-projects

src/main/java/net/sf/jsqlparser/util/TablesNamesFinder.java

@@ -383,8 +383,8 @@ protected String extractTableName(Table table) {
     }
 
     @Override
-    public <S> Void visit(Table tableName, S context) {
-        String tableWholeName = extractTableName(tableName);
+    public <S> Void visit(Table table, S context) {
+        String tableWholeName = extractTableName(table);
         if (!otherItemNames.contains(tableWholeName)) {
             tables.add(tableWholeName);
         }
@@ -1149,13 +1149,12 @@ public void visit(CreateView createView) {
 
     @Override
     public <S> Void visit(Alter alter, S context) {
-        throwUnsupported(alter);
-        return null;
+        return alter.getTable().accept(this, context);
     }
 
     @Override
     public void visit(Alter alter) {
-        StatementVisitor.super.visit(alter);
+        alter.getTable().accept(this, null);
     }
 
     @Override

src/test/java/net/sf/jsqlparser/util/TablesNamesFinderTest.java

@@ -539,19 +539,19 @@ void testTableRenamingIssue2028() throws JSQLParserException {
                 ";";
         //@formatter:on
 
-        TablesNamesFinder finder = new TablesNamesFinder<Void>() {
+        TablesNamesFinder<Void> finder = new TablesNamesFinder<>() {
             @Override
-            public <S> Void visit(Table tableName, S context) {
-                String schemaName = tableName.getSchemaName();
+            public <S> Void visit(Table table, S context) {
+                String schemaName = table.getSchemaName();
                 if (schemaName != null && IGNORE_SCHEMAS.contains(schemaName.toLowerCase())) {
-                    return super.visit(tableName, context);
+                    return super.visit(table, context);
                 }
-                String originTableName = tableName.getName();
-                tableName.setName(prefix + originTableName);
+                String originTableName = table.getName();
+                table.setName(prefix + originTableName);
                 if (originTableName.startsWith("`")) {
-                    tableName.setName("`" + prefix + originTableName.replace("`", "") + "`");
+                    table.setName("`" + prefix + originTableName.replace("`", "") + "`");
                 }
-                return super.visit(tableName, context);
+                return super.visit(table, context);
             }
         };
         finder.init(false);
@@ -561,5 +561,17 @@ public <S> Void visit(Table tableName, S context) {
 
         TestUtils.assertStatementCanBeDeparsedAs(statement, expected, true);
     }
+
+    @Test
+    void testAlterTableIssue2062() throws JSQLParserException {
+        String sqlStr = "ALTER TABLE the_cool_db.the_table\n"
+                + "    ADD test VARCHAR (40)\n"
+                + ";";
+        Set<String> tables = TablesNamesFinder.findTablesOrOtherSources(sqlStr);
+        assertThat(tables).containsExactlyInAnyOrder("the_cool_db.the_table");
+
+        tables = TablesNamesFinder.findTables(sqlStr);
+        assertThat(tables).containsExactlyInAnyOrder("the_cool_db.the_table");
+    }
 }