mybatis-plus分页原理

不知道大家使用过程有没有发现，如果你的mybatis-plus执行数据库操作方法中有

new Page<>(page, size)对象参数，他就会自动进行分页操作。

是不是很神奇的存在，今天我们就针对其源码深入探讨一番。

。。。。。。。。。。。以下内容纯摘抄，不要喷我啊，我注明了转载和出处，旨在分享技术。。。。。。。。

第一步：引入依赖

引入依赖，不多逼逼

第二步：写配置bean

将分页插件注入

第三步：观察实现

首先是sql的拦截方法

@SuppressWarnings("unchecked")
@Override
public Object intercept(Invocation invocation) throws Throwable {
StatementHandler statementHandler = PluginUtils.realTarget(invocation.getTarget());
MetaObject metaObject = SystemMetaObject.forObject(statementHandler);

// SQL 解析
this.sqlParser(metaObject);

// 先判断是不是SELECT操作 (2019-04-10 00:37:31 跳过存储过程)
MappedStatement mappedStatement = (MappedStatement) metaObject.getValue("delegate.mappedStatement");
if (SqlCommandType.SELECT != mappedStatement.getSqlCommandType()
|| StatementType.CALLABLE == mappedStatement.getStatementType()) {
return invocation.proceed();
}

// 针对定义了rowBounds，做为mapper接口方法的参数
BoundSql boundSql = (BoundSql) metaObject.getValue("delegate.boundSql");
Object paramObj = boundSql.getParameterObject();

// 判断参数里是否有page对象
IPage<?> page = null;
if (paramObj instanceof IPage) {
page = (IPage<?>) paramObj;
} else if (paramObj instanceof Map) {
for (Object arg : ((Map<?, ?>) paramObj).values()) {
if (arg instanceof IPage) {
page = (IPage<?>) arg;
break;
}
}
}

/*
* 不需要分页的场合，如果 size 小于 0 返回结果集
*/
if (null == page || page.getSize() < 0) {
return invocation.proceed();
}

if (this.limit > 0 && this.limit <= page.getSize()) {
//处理单页条数限制
handlerLimit(page);
}

String originalSql = boundSql.getSql();
Connection connection = (Connection) invocation.getArgs()[0];
DbType dbType = StringUtils.isNotBlank(dialectType) ? DbType.getDbType(dialectType)
: JdbcUtils.getDbType(connection.getMetaData().getURL());

if (page.isSearchCount()) {
SqlInfo sqlInfo = SqlParserUtils.getOptimizeCountSql(page.optimizeCountSql(), countSqlParser, originalSql);
this.queryTotal(sqlInfo.getSql(), mappedStatement, boundSql, page, connection);
if (page.getTotal() <= 0) {
return null;
}
}

String buildSql = concatOrderBy(originalSql, page);
DialectModel model = DialectFactory.buildPaginationSql(page, buildSql, dbType, dialectClazz);
Configuration configuration = mappedStatement.getConfiguration();
List<ParameterMapping> mappings = new ArrayList<>(boundSql.getParameterMappings());
Map<String, Object> additionalParameters = (Map<String, Object>) metaObject.getValue("delegate.boundSql.additionalParameters");
model.consumers(mappings, configuration, additionalParameters);
metaObject.setValue("delegate.boundSql.sql", model.getDialectSql());
metaObject.setValue("delegate.boundSql.parameterMappings", mappings);
return invocation.proceed();
}

首先是在Sql解析器中对sql进行解析和优化，然后其中这一步对分页做了判断

mybatis-plus的默认pageSize是500，随后取出sql，建立数据库连接

随后

点进去可以发现

主要是针对count进行了优化，在Page对象中可以发现 optimizeCountSql这个bool默认是一直true的，所以一般默认的优化措施是在 JsqlParserCountOptimize 这个对象中，对sql优化完成后，进入到queryTotal这个方法，开始了一次count，并将其赋值到page对象：