Java Web应用中打包多个CSV文件并直接流式传输到浏览器

FileOutputStream baos = new FileOutputStream("myZip.zip"); // 创建一个本地文件输出流
ZipOutputStream zos = new ZipOutputStream(baos); // ZIP内容写入本地文件

for(String sCurrent : selectedFiles){
    zos.putNextEntry(new ZipEntry(new File(sCurrent).getName()));
    Files.copy(Paths.get(sCurrent), zos);
    zos.closeEntry();
}
zos.close(); // 关闭ZIP输出流，完成本地ZIP文件的写入

// 尝试发送响应，但此时response.getOutputStream()并未获得myZip.zip的内容
response.getOutputStream().flush();
response.getOutputStream().close();

// 核心改动：ZipOutputStream直接包裹response.getOutputStream()
ZipOutputStream zos = new ZipOutputStream(response.getOutputStream());

response.setContentType("application/zip");
// filename 建议使用英文或进行URL编码，以避免中文乱码问题
response.setHeader("Content-Disposition", "attachment; filename=\"my_archive.zip\"");

// 使用try-with-resources确保ZipOutputStream及其底层流被正确关闭
try (ZipOutputStream zos = new ZipOutputStream(response.getOutputStream())) {
    // ... 写入ZIP条目和文件内容 ...
} catch (IOException e) {
    // ... 处理异常 ...
}

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.Arrays;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
import java.nio.charset.StandardCharsets; // 用于指定ZIP文件名的编码

@WebServlet("/downloadZip") // 配置Servlet的访问路径
public class ZipFileDownloadServlet extends HttpServlet {

    @Override
    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        // 假设这是您要打包的CSV文件路径列表
        // 实际应用中，这些路径可能来自数据库查询、文件系统扫描、用户上传等
        List<String> selectedFilePaths = Arrays.asList(
            "/data/reports/sales_2023.csv",
            "/data/reports/customers_export.csv",
            "/data/temp/another_data_file.csv"
        );

        // 设置HTTP响应头，指示浏览器下载一个ZIP文件
        response.setContentType("application/zip");
        // 设置下载的文件名，注意对非ASCII字符的编码处理
        // 这里假设文件名是英文，如果包含中文，需要进行URL编码或使用特定的编码方式
        response.setHeader("Content-Disposition", "attachment; filename=\"my_csv_archive.zip\"");

        // 使用try-with-resources确保ZipOutputStream及其底层流（response.getOutputStream()）被正确关闭
        try (ZipOutputStream zos = new ZipOutputStream(response.getOutputStream(), StandardCharsets.UTF_8)) { // 指定UTF-8编码处理文件名
            for (String filePath : selectedFilePaths) {
                Path sourcePath = Paths.get(filePath);

                // 检查文件是否存在且可读，避免不必要的错误
                if (Files.exists(sourcePath) && Files.isReadable(sourcePath)) {
                    // 创建ZIP文件中的一个条目。使用原始文件名作为条目名。
                    ZipEntry zipEntry = new ZipEntry(sourcePath.getFileName().toString());
                    zos.putNextEntry(zipEntry); // 开始写入新的ZIP条目

                    // 将源文件的所有字节高效地复制到ZIP输出流中
                    Files.copy(sourcePath, zos);

                    zos.closeEntry(); // 关闭当前ZIP文件条目，准备下一个
                } else {
                    // 如果文件不存在或不可读，记录警告或跳过
                    System.err.println("Warning: File not found or not readable, skipping: " + filePath);
                    // 实际应用中可能需要更详细的错误处理，例如向用户反馈哪些文件未能包含
                }
            }
            // try-with-resources 会自动调用 zos.close()，进而关闭 response.getOutputStream()
        } catch (IOException e) {
            // 捕获在ZIP生成或传输过程中可能发生的IO异常
            System.err.println("Error generating or streaming ZIP file: " + e.getMessage());
            // 向客户端返回一个错误状态码和信息
            response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR);
            response.getWriter().write("Failed to generate or download the ZIP file due to an internal error.");
        }
    }
}