Class SqlParser

src/main/java/com/sql/parse/util/SqlParser.java:34–378

@date 2019年4月30日上午09:58:02 @desc sql解析器

 * @desc sql解析器
 */
public class SqlParser {
    private Statement statement;

    private String type;

    public SqlParser(String sql) {
        String normalizedSql = sql.toUpperCase().trim();
        if (normalizedSql.indexOf("SELECT") == 0) {
            statement = com.sql.parse.statement.Select.parse(sql);
            type = "SELECT";
        } else if (normalizedSql.indexOf("INSERT") == 0) {
            statement = com.sql.parse.statement.Insert.parse(sql);
            type = "INSERT";
        } else if (normalizedSql.indexOf("UPDATE") == 0) {
            statement = com.sql.parse.statement.Update.parse(sql);
            type = "UPDATE";
        } else if (normalizedSql.indexOf("DELETE") == 0) {
            statement = com.sql.parse.statement.Delete.parse(sql);
            type = "DELETE";
        } else {
            throw BaseException.errorCode(SqlParseConstant.CODE_010);
        }
    }

    public String getType() {
        return type;
    }

    public Statement getStatement() {
        return statement;
    }

    /**
     * @title: getTableNames
     * @desc 获取表名
     */
    public List<String> getTableNames() {
        switch (SqlParseConfig.getHandle()) {
            case "com.github.jsqlparser":{
                TablesNamesFinder tablesNamesFinder = new TablesNamesFinder();
                if ("SELECT".equals(type)) {
                    Select select = (Select) statement;
                    return tablesNamesFinder.getTableList(select.getSelect());
                } else if ("INSERT".equals(type)) {
                    Insert insert = (Insert) statement;
                    return tablesNamesFinder.getTableList(insert.getInsert());
                } else if ("UPDATE".equals(type)) {
                    Update update = (Update) statement;
                    return tablesNamesFinder.getTableList(update.getUpdate());
                } else if ("DELETE".equals(type)) {
                    Delete delete = (Delete) statement;
                    return tablesNamesFinder.getTableList(delete.getDelete());
                } else {
                    throw BaseException.errorCode(SqlParseConstant.CODE_011);
                }
            }
            case "com.alibaba.druid":{
                List<String> list = new ArrayList<>();