手把手教你开发 MyBatis 插件

然后在 XML 文件中定义相关 SQL：

    select * from user


可以看到，在 SQL 定义时，压根不用管分页的事情，MyBatis 会查询到所有的数据，然后在内存中进行分页处理。
Mapper 中方法的调用方式如下：
@Test
public void test3() {
    UserMapper userMapper = sqlSessionFactory.openSession().getMapper(UserMapper.class);
    RowBounds rowBounds = new RowBounds(1,2);
    List list = userMapper.getAllUsersByPage(rowBounds);
    for (User user : list) {
 System.out.println("user = " + user);
    }
}

构建 RowBounds 时传入两个参数，分别是 offset 和 limit，对应分页 SQL 中的两个参数。也可以通过 RowBounds.DEFAULT 的方式构建一个 RowBounds 实例，这种方式构建出来的 RowBounds 实例，offset 为 0，limit 则为 Integer.MAX_VALUE，也就相当于不分页。
这就是 MyBatis 中提供的一个很不实用的内存分页功能。
了解了 MyBatis 自带的内存分页之后，接下来我们就可以来看看如何自定义分页插件了。
4.2 自定义分页插件
首先要声明一下，这里松哥带大家自定义 MyBatis 分页插件，主要是想通过这个东西让小伙伴们了解自定义 MyBatis 插件的一些条条框框，了解整个自定义插件的流程，分页插件并不是我们的目的，自定义分页插件只是为了让大家的学习过程变得有趣一些而已。
接下来我们就来开启自定义分页插件之旅。
首先我们需要自定义一个 RowBounds，因为 MyBatis 原生的 RowBounds 是内存分页，并且没有办法获取到总记录数（一般分页查询的时候我们还需要获取到总记录数），所以我们自定义 PageRowBounds，对原生的 RowBounds 功能进行增强，如下：
public class PageRowBounds extends RowBounds {
    private Long total;

    public PageRowBounds(int offset, int limit) {
 super(offset, limit);
    }

    public PageRowBounds() {
    }

    public Long getTotal() {
 return total;
    }

    public void setTotal(Long total) {
 this.total = total;
    }
}

可以看到，我们自定义的 PageRowBounds 中增加了 total 字段，用来保存查询的总记录数。
接下来我们自定义拦截器 PageInterceptor，如下：
@Intercepts(@Signature(
 type = Executor.class,
 method = "query",
 args = {MappedStatement.class, Object.class, RowBounds.class, ResultHandler.class}
))
public class PageInterceptor implements Interceptor {
    @Override
    public Object intercept(Invocation invocation) throws Throwable {
 Object[] args = invocation.getArgs();
 MappedStatement ms = (MappedStatement) args[0];
 Object parameterObject = args[1];
 RowBounds rowBounds = (RowBounds) args[2];
 if (rowBounds != RowBounds.DEFAULT) {
     Executor executor = (Executor) invocation.getTarget();
     BoundSql boundSql = ms.getBoundSql(parameterObject);
     Field additionalParametersField = BoundSql.class.getDeclaredField("additionalParameters");
     additionalParametersField.setAccessible(true);
     Map additionalParameters = (Map) additionalParametersField.get(boundSql);
     if (rowBounds instanceof PageRowBounds) {
  MappedStatement countMs = newMappedStatement(ms, Long.class);
  CacheKey countKey = executor.createCacheKey(countMs, parameterObject, RowBounds.DEFAULT, boundSql);
  String countSql = "select count(*) from (" + boundSql.getSql() + ") temp";
  BoundSql countBoundSql = new BoundSql(ms.getConfiguration(), countSql, boundSql.getParameterMappings(), parameterObject);
  Set keySet = additionalParameters.keySet();
  for (String key : keySet) {
      countBoundSql.setAdditionalParameter(key, additionalParameters.get(key));
  }
  List