MyBatis-Plus对于大数据量查询，采用分页查询按批次处理结果，通用工具封装

Posted 2022-01-24 抓手

定义业务处理接口

import java.util.List;

/**
 * @author 向振华
 * @date 2021/12/30 11:30
 */
public interface PageService<T> 

    /**
     * 业务处理
     *
     * @param records
     */
    void business(List<T> records);

定义工具类，这里提供了两种实现方式

import com.baomidou.mybatisplus.core.conditions.Wrapper;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.baomidou.mybatisplus.extension.service.IService;
import org.springframework.util.CollectionUtils;

/**
 * @author 向振华
 * @date 2021/12/30 1:41
 */
public class PageUtils 

    public static <T> void pageTurner(LambdaQueryChainWrapper<T> lambdaQueryChainWrapper, PageService<T> pageService) 
        // 当前页
        long current = 1;
        // 分页大小，可以修改
        long size = 200;
        for (; ; ) 
            // 分页
            Page<T> page = new Page<>(current, size);
            // 执行
            lambdaQueryChainWrapper.page(page);
            // 结果为空，直接返回
            if (CollectionUtils.isEmpty(page.getRecords())) 
                return;
            
            // 业务处理
            pageService.business(page.getRecords());
            // 说明没有下一页，直接返回
            if (current * size >= page.getTotal()) 
                return;
            
            // 下一页
            current++;
        
    

    public static <T> void pageTurner(IService<T> iService, Wrapper<T> queryWrapper, PageService<T> pageService) 
        // 当前页
        long current = 1;
        // 分页大小，可以修改
        long size = 200;
        for (; ; ) 
            // 分页
            Page<T> page = new Page<>(current, size);
            // 执行
            iService.page(page, queryWrapper);
            // 结果为空，直接返回
            if (CollectionUtils.isEmpty(page.getRecords())) 
                return;
            
            // 业务处理
            pageService.business(page.getRecords());
            // 说明没有下一页，直接返回
            if (current * size >= page.getTotal()) 
                return;
            
            // 下一页
            current++;
        
    

使用方式

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
import com.bzcst.bop.charge.Application;
import com.bzcst.bop.oms.orm.model.po.UserOrder;
import com.bzcst.bop.oms.orm.service.UserOrderService;
import lombok.extern.slf4j.Slf4j;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import javax.annotation.Resource;
import java.util.List;

/**
 * @author 向振华
 * @date 2021/11/11 19:34
 */
@Slf4j
@SpringBootTest(classes = Application.class)
@RunWith(SpringRunner.class)
public class OrderTest 

    @Resource
    private UserOrderService userOrderService;

    @Test
    public void name1() 

        LambdaQueryChainWrapper<UserOrder> lqcw = userOrderService.lambdaQuery().eq(UserOrder::getUserId, 29L);

        PageUtils.pageTurner(lqcw, new PageService<UserOrder>() 
            @Override
            public void business(List<UserOrder> records) 
                for (UserOrder record : records) 
                    System.out.println(record);
                
            
        );
    

    @Test
    public void name2() 

        LambdaQueryWrapper<UserOrder> lqw = new LambdaQueryWrapper<>();
        lqw.eq(UserOrder::getUserId, 29L);

        // 如果不需要查询条件，lqw传null
        PageUtils.pageTurner(userOrderService, lqw, new PageService<UserOrder>() 
            @Override
            public void business(List<UserOrder> records) 
                for (UserOrder record : records) 
                    System.out.println(record);
                
            
        );