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

相关推荐
色空大师11 分钟前
23种设计模式
java·开发语言·设计模式
闲人一枚(学习中)12 分钟前
设计模式-创建型-建造者模式
java·设计模式·建造者模式
2202_7544215429 分钟前
生成MPSOC以及ZYNQ的启动文件BOOT.BIN的小软件
java·linux·开发语言
蓝染-惣右介32 分钟前
【MyBatisPlus·最新教程】包含多个改造案例,常用注解、条件构造器、代码生成、静态工具、类型处理器、分页插件、自动填充字段
java·数据库·tomcat·mybatis
小林想被监督学习33 分钟前
idea怎么打开两个窗口,运行两个项目
java·ide·intellij-idea
HoneyMoose34 分钟前
IDEA 2024.3 版本更新主要功能介绍
java·ide·intellij-idea
我只会发热36 分钟前
Java SE 与 Java EE:基础与进阶的探索之旅
java·开发语言·java-ee
是老余37 分钟前
本地可运行,jar包运行错误【解决实例】:通过IDEA的maven package打包多模块项目
java·maven·intellij-idea·jar
crazy_wsp38 分钟前
IDEA怎么定位java类所用maven依赖版本及引用位置
java·maven·intellij-idea
.Ayang40 分钟前
tomcat 后台部署 war 包 getshell
java·计算机网络·安全·web安全·网络安全·tomcat·网络攻击模型