MybatisPlus的多数据源及SQL拦截

MybatisPlus多数据源情况下的分页,MyBatisPlus下的SQL拦截增加替换语句

MybatisPlus多数据源配置

分页

添加mybatisPlus的PaginationInnerInterceptor 插件,并在插件中指定多个dbType,这里以MySQL和PostgreSQL为例。

	MybatisplusInterceptor interceptor = new MybatisPlusInterceptor();
	PaginationInnerInterceptor paginationInnerInterceptor = new PaginationInnerInterceptor(DbType.MYSQL);
	paginationInnerInterceptor.setDbType(DBTypeEnum.POSTGRE_SQL);
	interceptor,addInnerInterceptor(paginationInnerInterceptor);

dbType

dbType 是 MyBatis-Plus 中的一个配置,用于指定数据库类型,以便正确地执行不同数据库的 SQL 语句。MyBatis-Plus 中提供了对多个数据库类型的支持,包括 MYSQL、Oracle、PostgreSQL、SQLServer 等。

在 MyBatis-Plus 中,可以同时配置多个 dbType。假设我们的项目同时使用了 MYSQL 和 PostgreSQL,配置文件进行配置:

mybatis-plus.db-config.db-type=mysql,postgresql

MySQL和PostgreSQL分页的区别

MySQL中limit和对应的offset的写法

limit offset,row_count
limit 0,20

PostgreSQL中limit和对应的offset的写法

LIMIT row_count OFFSET offset
limit 20 offset 0

在MySQL中,OFFSET和LIMIT一般一起使用,以返回结果集中的一部分。在postgreSQL中,LIMIT子句也可以作为独立的语句使用,以限制返回的行数,而不指定偏移量。

拦截MybatisPlus的SQL,对SQL进行修改

在使用MybatisPlus作为ORM框架操作数据库的项目中,如果项目需要实现数据权限的功能时,可以通过配置相关的拦截器来增强 MyBatisPlus 的功能,从而对 MyBatisPlus 生成的 SQL 语句进行替换和增强,甚至可以实现自己的 SQL 语句。

通过 AOP 的方式拦截com.baomidou.mybatisplus.core.mapper.BaseMapper中的所有方法,然后获取方法的参数,判断是否包含 MyBatisPlus 的 QueryWrapper 对象,如果有的话就获取其中的 WHERE 条件,并根据自己的需求进行 SQL 语句的替换和增强,然后设置替换后的 WHERE 条件。可以方便地替换和增强 MyBatisPlus 生成的 SQL 语句,同时保持原有的 MyBatisPlus 功能不变。

@Aspect
@Component
public class SqlReplaceAspect {

    @Pointcut("execution(* com.baomidou.mybatisplus.core.mapper.BaseMapper.*(..))")
    public void pointcut() {}

    @Around("pointcut()")
    public Object replaceSql(ProceedingJoinPoint point) throws Throwable {
        // 获取方法参数
        Object[] args = point.getArgs();
        // 获取 MyBatis-Plus 的 QueryWrapper 对象
        if (args != null && args.length > 0) {
            for (Object arg : args) {
                if (arg instanceof QueryWrapper) {
                    QueryWrapper<?> wrapper = (QueryWrapper<?>) arg;
                    // 获取 WHERE 条件
                    String str = wrapper.getSqlSegment();
                    // 替换 SQL 语句
                    String newStr = str.replaceAll("and", "AND").replaceAll("or", "OR");
                    // 设置替换后的 SQL 语句
                    wrapper.setSqlSegment(newStr);
                    break;
                }
            }
        }

        return point.proceed();
    }
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值