【组件】JAVA解析SQL语句jsqlparser

穆熙沐

 关注

阅读 74

2022-12-13


jsqlparser

依赖maven

<!-- https://mvnrepository.com/artifact/com.github.jsqlparser/jsqlparser -->
        <dependency>
            <groupId>com.github.jsqlparser</groupId>
            <artifactId>jsqlparser</artifactId>
            <version>4.3</version>
        </dependency>

public static void main(String[] args) throws Exception {


        String sql = """
                UPDATE
                 t1 q
                 SET q.name  = '张三'
                """;
//        sql = """
//DELETE t1 FROM t1,t2 WHERE t1.id=t2.id
//               """;

        Statement parse = CCJSqlParserUtil.parse(sql);
        //字典
        Dict tableMap = new Dict();
        Dict tableFullMap = new Dict();

        Table table = null;
        FromItem fromItem = null;
        List<Join> joins = null;
        List<Table> usingList = null;
        Expression where = null;
        if(parse instanceof Insert){
            log.info("增");
            Insert insert = (Insert) parse;
            table = insert.getTable();

        }else if(parse instanceof Delete){
            log.info("删");
            Delete delete = (Delete) parse;
            table = delete.getTable();
            joins = delete.getJoins();
            usingList = delete.getUsingList();
            where = delete.getWhere();

        }else if(parse instanceof Update){
            log.info("改");
            Update update = (Update) parse;
            table = update.getTable();
            joins = update.getJoins();
            fromItem = update.getFromItem();
            where = update.getWhere();

        }else if(parse instanceof Select){
            log.info("查");

        }


        if(table != null){
            log.info("表名:{}", table.getName());
            if(table.getAlias() != null){
                tableMap.set(table.getAlias().getName(), table.getName());
            }else{
                tableMap.set(table.getName(), table.getName());
            }
            tableFullMap.set(table.getName(), table);
        }

        if(fromItem != null){
            Table fromItemTable = (Table) fromItem; // table.getAlias().getName(), table.getName()
            log.info("fromItem表名:{}", fromItemTable.getName());
            if(fromItemTable.getAlias() != null){
                tableMap.set(fromItemTable.getAlias().getName(), fromItemTable.getName());
            }else{
                tableMap.set(fromItemTable.getName(), fromItemTable.getName());
            }
            tableFullMap.set(fromItemTable.getName(), fromItemTable);
        }

        /**
         * 联查
         */
        if(CollUtil.isNotEmpty(joins)){
            for (Join join : joins) {
                FromItem rightItem = join.getRightItem();
                Table table2 = (Table) rightItem; // (table.getAlias().getName(), table.getName());
                log.info("联查表:{}", table2.getName());
                if(table2.getAlias() != null){
                    tableMap.set(table2.getAlias().getName(), table2.getName());
                }else{
                    tableMap.set(table2.getName(), table2.getName());
                }
                tableFullMap.set(table2.getName(), table2);

                // on 条件分析   //和where一样
                Collection<Expression> onExpressions = join.getOnExpressions();
                for (Expression expression : onExpressions) {
//                EqualsTo equalsTo = (EqualsTo) expression;
                    if (expression instanceof Column) {
                        Column rightColumn = (Column) expression;
                        log.info("联查条件 ON,表名:{}, 字段名:{}, 关联条件:{}", tableMap.get(rightColumn.getTable().toString()), rightColumn.getColumnName());
                    }
                }
            }
        }


        // where 分析
        // where 只有两种情况，一种就是简单的a = b，另外就是 a = b or ... and， 这种就是AndExpression
        if(where != null){
            if (where instanceof EqualsTo) {
                EqualsTo equalsTo = (EqualsTo) where;
                Expression rightExpression = equalsTo.getRightExpression();
                Expression leftExpression = equalsTo.getLeftExpression();
                if (rightExpression instanceof Column) {
                    Column rightColumn = (Column) rightExpression;
                    log.info("条件 RIGHT,表名:{}, 字段名:{}, 值:{}", tableMap.get(rightColumn.getTable().toString()), rightColumn.getColumnName(), equalsTo.getRightExpression());
                }
                if (leftExpression instanceof Column) {
                    Column leftColumn = (Column) leftExpression;
                    log.info("条件 LEFT,表名:{}, 字段名:{}, 值:{}", tableMap.get(leftColumn.getTable().toString()), leftColumn.getColumnName(), equalsTo.getRightExpression());
                }
            }else if(where instanceof AndExpression){
                AndExpression andExpression = (AndExpression) where;
                Expression leftExpression = andExpression.getLeftExpression();
//            doExpression(leftExpression);
                Expression rightExpression = andExpression.getRightExpression();
//            doExpression(rightExpression);
            }
        }

        /**
         * usingList
         */
        if(CollUtil.isNotEmpty(usingList)){
            for (Table table2 : usingList) {

                if(table2.getAlias() != null){
                    tableMap.set(table2.getAlias().getName(), table2.getName());
                }else{
                    tableMap.set(table2.getName(), table2.getName());
                }
                log.info("using表名:{}", table2.getName());

            }
        }

//        System.out.println(parse);
//
//        // 使用工具类把SQL转换为Select对象
//        Select select = (Select) CCJSqlParserUtil.parse("""
//                 SELECT
//                                       *
//                                        FROM
//                                        user u
//                """);
//
//        SelectBody selectBody = select.getSelectBody();
//        PlainSelect plainSelect = (PlainSelect) selectBody;


    }

