海量数据导出

package test;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.ExcelWriter;
import com.alibaba.excel.write.metadata.WriteSheet;
import lombok.extern.slf4j.Slf4j;

import java.io.FileOutputStream;
import java.io.IOException;
import java.util.List;

/**
 * 海量数据导出
 *
 * @author LBY
 * @date 2023年11月17日 02:11
 */

@Slf4j
public class Demo {
    public void exportData() throws IOException {
        // 总数据量
        int totalSize = dataService.getDataCount();
        // 分页大小
        int pageSize = 10000;
        // 总页码
        int totalPage = (totalSize + pageSize - 1) / pageSize;
        // 设置文件名
        String fileName = "导出数据.xlsx";
        // 创建输出流，写到本地或者云对象存储中
        FileOutputStream outputStream = new FileOutputStream("output.xlsx");
        // 创建excel对象
        ExcelWriter excelWriter = EasyExcel.write(outputStream).build();
        // 分批写入数据
        for (int i = 1; i <= totalPage; i++) {
            // 计算起始索引
            int startIndex = (i - 1) * pageSize;
            // 计算结束索引
            int endIndex = i * pageSize;
            if (endIndex > totalSize) {
                endIndex = totalSize;
            }
            // 分批查询获取数据
            List dataList = dataService.getDataList(startIndex, endIndex);
            // 创建Sheet对象
            WriteSheet writeSheet = EasyExcel.writerSheet("数据" + i).build();
            excelWriter.write(dataList, writeSheet);
        }
        // 关闭excelWriter对象
        excelWriter.finish();
        // 关闭输出流
        outputStream.flush();
        outputStream.close();
    }
}