Antlr SQL Parser 获得表名

要获得 SQL 查询语句中的表名，可以使用 ANTLR4 SQL Parser 提供的访问器（Visitor）模式来遍历 AST（抽象语法树）。以下是一个示例代码，演示如何从 SELECT 查询语句中获取表名：

public class TableNameVisitor extends SQLBaseVisitor<Void> {
    private String tableName;

    @Override
    public Void visitTableName(SQLParser.TableNameContext ctx) {
        tableName = ctx.getText();
        return null;
    }

    public String getTableName() {
        return tableName;
    }
}

// 创建 ANTLR 输入流
ANTLRInputStream input = new ANTLRInputStream("SELECT * FROM mytable");

// 创建 SQLLexer
SQLLexer lexer = new SQLLexer(input);

// 创建 ANTLR 编译器
CommonTokenStream tokens = new CommonTokenStream(lexer);
SQLParser parser = new SQLParser(tokens);

// 解析 SQL 查询语句并生成 AST
SelectStatementContext selectStatement = parser.selectStatement();

// 访问 AST 并获取表名
TableNameVisitor visitor = new TableNameVisitor();
visitor.visit(selectStatement);
String tableName = visitor.getTableName();

在上面的代码中，我们创建了一个名为 `TableNameVisitor` 的访问器，它继承自 `SQLBaseVisitor`，并重写了 `visitTableName` 方法来获取表名。在代码的最后，我们创建了一个 `TableNameVisitor` 实例，并调用 `visit` 方法遍历 AST。最后，我们可以通过 `getTableName` 方法获取到表名。

需要注意的是，上面的代码只适用于 SELECT 查询语句。如果要处理其他类型的查询语句，需要根据语法规则进行相应的修改。