package com.inovance.paas.tpm.infra.util;
import org.apache.commons.compress.utils.IOUtils;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;
public class DownloadZipUtil {
public static void downloadZip(HttpServletResponse response, String sourceFolderPath, String zipName, String filePath){
// 获取文件分隔符（动态适配操作系统）
String separator = File.separator;
String zipFilePath = filePath + "zip"+ separator + zipName +".zip"; // 生成的ZIP文件路径
FileOutputStream fos;ZipOutputStream zos;try {fos = new FileOutputStream(zipFilePath);zos = new ZipOutputStream(fos);addFolderToZip("", sourceFolderPath, zos);zos.close();fos.close();File downloadFile = new File(zipFilePath);FileInputStream inputStream = new FileInputStream(downloadFile);response.setContentType("application/zip");response.setHeader("Content-Disposition", "attachment; filename=\"" + downloadFile.getName() + "\"");IOUtils.copy(inputStream, response.getOutputStream()); // 使用IOUtils将ZIP响应给客户端inputStream.close();response.flushBuffer();doGet(response, zipFilePath, zipName);} catch (FileNotFoundException e) {e.printStackTrace();} catch (IOException e) {e.printStackTrace();}
}private static void addFileToZip(String path, String srcFile, ZipOutputStream zip) throws IOException {File folder = new File(srcFile);if (folder.isDirectory()) {addFolderToZip(path, srcFile, zip);} else {byte[] buf = new byte[1024];int len;FileInputStream in = new FileInputStream(srcFile);zip.putNextEntry(new ZipEntry(path + "/" + folder.getName()));while ((len = in.read(buf)) > 0) {zip.write(buf, 0, len);}}
}private static void addFolderToZip(String path, String srcFolder, ZipOutputStream zip) throws IOException {File folder = new File(srcFolder);for (String fileName : folder.list()) {if (path.equals("")) {addFileToZip(folder.getName(), srcFolder + "/" + fileName, zip);} else {addFileToZip(path + "/" + folder.getName(), srcFolder + "/" + fileName, zip);}}
}protected static void doGet(HttpServletResponse response, String filePath, String zipName) throws IOException {//设置文件路径File downloadFile = new File(filePath);FileInputStream inStream = new FileInputStream(downloadFile);//设置下载的文件类型为zipresponse.setContentType("application/zip");//设置文件名，此处以zipFile.zip为例response.setHeader("Content-Disposition", "attachment; filename="+zipName+".zip");//获取输出流ServletOutputStream outputStream = response.getOutputStream();//创建缓冲区byte[] buffer = new byte[1024];int bytesRead = -1;//将文件写入输出流while ((bytesRead = inStream.read(buffer)) != -1) {outputStream.write(buffer, 0, bytesRead);}inStream.close();outputStream.flush();outputStream.close();
}
}