从JAVA中的sql语句获取表名和查询类型

时间:2015-06-01 13:23:55

标签: java jsqlparser

我是jsqlparser的新手并尝试解析sql语句以获取表名及其查询类型(在Java中)。

对于eg1。
INSERT INTO Customers (CustomerName, Country) SELECT SupplierName, Country FROM Suppliers WHERE Country='Germany'
在此查询中,我想获取表名Suppliers及其查询类型SELECT 同样,表名Suppliers及其查询类型INSERT

对于eg2。
UPDATE CUSTOMERS SET SALARY = SALARY * 0.25 WHERE AGE IN (SELECT AGE FROM CUSTOMERS_BKP WHERE AGE >= 27 )
在此查询中,我想获取表名CUSTOMERS_BKP及其查询类型SELECT 同样,表名CUSTOMERS及其查询类型UPDATE

对于eg3。
UPDATE CUSTOMERS SET SALARY = SALARY * 0.25 WHERE AGE IN (SELECT AGE FROM CUSTOMERS WHERE AGE >= 27 )
(我知道这个查询可以简化,但例如我已粘贴在这里。)
在此查询中,我想获取表名CUSTOMERS及其查询类型SELECT 同样,表名CUSTOMERS及其查询类型UPDATE
注意:此处的表名相同但查询类型不同。

同样,我想获取任何复杂sql查询的表名及其查询类型。

你能帮帮我吗?

如果您希望我提供任何详细信息,请发表评论。

2 个答案:

答案 0 :(得分:0)

您提交的问题很棘手,因为JSqlParsers API缺少一些扩展点。为了纠正这个问题,我在github上创建了一些问题。所以这里有一个小样本来运行。输出是:

eg1
INSERT - Customers
SELECT - [Suppliers]
eg2
UPDATE - [CUSTOMERS]
SELECT - [CUSTOMERS_BKP]
eg3
UPDATE - [CUSTOMERS]
SELECT - [CUSTOMERS]

这里的主要工具是 JSqlParser TablesNamesFinder ,它获取了某些SQL的所有表名。现在我们必须修改它,以单独的方式获取所有基于select语句的表名。这是使用 TablesNamesFinderExt 完成的,它也可以使用插入语句更正子选择问题。

简而言之,查找程序识别子选择并设置全局标记 inSelect 以填充单独的 mySelectTableList ,以便后续调用访问(表...)

import java.util.ArrayList;
import java.util.List;
import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.schema.Table;
import net.sf.jsqlparser.statement.insert.Insert;
import net.sf.jsqlparser.statement.select.SubSelect;
import net.sf.jsqlparser.statement.update.Update;
import net.sf.jsqlparser.util.TablesNamesFinder;

public class SimpleSqlParserTableNames2 {

    public static void main(String args[]) throws JSQLParserException {
        System.out.println("eg1");
        Insert insert = (Insert)CCJSqlParserUtil.parse("INSERT INTO Customers (CustomerName, Country) SELECT SupplierName, Country FROM Suppliers WHERE Country='Germany'");
        TablesNamesFinderExt finder = new TablesNamesFinderExt();

        System.out.println("INSERT - " + insert.getTable());
        finder.getTableList(insert.getSelect());
        System.out.println("SELECT - " + finder.getSelectTableList());

        System.out.println("eg2");
        Update update = (Update)CCJSqlParserUtil.parse("UPDATE CUSTOMERS SET SALARY = SALARY * 0.25\n" +
                    "  WHERE AGE IN (SELECT AGE FROM CUSTOMERS_BKP WHERE AGE >= 27 )");
        finder = new TablesNamesFinderExt();
        System.out.println("UPDATE - " + update.getTables());
        finder.getTableList(update);
        System.out.println("SELECT - " + finder.getSelectTableList());

        System.out.println("eg3");
        update = (Update)CCJSqlParserUtil.parse("UPDATE CUSTOMERS SET SALARY = SALARY * 0.25\n" +
                    "WHERE AGE IN (SELECT AGE FROM CUSTOMERS WHERE AGE >= 27 )");
        finder = new TablesNamesFinderExt();
        System.out.println("UPDATE - " + update.getTables());
        finder.getTableList(update);
        System.out.println("SELECT - " + finder.getSelectTableList());
    }

    static class TablesNamesFinderExt extends TablesNamesFinder {
        List<String> mySelectTableList = new ArrayList<>();
        boolean inSelect = true;
        /**
         * To solve JSqlParsers Problem in getting tablenames from subselect using an Insert
         * statement.
         *
         * @param insert
         * @return
         */
        @Override
        public List<String> getTableList(Insert insert) {
            List<String> list = super.getTableList(insert);
            if (insert.getSelect() != null) {
                insert.getSelect().getSelectBody().accept(this);
            }
            return list;
        }

        @Override
        public void visit(SubSelect subSelect) {
            inSelect = true;
            super.visit(subSelect);
        }

        @Override
        public void visit(Table tableName) {
            super.visit(tableName); 
            if (inSelect && !mySelectTableList.contains(tableName.getFullyQualifiedName()))
                mySelectTableList.add(tableName.getFullyQualifiedName());
        }

        public List<String> getSelectTableList() {
            return mySelectTableList;
        }

    }
}

答案 1 :(得分:0)

如果你只需要表格形式的SQL,你需要一个超轻,超快的libray(完整的SQL解析器会被杀死)

只需在你的pom中添加以下内容

即可
<dependency>
    <groupId>com.github.mnadeem</groupId>
    <artifactId>sql-table-name-parser</artifactId>
    <version>0.0.1</version>
</dependency>

并使用以下说明

new TableNameParser(sql).tables()

有关详细信息,请参阅project

免责声明:我是所有者