import java.io.*;
import java.nio.file.*;
import java.util.zip.*;

public class ZipDemo {
    public static void main(String[] args) throws IOException {
        // 源文件夾路徑：要壓縮的文件所在的目錄
        String sourceDir = "path/to/folder";

        // 輸出 ZIP 文件路徑：生成的壓縮包保存的位置和名稱
        String zipFilePath = "output.zip";

        // 使用 try-with-resources 自動關閉資源（FileOutputStream 和 ZipOutputStream）
        try (FileOutputStream fos = new FileOutputStream(zipFilePath);
             ZipOutputStream zos = new ZipOutputStream(fos)) {

            // 設置 ZIP 壓縮級別為最佳壓縮（壓縮率最高，但速度較慢）
            zos.setLevel(Deflater.BEST_COMPRESSION);

            // 遍歷源目錄下的所有文件（包括子目錄中的文件）
            Files.walk(Paths.get(sourceDir))
                 // 過濾掉目錄，只保留文件
                 .filter(path -> !Files.isDirectory(path))
                 // 對每個文件執(zhí)行以下操作
                 .forEach(path -> {
                     try {
                         // 構建相對路徑字符串：sourceDir + 文件分隔符 + 文件名
                         // 注意：這會導致 ZIP 中的路徑包含 sourceDir 的前綴
                         String relativePath = sourceDir + File.separator + path.getFileName();

                         // 創(chuàng)建一個 ZIP 條目（即 ZIP 包中的一個文件）
                         ZipEntry zipEntry = new ZipEntry(relativePath);

                         // 開始寫入該條目到 ZIP 流中
                         zos.putNextEntry(zipEntry);

                         // 讀取文件內容為字節(jié)數(shù)組
                         byte[] bytes = Files.readAllBytes(path);

                         // 將文件內容寫入 ZIP 輸出流
                         zos.write(bytes, 0, bytes.length);

                         // 關閉當前 ZIP 條目，準備寫入下一個文件
                         zos.closeEntry();

                     } catch (IOException e) {
                         // 如果寫入失敗，打印錯誤信息并輸出異常堆棧
                         System.err.println("寫入 ZIP 失敗: " + path);
                         e.printStackTrace();
                     }
                 });
        }
        // try-with-resources 會自動關閉 fos 和 zos，無需手動 close()
    }
}

response.setContentType("application/zip");
response.setHeader("Content-Disposition", "attachment; filename=\"streamed_files.zip\"");

try (ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream(), StandardCharsets.UTF_8)) {
    zipOut.setLevel(Deflater.BEST_SPEED);

    for (User u : userList) {
        String fileName = String.format("%sxxxx.docx", u.getNickName());

        try {
            zipOut.putNextEntry(new ZipEntry(fileName));
            byte[] docBytes = genDocx(u).toByteArray();
            zipOut.write(docBytes, 0, docBytes.length);
            zipOut.closeEntry();
        } catch (IOException e) {
            log.error("寫入 ZIP 條目失敗: {}", fileName, e);
            throw new IOException("生成 ZIP 條目時出錯: " + fileName, e);
        }
    }

} catch (IOException ex) {
    log.error("導出 ZIP 異常", ex);
    throw new RuntimeException("導出異常");
}

response.setContentType("application/zip");

response.setHeader("Content-Disposition", "attachment; filename=\"streamed_files.zip\"");

try (ZipOutputStream zipOut = new ZipOutputStream(response.getOutputStream(), StandardCharsets.UTF_8)) {

zipOut.setLevel(Deflater.BEST_SPEED);

zipOut.putNextEntry(new ZipEntry(fileName));

byte[] docBytes = genDocx(u).toByteArray();
zipOut.write(docBytes, 0, docBytes.length);

zipOut.closeEntry();