mybatis plus 使用hutool 分页导出 千万级别数据

工具类

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.IoUtil;
import cn.hutool.core.lang.UUID;
import cn.hutool.core.util.ObjectUtil;
import cn.hutool.poi.excel.ExcelWriter;
import lombok.extern.slf4j.Slf4j;

import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;
import java.util.function.BiFunction;
import java.util.function.Supplier;

@Slf4j
public class ExcelUtil {


    private static final Integer PAGE_SIZE = 100000;

    private static final Integer SHEET_MAX_SIZE = 100000;
    
    public static <T> void executeTask(HttpServletResponse response, Supplier<Long> countSupplier
            , BiFunction<Integer, Integer, List<T>> queryFunc
            , LinkedHashMap<String, String> header
            , Integer width
            , String excelName) {
        ExcelWriter writer;
        ServletOutputStream out = null;
        Long totalCount = countSupplier.get();
        log.info("总条数;{}", totalCount);
        int writeCount = 0;
        long pageTotal = (totalCount % PAGE_SIZE) > 0 ? (totalCount / PAGE_SIZE) + 1 : totalCount / PAGE_SIZE;
        writer = cn.hutool.poi.excel.ExcelUtil.getBigWriter(UUID.fastUUID()+".xlsx", "sheet1");
        int sheetNo = writeCount / SHEET_MAX_SIZE;
        for (int pageNo = 1; pageNo <= pageTotal; pageNo++) {
            writer.setSheet("sheet" + pageNo);
            List<T> list = queryFunc.apply(pageNo, PAGE_SIZE);
            log.info("pageNo:{},resultSize:{}",pageNo,list.size());

            writeCount += list.size();
            AtomicReference<Integer> index = new AtomicReference<>(0);
            header.forEach((k, v) -> {
                writer.addHeaderAlias(k, v);
                writer.setColumnWidth(index.get(), width);
                index.getAndSet(index.get() + 1);
            });
//            writer.autoSizeColumnAll();
            writer.setOnlyAlias(true);
            writer.write(list, true);

        }
        try {
            response.setContentType("application/vnd.ms-excel;charset=utf-8");
            response.setHeader("Content-disposition", "attachment; filename=" + URLEncoder.encode(excelName + "-" + DateUtil.today() + ".xls", "utf-8"));
            out = response.getOutputStream();
            writer.flush(out, true);
        } catch (IOException e) {
            log.error("导出异常错误!!");
            throw new RuntimeException("导出异常");
        } finally {
            if (ObjectUtil.isNotEmpty(writer)) {
                writer.close();
            }
            IoUtil.close(out);
        }

    }


    /**
     * @param response
     * @param maps      导出数据
     * @param header    头部
     * @param width     表格宽度
     * @param excelName 导出文件名字
     * @throws IOException
     */
    public static void exportExcel(HttpServletResponse response, List<Map<String, Object>> maps, LinkedHashMap<String, String> header, Integer width, String excelName) {
        ExcelWriter writer = null;
        ServletOutputStream out = null;
        try {
            writer = cn.hutool.poi.excel.ExcelUtil.getBigWriter();
            AtomicReference<Integer> index = new AtomicReference<>(0);
            ExcelWriter finalWriter = writer;
            header.forEach((k, v) -> {
                finalWriter.addHeaderAlias(k, v);
                finalWriter.setColumnWidth(index.get(), width);
                index.getAndSet(index.get() + 1);
            });
            finalWriter.setOnlyAlias(true);
            writer.write(maps, true);

            response.setContentType("application/vnd.ms-excel;charset=utf-8");
            response.setHeader("Content-disposition", "attachment; filename=" + URLEncoder.encode(excelName + "-" + DateUtil.today() + ".xls", "utf-8"));

            out = response.getOutputStream();
            writer.flush(out, true);
        } catch (IOException e) {
            log.error("导出异常错误!!");
            throw new RuntimeException("导出异常");
        } finally {
            if (ObjectUtil.isNotEmpty(writer)) {
                writer.close();
            }
            IoUtil.close(out);
        }
    }

    /**
     * @param response
     * @param maps      导出数据
     * @param header    头部
     * @param width     表格宽度
     * @param excelName 导出文件名字
     * @param doAction  接口函数
     * @throws IOException
     */
    public static void exportExcel(HttpServletResponse response, List<Map<String, Object>> maps, LinkedHashMap<String, String> header, Integer width, String excelName, Action doAction) {
        ExcelWriter writer = null;
        ServletOutputStream out = null;
        try {
            writer = cn.hutool.poi.excel.ExcelUtil.getBigWriter();
            AtomicReference<Integer> index = new AtomicReference<>(0);
            ExcelWriter finalWriter = writer;
            doAction.doAction(writer);
            header.forEach((k, v) -> {
                finalWriter.addHeaderAlias(k, v);
                finalWriter.setColumnWidth(index.get(), width);
                index.getAndSet(index.get() + 1);
            });
            finalWriter.setOnlyAlias(true);
            writer.write(maps, true);

            response.setContentType("application/vnd.ms-excel;charset=utf-8");
            response.setHeader("Content-disposition", "attachment; filename=" + URLEncoder.encode(excelName + "-" + DateUtil.today() + ".xls", "utf-8"));


            out = response.getOutputStream();
            writer.flush(out, true);
        } catch (IOException e) {
            log.error("导出异常错误!!");
            throw new RuntimeException("导出异常");
        } finally {
            writer.close();
            IoUtil.close(out);
        }
    }

    public interface Action {
        void doAction(ExcelWriter writer);
    }
}

调用类

        ExcelUtil.executeTask(response, () -> baseMapper.getCount(tabInsBase), (no, size) -> {
            List<TabInsBase> records = this.getPage(new Page<>(no, size), tabInsBase).getRecords();
            return records;
        }, linkedHashMap, 50, "导出名称");

  • 9
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值