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

猜你喜欢

转载自blog.csdn.net/qq_41941497/article/details/131171534