/**
 * 壓縮文件(支持單個(gè)文件和單個(gè)文件夾)
 * @param sourceFile 被壓縮文件/文件夾
 * @param zipFile Zip文件
 */
public static void zipCompress(File sourceFile, File zipFile) {
    try (ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(zipFile))) {
        // 設(shè)置壓縮方法
        zos.setMethod(ZipOutputStream.DEFLATED);
        zos.setLevel(Deflater.BEST_COMPRESSION); // 默認(rèn)為-1,壓縮級(jí)別，1速度快，效率低，9 速度滿，效率高
        // zos.setLevel(Deflater.BEST_SPEED);
        zos.setComment("zip文件說明");
        // 處理文件夾
        if (sourceFile.exists() && sourceFile.isDirectory() && Objects.nonNull(sourceFile.listFiles())){
            Arrays.stream(Objects.requireNonNull(sourceFile.listFiles())).forEach(file -> {
                addZipFile(file, zos);
            });
        }else{
            addZipFile(sourceFile, zos);
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}

/**
 * 向ZIP中添加文件
 * @param file 源文件
 * @param zos zip輸出流
 */
private static void addZipFile(File file, ZipOutputStream zos){
    if (!file.exists() || file.isDirectory()){
        throw new RuntimeException("文件不存在或該文件為文件夾，請(qǐng)檢查");
    }
    try {
        // 讀入文件
        FileInputStream fis = new FileInputStream(file);
        // 創(chuàng)建壓縮對(duì)象并設(shè)置一些屬性
        ZipEntry entry = new ZipEntry(file.getName());
        entry.setMethod(ZipEntry.DEFLATED); // 壓縮方法默認(rèn)為DEFLATED
        // entry.setMethod(ZipEntry.STORED); // STORED（不壓縮）。當(dāng)使用STORED壓縮方法時(shí)，需要設(shè)置未壓縮的數(shù)據(jù)大小和CRC-32校驗(yàn)和，否則壓縮和解壓縮時(shí)會(huì)出現(xiàn)錯(cuò)誤。
        entry.setSize(file.length()); // 設(shè)置未壓縮的數(shù)據(jù)大小，這里設(shè)置的是文件大小
        // 計(jì)算 CRC-32 校驗(yàn)碼
        // byte[] data = Files.readAllBytes(file.toPath());
        // CRC32 crc = new CRC32();
        // crc.update(data);
        // entry.setCrc(crc.getValue()); // 設(shè)置CRC-32校驗(yàn)和，用于保證壓縮后的數(shù)據(jù)完整性，盡量別手動(dòng)設(shè)置，可以通過CRC-32計(jì)算
        entry.setCompressedSize(file.length()); // 設(shè)置壓縮后的數(shù)據(jù)大小，這里設(shè)置的是使用DEFLATED方法壓縮后的數(shù)據(jù)大小
        entry.setExtra(new byte[]{}); // 設(shè)置額外的數(shù)據(jù)，這里設(shè)置為空
        entry.setComment("file comment"); // 設(shè)置ZipEntry的注釋，即文件說明
        entry.setCreationTime(FileTime.from(Instant.now())); // 設(shè)置文件的創(chuàng)建時(shí)間
        entry.setLastAccessTime(FileTime.from(Instant.now())); // 設(shè)置文件的最后訪問時(shí)間
        entry.setLastModifiedTime(FileTime.from(Instant.now())); // 設(shè)置文件的最后修改時(shí)間。
        // 向ZIP輸出流中添加一個(gè)ZIP實(shí)體，構(gòu)造方法中的name參數(shù)指定文件在ZIP包中的文件名
        zos.putNextEntry(entry);
        // 向ZIP實(shí)體中寫入內(nèi)容
        byte[] buf = new byte[1024];
        int len;
        while ((len = fis.read(buf)) > 0) {
            zos.write(buf, 0, len);
        }
        // 關(guān)閉ZipEntry
        zos.closeEntry();
    } catch (IOException e) {
        e.printStackTrace();
    }
}

/**
 * 解壓縮ZIP文件
 * @param zipFile ZIP文件
 * @param destDir 目標(biāo)路徑
 */
public static void zipDecompress(File zipFile, File destDir) {
    byte[] buffer = new byte[1024];
    try (ZipInputStream zis = new ZipInputStream(new FileInputStream(zipFile))) {
        ZipEntry entry = zis.getNextEntry();
        while (entry != null) {
            File file = new File(destDir, entry.getName());
            if (entry.isDirectory()) {
                file.mkdirs();
            } else {
                File parent = file.getParentFile();
                if (!parent.exists()) {
                    parent.mkdirs();
                }
                try (FileOutputStream fos = new FileOutputStream(file)) {
                    int len;
                    while ((len = zis.read(buffer)) > 0) {
                        fos.write(buffer, 0, len);
                    }
                }
            }
            entry = zis.getNextEntry();
        }
    } catch (IOException e) {
        e.printStackTrace();
    }
}

@Test
void testJavaUtilZip(){
    // 測(cè)試壓縮和解壓縮單文件ZIP
    // 被壓縮的MP4單文件，大小112.5MB
    File inputFile = new File("/Users/zlc/Documents/own/images/GPT-4 Developer Livestream.mp4");
    // ZIP文件路徑
    File zipFile = new File("/Users/zlc/Documents/own/mp4.zip");
    // ZIP 解壓縮路徑
    File unzipFile = new File("/Users/zlc/Documents/own/unzip");
    long start = System.currentTimeMillis();
    // 壓縮文件
    ZipFileUtil.zipCompress(inputFile, zipFile);
    long end = System.currentTimeMillis();
    System.out.println("ZIP-壓縮單文件耗時(shí):" + (end - start) + "毫秒");
    start = System.currentTimeMillis();
    ZipFileUtil.zipDecompress(zipFile, unzipFile);
    end = System.currentTimeMillis();
    System.out.println("ZIP-解壓縮單文件耗時(shí):" + (end - start) + "毫秒");
    // 單文件夾多文件壓縮和解壓縮測(cè)試
    // 文件夾大小2.42G
    File inputFiles = new File("/Users/zlc/Documents/own/images");
    File zipFiles = new File("/Users/zlc/Documents/own/imagesZip.zip");
    File unzipFiles = new File("/Users/zlc/Documents/own/imagesUnzip");
    start = System.currentTimeMillis();
    // 壓縮文件
    ZipFileUtil.zipCompress(inputFiles, zipFiles);
    end = System.currentTimeMillis();
    System.out.println("ZIP - 多文件壓縮耗時(shí):" + (end - start) + "毫秒");
    start = System.currentTimeMillis();
    ZipFileUtil.zipDecompress(zipFiles, unzipFiles);
    end = System.currentTimeMillis();
    System.out.println("ZIP - 多文件解壓縮耗時(shí):" + (end - start) + "毫秒");
}

// 不同壓縮等級(jí)下的處理耗時(shí) 
BEST_COMPRESSION 2.43G 文件夾耗時(shí)119801毫秒快兩分鐘了
BEST_SPEED 2.43G 文件夾耗時(shí)112646毫秒 也沒差多少，但確實(shí)快了

  /**
 * 下載ZIP
 * @param response HttpServletResponse 響應(yīng)流
 * @return zip file 
 */
@GetMapping(value = "/downloadZip")
public String downloadZipFile(HttpServletResponse response) {
    // 設(shè)置響應(yīng)頭
    response.setContentType("application/octet-stream");
    response.setHeader("Content-Disposition", "attachment; filename=download.zip");
    try (ZipOutputStream outputStream = new ZipOutputStream(response.getOutputStream())) {
        ExcelWriter writer = new ExcelWriter(true);
        List<String> header = Arrays.asList("開始日期", "結(jié)束日期", "算法廠商", "期末資產(chǎn)總額", "累計(jì)成交金額",
                "期間委托筆數(shù)", "期間成交筆數(shù)", "期間撤單筆數(shù)");
        writer.writeHeadRow(header);
        List<List<String>> rows = new ArrayList<>();
        rows.add(Arrays.asList("張三", "里斯", "男", "張三", "里斯", "男", "張三", "里斯"));
        rows.add(Arrays.asList("李四", "王武", "女", "張三", "里斯", "男", "張三", "里斯"));
        writer.write(rows);
        writer.passRows(1);
        List<String> header1 = Arrays.asList("日期", "資金賬號(hào)", "算法母單編號(hào)", "委托編號(hào)", "交易所", "股票代碼",
                "委托數(shù)量", "交易方向", "訂單類型", "委托價(jià)格", "委托狀態(tài)", "累計(jì)成交數(shù)量", "累計(jì)成交金額");
        writer.writeHeadRow(header1);
        writer.autoSizeColumn(writer.getCurrentRow());
        List<List<String>> rows2 = new ArrayList<>();
        rows2.add(Arrays.asList("張三", "里斯", "男", "張三", "里斯", "男", "張三", "里斯", "找大大", "趙打打", "炸", "茅臺(tái)", "米線"));
        rows2.add(Arrays.asList("李四", "王武", "女", "張三", "里斯", "男", "張三", "里斯", "找大大", "趙打打", "炸", "茅臺(tái)", "米線"));
        writer.write(rows2);
        writer.autoSizeColumnAll();
        ZipEntry entry = new ZipEntry("數(shù)據(jù).xlsx");
        outputStream.putNextEntry(entry);
        writer.flush(outputStream);
        writer.close();
        outputStream.closeEntry();
        ZipEntry entry1 = new ZipEntry("母單.csv");
        outputStream.putNextEntry(entry1);
        CsvWriter csvWriter1 = new CsvWriter(new OutputStreamWriter(outputStream));
        String[] csvHead1 = {"日期", "資金賬號(hào)", "算法廠商", "算法", "算法母單編號(hào)", "交易所", "股票代碼", "委托數(shù)量", "交易方向", "啟動(dòng)時(shí)間", "停止時(shí)間", "算法狀態(tài)"};
        csvWriter1.writeLine(csvHead1);
        String[] csvData1 = {"20230203", "98830901", "XX", "TWAP", "12984", "SH", "600000", "10000", "4", "102311", "112311", "1"};
        String[] csvData2 = {"20230203", "98830901", "XX", "TWAP", "12984", "SH", "600000", "10000", "4", "102311", "112311", "1"};
        csvWriter1.writeLine(csvData1);
        csvWriter1.writeLine(csvData2);
        csvWriter1.flush();
        outputStream.closeEntry();
        ZipEntry entry2 = new ZipEntry("子單.csv");
        outputStream.putNextEntry(entry2);
        CsvWriter csvWriter2 = new CsvWriter(new OutputStreamWriter(outputStream));
        String[] csvHead2 = {"日期", "資金賬號(hào)", "算法母單編號(hào)", "委托編號(hào)", "交易所", "股票代碼", "委托數(shù)量", "交易方向", "訂單類型", "委托價(jià)格", "委托狀態(tài)", "累計(jì)成交數(shù)量", "累計(jì)成交金額"};
        csvWriter2.writeLine(csvHead2);
        String[] csvData3 = {"20230203", "98830901", "12984", "ord1122", "SH", "600000", "1000", "4", "1", "7.23", "1", "800", "5600"};
        String[] csvData4 = {"20230203", "98830901", "12984", "ord1122", "SH", "600000", "1000", "4", "1", "7.23", "1", "800", "5600"};
        csvWriter2.writeLine(csvData3);
        csvWriter2.writeLine(csvData4);
        csvWriter2.flush();
        outputStream.closeEntry();
        outputStream.finish();
        return "success";
    } catch (Exception e) {
        return "faild";
    }
}