admin/mybatis-flex
1
0
Fork 0
mirror of https://gitee.com/mybatis-flex/mybatis-flex.git synced 2025-12-08 09:38:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

optimize paginate method

Browse Source
This commit is contained in:
开源海哥 2023-05-22 08:07:33 +08:00
parent 5c47927241
commit e45df191df
2 changed files with 31 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
mybatis-flex-core/src/main/java/com/mybatisflex/core/BaseMapper.java
View File

@ -575,16 +575,25 @@ public interface BaseMapper<T> {
default <R> Page<R> paginateAs(Page<R> page, QueryWrapper queryWrapper, Class<R> asType) { default <R> Page<R> paginateAs(Page<R> page, QueryWrapper queryWrapper, Class<R> asType) {
List<QueryColumn> selectColumns = CPI.getSelectColumns(queryWrapper); List<QueryColumn> selectColumns = CPI.getSelectColumns(queryWrapper);
List<QueryOrderBy> orderBys = CPI.getOrderBys(queryWrapper);
List<Join> joins = CPI.getJoins(queryWrapper); List<Join> joins = CPI.getJoins(queryWrapper);
boolean removedJoins = true; boolean removedJoins = true;
// 只有 totalRow 小于 0 的时候才会去查询总量 // 只有 totalRow 小于 0 的时候才会去查询总量
// 这样方便用户做总数缓存，而非每次都要去查询总量 // 这样方便用户做总数缓存，而非每次都要去查询总量
// 一般的分页场景中，只有第一页的时候有必要去查询总量，第二页以后是不需要的 // 一般的分页场景中，只有第一页的时候有必要去查询总量，第二页以后是不需要的
if (page.getTotalRow() < 0) { if (page.getTotalRow() < 0) {
//移除 select
CPI.setSelectColumns(queryWrapper, Collections.singletonList(count().as("total"))); CPI.setSelectColumns(queryWrapper, Collections.singletonList(count().as("total")));
//移除 OrderBy
if (CollectionUtil.isNotEmpty(orderBys)) {
CPI.setOrderBys(queryWrapper, null);
}
//移除 left join
if (joins != null && !joins.isEmpty()) { if (joins != null && !joins.isEmpty()) {
for (Join join : joins) { for (Join join : joins) {
if (!Join.TYPE_LEFT.equals(CPI.getJoinType(join))) { if (!Join.TYPE_LEFT.equals(CPI.getJoinType(join))) {
@ -615,6 +624,7 @@ public interface BaseMapper<T> {
CPI.setJoins(queryWrapper, null); CPI.setJoins(queryWrapper, null);
} }
long count = selectCountByQuery(queryWrapper); long count = selectCountByQuery(queryWrapper);
page.setTotalRow(count); page.setTotalRow(count);
} }
@ -626,6 +636,11 @@ public interface BaseMapper<T> {
//重置 selectColumns //重置 selectColumns
CPI.setSelectColumns(queryWrapper, selectColumns); CPI.setSelectColumns(queryWrapper, selectColumns);
//重置 orderBys
if (CollectionUtil.isNotEmpty(orderBys)) {
CPI.setOrderBys(queryWrapper, orderBys);
}
//重置 join //重置 join
if (removedJoins) { if (removedJoins) {
CPI.setJoins(queryWrapper, joins); CPI.setJoins(queryWrapper, joins);

15
mybatis-flex-core/src/main/java/com/mybatisflex/core/row/RowMapper.java
View File

@ -404,6 +404,9 @@ public interface RowMapper {
CPI.setFromIfNecessary(queryWrapper, tableName); CPI.setFromIfNecessary(queryWrapper, tableName);
List<QueryColumn> selectColumns = CPI.getSelectColumns(queryWrapper); List<QueryColumn> selectColumns = CPI.getSelectColumns(queryWrapper);
List<QueryOrderBy> orderBys = CPI.getOrderBys(queryWrapper);
List<Join> joins = CPI.getJoins(queryWrapper); List<Join> joins = CPI.getJoins(queryWrapper);
boolean removedJoins = true; boolean removedJoins = true;
@ -412,8 +415,15 @@ public interface RowMapper {
// 一般的分页场景中，只有第一页的时候有必要去查询总量，第二页以后是不需要的 // 一般的分页场景中，只有第一页的时候有必要去查询总量，第二页以后是不需要的
if (page.getTotalRow() < 0) { if (page.getTotalRow() < 0) {
//移除 seelct
CPI.setSelectColumns(queryWrapper, Collections.singletonList(count().as("total"))); CPI.setSelectColumns(queryWrapper, Collections.singletonList(count().as("total")));
//移除 OrderBy
if (CollectionUtil.isNotEmpty(orderBys)) {
CPI.setOrderBys(queryWrapper, null);
}
//移除 left join
if (joins != null && !joins.isEmpty()) { if (joins != null && !joins.isEmpty()) {
for (Join join : joins) { for (Join join : joins) {
if (!Join.TYPE_LEFT.equals(CPI.getJoinType(join))) { if (!Join.TYPE_LEFT.equals(CPI.getJoinType(join))) {
@ -455,6 +465,11 @@ public interface RowMapper {
//重置 selectColumns //重置 selectColumns
CPI.setSelectColumns(queryWrapper, selectColumns); CPI.setSelectColumns(queryWrapper, selectColumns);
//重置 orderBys
if (CollectionUtil.isNotEmpty(orderBys)) {
CPI.setOrderBys(queryWrapper, orderBys);
}
//重置 join //重置 join
if (removedJoins) { if (removedJoins) {
CPI.setJoins(queryWrapper, joins); CPI.setJoins(queryWrapper, joins);