工具

import cn.hutool.core.collection.CollUtil;
import cn.hutool.core.lang.Dict;
import lombok.extern.slf4j.Slf4j;
import net.sf.jsqlparser.JSQLParserException;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.parser.CCJSqlParserUtil;
import net.sf.jsqlparser.schema.Table;
import net.sf.jsqlparser.statement.Statement;
import net.sf.jsqlparser.statement.delete.Delete;
import net.sf.jsqlparser.statement.insert.Insert;
import net.sf.jsqlparser.statement.select.FromItem;
import net.sf.jsqlparser.statement.select.Join;
import net.sf.jsqlparser.statement.select.Select;
import net.sf.jsqlparser.statement.update.Update;

import java.util.HashSet;
import java.util.List;
import java.util.Set;

@Slf4j
public class SqlUtil {


    /**
     * 通过sql 解析出涉及到的表
     * @param sql
     * @return
     * @throws Exception
     */
    public static Set<String> getSqlTables(String sql) {


//        String sql = """
//                UPDATE
//                 user u
//                 SET u.name  = '张三'
//                """;
//        sql = """
//DELETE t1 FROM t1,t2 WHERE t1.id=t2.id
//               """;

        Set<String> tables = new HashSet<>();

        Statement parse = null;
        try {
            parse = CCJSqlParserUtil.parse(sql);
        } catch (JSQLParserException e) {
            e.printStackTrace();
            return tables;
        }
        //字典
        Dict tableMap = new Dict();
        Dict tableFullMap = new Dict();

        Table table = null;
        FromItem fromItem = null;
        List<Join> joins = null;
        List<Table> usingList = null;
        Expression where = null;
        if(parse instanceof Insert){
            log.info("增");
            Insert insert = (Insert) parse;
            table = insert.getTable();

        }else if(parse instanceof Delete){
            log.info("删");
            Delete delete = (Delete) parse;
            table = delete.getTable();
            joins = delete.getJoins();
            usingList = delete.getUsingList();
            where = delete.getWhere();

        }else if(parse instanceof Update){
            log.info("改");
            Update update = (Update) parse;
            table = update.getTable();
            joins = update.getJoins();
            fromItem = update.getFromItem();
            where = update.getWhere();

        }else if(parse instanceof Select){
//            log.info("查");

        }


        if(table != null){
//            log.info("表名:{}", table.getName());
//            if(table.getAlias() != null){
//                tableMap.set(table.getAlias().getName(), table.getName());
//            }else{
//                tableMap.set(table.getName(), table.getName());
//            }
//            tableFullMap.set(table.getName(), table);
            tables.add(table.getName());
        }

        if(fromItem != null){
            Table fromItemTable = (Table) fromItem; // table.getAlias().getName(), table.getName()
//            log.info("fromItem表名:{}", fromItemTable.getName());
//            if(fromItemTable.getAlias() != null){
//                tableMap.set(fromItemTable.getAlias().getName(), fromItemTable.getName());
//            }else{
//                tableMap.set(fromItemTable.getName(), fromItemTable.getName());
//            }
//            tableFullMap.set(fromItemTable.getName(), fromItemTable);
            tables.add(fromItemTable.getName());
        }

        /**
         * 联查
         */
        if(CollUtil.isNotEmpty(joins)){
            for (Join join : joins) {
                FromItem rightItem = join.getRightItem();
                Table table2 = (Table) rightItem; // (table.getAlias().getName(), table.getName());
//                log.info("联查表:{}", table2.getName());
//                if(table2.getAlias() != null){
//                    tableMap.set(table2.getAlias().getName(), table2.getName());
//                }else{
//                    tableMap.set(table2.getName(), table2.getName());
//                }
//                tableFullMap.set(table2.getName(), table2);
                tables.add(table2.getName());

                // on 条件分析   //和where一样
//                Collection<Expression> onExpressions = join.getOnExpressions();
//                for (Expression expression : onExpressions) {
                EqualsTo equalsTo = (EqualsTo) expression;
//                    if (expression instanceof Column) {
//                        Column rightColumn = (Column) expression;
//                        log.info("联查条件 ON,表名:{}, 字段名:{}, 关联条件:{}", tableMap.get(rightColumn.getTable().toString()), rightColumn.getColumnName());
//                    }
//                }
            }
        }


        // where 分析
        // where 只有两种情况，一种就是简单的a = b，另外就是 a = b or ... and， 这种就是AndExpression
//        if(where != null){
//            if (where instanceof EqualsTo) {
//                EqualsTo equalsTo = (EqualsTo) where;
//                Expression rightExpression = equalsTo.getRightExpression();
//                Expression leftExpression = equalsTo.getLeftExpression();
//                if (rightExpression instanceof Column) {
//                    Column rightColumn = (Column) rightExpression;
//                    log.info("条件 RIGHT,表名:{}, 字段名:{}, 值:{}", tableMap.get(rightColumn.getTable().toString()), rightColumn.getColumnName(), equalsTo.getRightExpression());
//                }
//                if (leftExpression instanceof Column) {
//                    Column leftColumn = (Column) leftExpression;
//                    log.info("条件 LEFT,表名:{}, 字段名:{}, 值:{}", tableMap.get(leftColumn.getTable().toString()), leftColumn.getColumnName(), equalsTo.getRightExpression());
//                }
//            }else if(where instanceof AndExpression){
//                AndExpression andExpression = (AndExpression) where;
//                Expression leftExpression = andExpression.getLeftExpression();
            doExpression(leftExpression);
//                Expression rightExpression = andExpression.getRightExpression();
            doExpression(rightExpression);
//            }
//        }

        /**
         * usingList
         */
        if(CollUtil.isNotEmpty(usingList)){
            for (Table table2 : usingList) {

//                if(table2.getAlias() != null){
//                    tableMap.set(table2.getAlias().getName(), table2.getName());
//                }else{
//                    tableMap.set(table2.getName(), table2.getName());
//                }
//                log.info("using表名:{}", table2.getName());
                tables.add(table2.getName());

            }
        }

//        System.out.println(parse);
//
//        // 使用工具类把SQL转换为Select对象
//        Select select = (Select) CCJSqlParserUtil.parse("""
//                 SELECT
//                                       *
//                                        FROM
//                                        user u
//                """);
//
//        SelectBody selectBody = select.getSelectBody();
//        PlainSelect plainSelect = (PlainSelect) selectBody;

        return tables;
    }

}

相关推荐

JSQLPARSER解析SQL知识入门

回望这一段人生 54 0 0

JSqlParser生成修改表定义SQL语句

梦幻之云 68 0 0

JsqlParser插件格式化SQL语句

小时候是个乖乖 63 0 0

【sql】sql中with as 介绍与使用jsqlparser解析sql

醉东枫 14 0 0

sql语句解析

舟海君 106 0 0

【Mysql系列】jsqlparser学习-SQL转换为java类

仲秋花似锦 22 0 0

Sql语句解析工具类

野见 22 0 0

SQL 语句解析过程详解

49路末班车 52 0 0

SQL语句解析类型---硬解析和软解析

耶也夜 74 0 0

SQL语句解析(alibaba-druid&metadata)

古月无语 44 0 0

精彩评论（0）

0 0 举报