MybatisPlus数据权限插件遇到PageHelper失效的解决办法

pota10

已于 2025-04-28 14:43:42 修改

阅读量433

点赞数 12

文章标签： java 数据库 mysql mybatis

于 2025-04-28 14:40:35 首次发布

本文链接：https://blog.csdn.net/qq_53692503/article/details/147581186

版权

问题描述

多租户系统里基于MybatisPlus提供的数据权限拦截器DataPermissionInterceptor实现了用户数据权限的隔离。但遇到使用了PageHelper的分页接口时，出现total和data的条数不一致的情况。查看日志发现，似乎在执行对应mapper方法之前，先执行了一条COUNT语句来计算total，而这条sql不会被拦截。

解决办法

PageHelper默认情况下会帮我们根据查询语句自动生成COUNT查询SQL。当我们在Mapper方法上添加数据权限注解时，由于生成的COUNT与分页查询SQL的MappedStatement.id不一致，拦截器会认为这条COUNT并没有被注解，因此直接放行了。

幸好，这条COUNT的id只是在原来的基础上加上“_COUNT”后缀，因此我们可以在DataPermissionInterceptor的BeforeQuery方法中做一个简单处理：

    @Override
    public void beforeQuery(Executor executor, MappedStatement ms, Object parameter, RowBounds rowBounds, ResultHandler resultHandler, BoundSql boundSql) throws SQLException {
        // 处理pageHelper产生的 _COUNT 后缀
        String mappedStatementId = ms.getId();
        if (mappedStatementId.endsWith("_COUNT")) {
            mappedStatementId = mappedStatementId.substring(0, mappedStatementId.length() - "_COUNT".length());
        }