Java 多文件内存生成并压缩下载方案

设计思路

实现多文件打包下载主要有两种常见路径：

1. 基于文件系统（思路一）：在服务器本地创建临时文件夹，逐个导出文件后执行压缩，下载完毕再清理临时文件。该方法逻辑直观但涉及磁盘 IO，且需处理文件清理异常。
2. 基于内存操作（思路二）：所有文件直接在内存中以字节数组形式存在，随后将这些字节数组依次写入压缩输出流。本文采用此方案，具有代码简洁、资源开销相对可控的优势。

准备条件

• Java 运行环境及 Servlet API (javax.servlet)
• EasyExcel 依赖用于快速生成 Excel 数据流
• Java 标准库中的 java.util.zip.ZipOutputStream 和 java.io.ByteArrayOutputStream

核心操作步骤

第1步：Controller 层接口定义

目的：接收前端请求并将处理转发给业务逻辑层。

代码入口：/zip/download2

// 示例 Controller 调用
@RequestMapping("/zip/download2")
public void downloadZip2(String name, HttpServletResponse httpResponse) {
    // 委托给 Service 层执行核心压缩逻辑
    documentService.downloadReport(name, httpResponse);
}

第2步：Service 层组装文件列表

目的：生成单个文件的字节数据并将其收集到列表中，随后触发压缩流程。

操作说明：调用具体的业务导出方法获取文件元数据（文件名和字节数组），合并后传入压缩方法。

public void downloadReport(String name, HttpServletResponse httpResponse){
    // 调用具体生成逻辑，返回包含文件名和byte数组的Map
    Map excelOut = exportExcel(name); 
    Map excelOut2 = exportExcel2(name);
    
    List<Map> fileList = new ArrayList<>();
    fileList.add(excelOut);
    fileList.add(excelOut2);
    
    // 执行内存压缩并下载
    zipFiles(fileList, httpResponse);
}

第3步：Excel 文件转字节数组

目的：将 ExcelWriter 的输出目标指向 ByteArrayOutputStream，从而将文件内容转化为字节数组存储于内存。

重要参数：excelFileName 用于后续解压后的文件名显示；out.toByteArray() 为核心数据包。

public Map<String, Object> exportExcel(String name) {
    List<User> userList = userDao.queryUsersByName(name);
    try (ByteArrayOutputStream out = new ByteArrayOutputStream()) {
        String excelFileName = "查询数据.xlsx";
        // 初始化 Writer 写入内存
        ExcelWriter writer = new ExcelWriter(out, ExcelTypeEnum.XLSX, true);
        Sheet sheet = new Sheet(1, 0, User.class);
        sheet.setAutoWidth(Boolean.TRUE);
        writer.write(userList, sheet);
        writer.finish();
        
        // 结果封装
        Map<String, Object> result = new HashMap<>();
        result.put("fileName", excelFileName);
        result.put("outByte", out.toByteArray());
        return result;
    } catch (Exception e) {
        throw new RuntimeException("Excel生成失败", e);
    }
}

第4步：ZipOutputStream 内存打包

目的：遍历文件字节列表，使用 ZipEntry 包装每个文件头信息，将字节写入 ZIP 流并直接输出至浏览器。

可观察的完成状态：HTTP 响应头已设置为 application/zip，客户端开始接收二进制流。

public void zipFiles(List<Map> fileList, HttpServletResponse httpResponse) throws IOException {
    // 设置 HTTP 响应属性
    httpResponse.setContentType("application/zip");
    httpResponse.setHeader("Content-Disposition", "attachment;fileName=" + URLEncoder.encode("附件.zip", "UTF-8"));
    
    // 直接连接 HTTP 输出流进行压缩
    try (ZipOutputStream zipOutputStream = new ZipOutputStream(httpResponse.getOutputStream())) {
        for (Map map : fileList) {
            String fileName = (String) map.get("fileName");
            byte[] fileBytes = (byte[]) map.get("outByte");
            
            if (fileBytes != null && fileBytes.length > 0) {
                ZipEntry zipEntry = new ZipEntry(fileName);
                zipOutputStream.putNextEntry(zipEntry);
                zipOutputStream.write(fileBytes);
                zipOutputStream.closeEntry(); // 记得关闭当前条目
            }
        }
    }
}

上述流程完成后，用户点击链接即可获取包含多个文件的 Zip 包。其实际表现如下：

排错建议

• ZipEntry 未关闭：在循环中写入完 write 后务必调用 closeEntry()，否则生成的压缩包可能损坏或无法解压。
• 内存溢出：若单文件或总量极大（如几百兆以上），该内存方案可能导致 OutOfMemoryError，此时应考虑切换回“思路一”基于磁盘临时目录的操作方式。
• 编码问题：确保文件名中包含中文时，使用 URLEncoder.encode(..., "UTF-8") 正确转义，防止乱码。

总结

本方案通过在 ByteArrayOutputStream 中缓冲数据并在 ZipOutputStream 中直接流转的方式，有效简化了文件 IO 步骤，适用于中大规模数据的轻量级聚合下载场景。以上就是基于 Java 内存流实现多文件生成并打包下载的详细内容，更多关于 Java EasyExcel 与压缩工具的资料请关注本站其它相关文章！