java:使用JSqlParser给sql语句增加tenant_id和deleted条件

# 示例代码

【pom.xml】

xml 复制代码
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-core</artifactId>
    <version>3.4.3.1</version>
</dependency>

【MyJSqlParserTest.java】

java 复制代码
package com.chz.myJSqlParser;

public class MyJSqlParserTest {

    public static void main(String[] args) throws JSQLParserException
    {
        Select stmt = (Select) CCJSqlParserUtil.parse(" SELECT col1 AS a, col2 AS b" +
                " FROM table_name " +
                " WHERE col_1 = 10" +
                " AND col_2 = 20");

        System.out.println("before ::: " + stmt.toString());

        PlainSelect plainSelect = (PlainSelect)stmt.getSelectBody();
        Expression whereExpression = plainSelect.getWhere();

        whereExpression = andTenantIdExpression(whereExpression, 1L);
        whereExpression = andDeletedExpression(whereExpression, true);

        plainSelect.setWhere(whereExpression);
        System.out.println("after ::: " + stmt);
    }

    private static BinaryExpression andTenantIdExpression(Expression where, Long tenantId) {
        EqualsTo equalsTo = new EqualsTo();
        equalsTo.setLeftExpression(new Column("tenant_id"));
        equalsTo.setRightExpression(new LongValue(tenantId));
        if (null != where) {
            if (where instanceof OrExpression) {
                return new AndExpression(equalsTo, new Parenthesis(where));
            } else {
                return new AndExpression(equalsTo, where);
            }
        }
        return equalsTo;
    }

    private static Expression andDeletedExpression(Expression where, Boolean deleted)
    {
        IsBooleanExpression isBooleanExpression = new IsBooleanExpression();
        isBooleanExpression.setLeftExpression(new Column("deleted"));
        isBooleanExpression.setIsTrue(deleted);
        if (null != where) {
            if (where instanceof OrExpression) {
                return new AndExpression(isBooleanExpression, new Parenthesis(where));
            } else {
                return new AndExpression(isBooleanExpression, where);
            }
        }
        return isBooleanExpression;
    }
}

运行【MyJSqlParserTest】

# 其它参考资料:

https://juejin.cn/post/7072190387342802981

相关推荐
七星静香24 分钟前
laravel chunkById 分块查询 使用时的问题
java·前端·laravel
Jacob程序员25 分钟前
java导出word文件(手绘)
java·开发语言·word
ZHOUPUYU25 分钟前
IntelliJ IDEA超详细下载安装教程(附安装包)
java·ide·intellij-idea
stewie628 分钟前
在IDEA中使用Git
java·git
Elaine20239144 分钟前
06 网络编程基础
java·网络
G丶AEOM1 小时前
分布式——BASE理论
java·分布式·八股
落落鱼20131 小时前
tp接口 入口文件 500 错误原因
java·开发语言
想要打 Acm 的小周同学呀1 小时前
LRU缓存算法
java·算法·缓存
镰刀出海1 小时前
Recyclerview缓存原理
java·开发语言·缓存·recyclerview·android面试
阿伟*rui3 小时前
配置管理,雪崩问题分析,sentinel的使用
java·spring boot·sentinel