mybatisplus 分页重写count方法

MyBatis-Plus是一个基于MyBatis的增强工具，提供了很多便捷的功能，其中包括分页查询。在使用MyBatis-Plus进行分页查询时，可以通过重写count方法来实现自定义的分页查询逻辑。

要重写count方法，需要创建一个自定义的分页拦截器，并在该拦截器中实现count方法的重写。以下是一个示例：

public class CustomPaginationInterceptor extends PaginationInterceptor {
    
    @Override
    public boolean optimizeCountSql() {
        return false; // 关闭内置的 count 查询
    }
    
    @Override
    public List<ISqlParser> getSqlParserList() {
        List<ISqlParser> sqlParserList = super.getSqlParserList();
        sqlParserList.add(new CustomCountSqlParser()); // 添加自定义的 count 查询解析器
        return sqlParserList;
    }
    
    public static class CustomCountSqlParser extends AbstractJsqlParser {
        
        @Override
        public SqlInfo parser(MetaObject metaObject, String sql) {
            MappedStatement mappedStatement = PluginUtils.getMappedStatement(metaObject);
            if (SqlCommandType.SELECT == mappedStatement.getSqlCommandType()) {
                // 判断是否为 count 查询
                if (sql.trim().toLowerCase().matches("^select\\s+count\\(.+")) {
                    // 自定义 count 查询逻辑
                    String newSql = "SELECT COUNT(1) FROM (" + sql + ") tmp";
                    return SqlInfo.newInstance().setSql(newSql);
                }
            }
            return null;
        }
    }
}

在上述示例中，我们创建了一个自定义的分页拦截器`CustomPaginationInterceptor`，并重写了`optimizeCountSql`方法，将内置的count查询关闭。然后，我们添加了一个自定义的count查询解析器`CustomCountSqlParser`，在该解析器中实现了自定义的count查询逻辑。

使用自定义的分页拦截器后，可以在MyBatis-Plus的分页查询中使用自定义的count查询逻辑。