java HSSF excel大数据导出内存溢出

原创

mob64ca12e63b18 2024-04-07 05:42:00 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e63b18的原创作品，请联系作者获取转载授权，否则将追究法律责任

解决Java HSSF Excel大数据导出内存溢出问题

在开发中，我们经常会遇到需要导出大量数据到Excel文件的情况，而Apache POI库是一个非常常用的Java库，用来处理Excel文件。在使用POI库中的HSSF模块来导出Excel文件时，如果导出的数据量较大，就容易出现内存溢出的问题。

问题分析

在使用HSSF模块导出Excel文件时，当数据量很大时，会将所有数据加载到内存中，导致内存溢出。这是因为HSSF是基于DOM的模式，会在内存中创建一个完整的Excel文档对象树，当数据量大时，占用的内存也会很大。

解决方案

为了解决内存溢出的问题，我们可以使用SXSSFWorkbook来替代HSSFWorkbook。SXSSFWorkbook是POI库中专门为大数据量而设计的类，可以在写入数据时将部分数据写入临时文件而不是全部加载到内存中，从而避免内存溢出。

下面是一个使用SXSSFWorkbook导出Excel文件的示例代码：

import org.apache.poi.ss.usermodel.*;
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
import java.io.FileOutputStream;
import java.io.IOException;

public class ExcelExportUtil {

    public static void exportExcel(String filePath, String[] headers, Object[][] data) {
        try (SXSSFWorkbook workbook = new SXSSFWorkbook()) {
            Sheet sheet = workbook.createSheet("Sheet1");

            Row headerRow = sheet.createRow(0);
            for (int i = 0; i < headers.length; i++) {
                Cell cell = headerRow.createCell(i);
                cell.setCellValue(headers[i]);
            }

            for (int i = 0; i < data.length; i++) {
                Row dataRow = sheet.createRow(i + 1);
                for (int j = 0; j < data[i].length; j++) {
                    Cell cell = dataRow.createCell(j);
                    if (data[i][j] instanceof String) {
                        cell.setCellValue((String) data[i][j]);
                    } else if (data[i][j] instanceof Double) {
                        cell.setCellValue((Double) data[i][j]);
                    }
                }
            }

            try (FileOutputStream fileOut = new FileOutputStream(filePath)) {
                workbook.write(fileOut);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public static void main(String[] args) {
        String[] headers = {"Name", "Age", "Country"};
        Object[][] data = {{"Alice", 25, "USA"},
                           {"Bob", 30, "Canada"},
                           {"Charlie", 35, "UK"}};

        exportExcel("data.xlsx", headers, data);
    }
}