利用 druid 的 sql parser 模块解析 sql 语句获得表名

druid 是阿里开源在 github 上面的数据库连接池，里面有一个专门解析 sql 语句的模块。

源码位置
SQL Parser 模块的介绍
相关 API

druid 使用说明：

druid 现在已经直接发布到了 maven 仓库中，可以直接引用。
1、新建一个 maven 工程并命名

2、在 pom.xml 文件中加入如下配置，当前最新使用 1.0.19

<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>druid</artifactId>
    <version>1.0.19</version>
</dependency>

3、示例代码，druid 的源码中有丰富的测试代码，具体路径如下
src/test/java/com.alibaba.druid.bvt.sql.mysql.visitor/
下面的 App.java 参考的是 MySqlSchemaStatVisitorTest_Delete.java 这个源文件

import java.util.List;

import com.alibaba.druid.sql.SQLUtils;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
import com.alibaba.druid.util.JdbcConstants;

/**
 *  获取表名
 */
public class App {
    public static void main(String[] args) {
        // String sql = "update t set name = 'x' where id < 100 limit 10";
        // String sql = "SELECT ID, NAME, AGE FROM USER WHERE ID = ? limit 2";
        // String sql = "select * from tablename limit 10";

        String sql = "select user from emp_table";
        String dbType = JdbcConstants.MYSQL;

        //格式化输出
        String result = SQLUtils.format(sql, dbType);
        System.out.println(result); // 缺省大写格式
        List<SQLStatement> stmtList = SQLUtils.parseStatements(sql, dbType);

        //解析出的独立语句的个数
        System.out.println("size is:" + stmtList.size());
        for (int i = 0; i < stmtList.size(); i++) {

            SQLStatement stmt = stmtList.get(i);
            MySqlSchemaStatVisitor visitor = new MySqlSchemaStatVisitor();
            stmt.accept(visitor);

            //获取表名称
            System.out.println("Tables : " + visitor.getCurrentTable());
            //获取操作方法名称,依赖于表名称
            System.out.println("Manipulation : " + visitor.getTables());
            //获取字段名称
            System.out.println("fields : " + visitor.getColumns());
        }
    }
}

4、结果

SELECT user
FROM emp_table
size is:1
Tables : emp_table
Manipulation : {emp_table=Select}
fields : [emp_table.user]