大家好,我是 V 哥。使用EasyExcel进行大数据量导出时容易导致内存溢出,特别是在导出百万级别的数据时。你有遇到过这种情况吗,以下是V 哥整理的解决该问题的一些常见方法,分享给大家,欢迎一起讨论:
EasyExcel大数据量导出常见方法
1. 分批写入
- EasyExcel支持分批写入数据,可以将数据分批加载到内存中,分批写入Excel文件,避免一次性将大量数据加载到内存中。
- 示例代码:
java
String fileName = "large_data.xlsx";
ExcelWriter excelWriter = EasyExcel.write(fileName).build();
WriteSheet writeSheet = EasyExcel.writerSheet("Sheet1").build();
// 假设每次写入10000条数据
int batchSize = 10000;
List<Data> dataList;
int pageIndex = 0;
do {
// 分页获取数据
dataList = getDataByPage(pageIndex++, batchSize);
excelWriter.write(dataList, writeSheet);
} while (dataList.size() == batchSize);
// 关闭资源
excelWriter.finish();
2. 设置合适的JVM内存
- 针对大数据导出场景,可以尝试增大JVM的内存分配,例如:
shell
java -Xms512M -Xmx4G -jar yourApp.jar
- 解释 :
-Xms512M
:设置初始堆大小为512MB。-Xmx4G
:设置最大堆大小为4GB。
3. 减少数据对象的复杂性
- 导出数据时,尽量简化数据对象,避免不必要的嵌套和多余字段的加载,以减少对象占用的内存空间。
4. 关闭自动列宽设置
- EasyExcel的自动列宽功能会占用大量内存,特别是在数据量较大的情况下。关闭自动列宽可以节省内存。
- 示例代码:
java
EasyExcel.write(fileName)
.registerWriteHandler(new SimpleWriteHandler()) // 不使用自动列宽
.sheet("Sheet1")
.doWrite(dataList);
5. 使用Stream导出(适合大数据)
- 利用
OutputStream
分批写入数据,减少内存消耗。通过BufferedOutputStream
可以进一步提高性能。 - 示例代码:
java
try (OutputStream out = new BufferedOutputStream(new FileOutputStream(fileName))) {
ExcelWriter excelWriter = EasyExcel.write(out).build();
WriteSheet writeSheet = EasyExcel.writerSheet("Sheet1").build();
int pageIndex = 0;
List<Data> dataList;
do {
dataList = getDataByPage(pageIndex++, batchSize);
excelWriter.write(dataList, writeSheet);
} while (dataList.size() == batchSize);
excelWriter.finish();
} catch (IOException e) {
e.printStackTrace();
}
6. 选择合适的数据导出工具
- 如果数据量非常大,可以考虑切换到支持更高性能的导出工具(如Apache POI的
SXSSFWorkbook
),适合导出百万级别数据量,但配置和使用会更复杂。
亮点来了,那要如何使用 POI 的 SXSSFWorkbook来导出百万级别的数据量呢?
Apache POI的SXSSFWorkbook 实现百万级别数据量的导出案例
使用Apache POI的SXSSFWorkbook
可以处理大数据量的Excel导出,因为SXSSFWorkbook
基于流式写入,不会将所有数据加载到内存中,而是使用临时文件进行缓存,这样可以显著减少内存消耗,适合百万级别数据的导出。下面我们来看一个完整的实现示例。
代码如下
java
import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
public class LargeDataExportExample {
public static void main(String[] args) {
// 文件输出路径
String filePath = "vg_large_data_export.xlsx";
// 导出百万级数据
exportLargeData(filePath);
}
private static void exportLargeData(String filePath) {
// 每次写入的批次大小
final int batchSize = 10000;
// 数据总条数
final int totalRows = 1_000_000;
// 创建SXSSFWorkbook对象,内存中只保留100行,超过的部分会写入临时文件
SXSSFWorkbook workbook = new SXSSFWorkbook(100);
workbook.setCompressTempFiles(true); // 启用临时文件压缩
// 创建工作表
Sheet sheet = workbook.createSheet("Large Data");
// 创建标题行
Row headerRow = sheet.createRow(0);
String[] headers = {"ID", "Name", "Age"};
for (int i = 0; i < headers.length; i++) {
Cell cell = headerRow.createCell(i);
cell.setCellValue(headers[i]);
}
int rowNum = 1; // 数据开始的行号
try {
// 按批次写入数据
for (int i = 0; i < totalRows / batchSize; i++) {
// 模拟获取每批数据
List<Data> dataList = getDataBatch(rowNum, batchSize);
// 将数据写入到Excel中
for (Data data : dataList) {
Row row = sheet.createRow(rowNum++);
row.createCell(0).setCellValue(data.getId());
row.createCell(1).setCellValue(data.getName());
row.createCell(2).setCellValue(data.getAge());
}
// 处理完成一批数据后,可以选择清除缓存数据,防止内存溢出
((SXSSFSheet) sheet).flushRows(batchSize); // 清除已写的行缓存
}
// 将数据写入文件
try (FileOutputStream fos = new FileOutputStream(filePath)) {
workbook.write(fos);
}
System.out.println("数据导出完成:" + filePath);
} catch (IOException e) {
e.printStackTrace();
} finally {
// 关闭workbook并删除临时文件
workbook.dispose();
}
}
/**
* 模拟分页获取数据
*/
private static List<Data> getDataBatch(int startId, int batchSize) {
List<Data> dataList = new ArrayList<>(batchSize);
for (int i = 0; i < batchSize; i++) {
dataList.add(new Data(startId + i, "Name" + (startId + i), 20 + (startId + i) % 50));
}
return dataList;
}
// 数据类
static class Data {
private final int id;
private final String name;
private final int age;
public Data(int id, String name, int age) {
this.id = id;
this.name = name;
this.age = age;
}
public int getId() {
return id;
}
public String getName() {
return name;
}
public int getAge() {
return age;
}
}
}
来解释一下代码
- SXSSFWorkbook :
SXSSFWorkbook(100)
表示内存中最多保留100行数据,超过的部分会写入临时文件,节省内存。 - 批次处理 :通过
batchSize
控制每批次写入的数据量,以减少内存消耗。totalRows
设置为1,000,000表示导出100万条数据。 - 模拟数据生成 :
getDataBatch
方法模拟分页获取数据,每次返回一批数据。 - 清除缓存行 :每次写入一批数据后,通过
flushRows(batchSize)
将缓存的行从内存中清除,以控制内存占用。 - 压缩临时文件 :
workbook.setCompressTempFiles(true)
启用临时文件压缩,进一步减少磁盘空间占用。
需要注意的事项
- 临时文件:SXSSFWorkbook会在系统临时文件夹中生成临时文件,需要确保磁盘空间足够。
- 资源释放 :完成数据写入后需要调用
workbook.dispose()
以清理临时文件。 - 性能优化 :可根据机器内存调整
batchSize
和SXSSFWorkbook
缓存行数,避免频繁刷新和内存溢出。