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();
}
}