import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
public class ZipUtil {

    private static final int BUFFER_SIZE = 1024 * 2;
    
    /**
     * 壓縮成ZIP 方法1
     * @param files         需要壓縮的文件集合
     * @param zipFilePath   壓縮文件路徑
     * @throws Exception
     */
    public static void toZip(List<File> files, String zipFilePath) throws Exception {
        ZipOutputStream zos = null;
        try {
            OutputStream out = new FileOutputStream(new File(zipFilePath));
            zos = new ZipOutputStream(out);
            for (File file : files) {
                byte[] buf = new byte[BUFFER_SIZE];
                zos.putNextEntry(new ZipEntry(file.getName()));
                FileInputStream in = new FileInputStream(file);
                int len;
                while ((len = in.read(buf)) != -1) {
                    zos.write(buf, 0, len);
                }
                zos.closeEntry();
                in.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (zos != null) {
                zos.close();
            }
        }
    }

    /**
     * 壓縮成ZIP 方法2
     * @param fileDirPath   壓縮文件夾路徑
     * @param zipFilePath   壓縮文路徑
     * @param keepDirStructure 是否保留原來(lái)的目錄結(jié)構(gòu),true:保留目錄結(jié)構(gòu);false:所有文件跑到壓縮包根目錄下(注意：
     *                      不保留目錄結(jié)構(gòu)可能會(huì)出現(xiàn)同名文件,會(huì)壓縮失敗)
     * @throws Exception
     */
    public static void toZip(String fileDirPath, String zipFilePath, boolean keepDirStructure) throws Exception {
        ZipOutputStream zos = null;
        try {   
            OutputStream out = new FileOutputStream(new File(zipFilePath));
            zos = new ZipOutputStream(out);
            File dirFile = new File(fileDirPath);
            compress(dirFile, zos, dirFile.getName(), keepDirStructure);
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (zos != null) {
                zos.close();
            }
        }
    }

    /**
     * 遞歸壓縮方法
     * @param sourceFile        源文件
     * @param zos               zip輸出流
     * @param name              壓縮后的文件夾/文件名稱
     * @param keepDirStructure  是否保留原來(lái)的目錄結(jié)構(gòu),true:保留目錄結(jié)構(gòu);
     *            false:所有文件跑到壓縮包根目錄下(注意：不保留目錄結(jié)構(gòu)可能會(huì)出現(xiàn)同名文件,會(huì)壓縮失敗)
     * @throws Exception
     */
    public static void compress(File sourceFile, ZipOutputStream zos, String name, boolean keepDirStructure) throws Exception {
        byte[] buf = new byte[BUFFER_SIZE];
        if (sourceFile.isFile()) {
            //向zip輸出流中添加一個(gè)zip實(shí)體，構(gòu)造器中name為zip實(shí)體的文件的名字
            zos.putNextEntry(new ZipEntry(name));
            //copy文件到zip輸出流中
            int len;
            FileInputStream in = new FileInputStream(sourceFile);
            while ((len = in.read(buf)) != -1) {
                zos.write(buf, 0, len);
            }
            zos.closeEntry();
            in.close();
        } else {
            File[] listFiles = sourceFile.listFiles();
            if (listFiles == null || listFiles.length == 0) {
                //需要保留原來(lái)的文件結(jié)構(gòu)時(shí),需要對(duì)空文件夾進(jìn)行處理
                if (keepDirStructure) {
                    //空文件夾的處理
                    zos.putNextEntry(new ZipEntry(name + "/"));
                    //沒(méi)有文件，不需要文件的copy
                    zos.closeEntry();
                }
            } else {
                for (File file : listFiles) {
                    // 判斷是否需要保留原來(lái)的文件結(jié)構(gòu)
                    if (keepDirStructure) {
                        // 注意：file.getName()前面需要帶上父文件夾的名字加一斜杠,不然最后壓縮包中就不能保留原來(lái)的文件結(jié)構(gòu),即：所有文件都跑到壓縮包根目錄下了
                        compress(file, zos, name + "/" + file.getName(), keepDirStructure);
                    } else {
                        compress(file, zos, file.getName(), keepDirStructure);
                    }
                }
            }
        }
    }
}