public class ZipUtils { private static final Logger logger = LoggerFactory.getLogger(ZipUtils.class); private static final int BUFFER_SIZE = 2 * 1024; /** * zip文件生成及导出 * * @param srcDir 压缩文件夹路径 zip * @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败) */ public static void zipAndExport(String srcDir, String zipDir, boolean KeepDirStructure, HttpServletResponse response) { toZip(srcDir, KeepDirStructure); downloadZip(response, zipDir); deleteZip(zipDir); } /** * 压缩成ZIP * * @param srcDir 压缩文件夹路径 zip * @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败) */ public static void toZip(String srcDir, boolean KeepDirStructure) { ZipOutputStream zos = null; try { String zipPath = srcDir + SystemConstants.ZIP; FileOutputStream outputStream = new FileOutputStream(zipPath); zos = new ZipOutputStream(outputStream); File sourceFile = new File(srcDir); compress(sourceFile, zos, sourceFile.getName(), KeepDirStructure); } catch (Exception e) { logger.error(e.getMessage(), e); throw new ServiceException("文件压缩失败!"); } finally { // 无异常关闭流,它将无条件的关闭一个可被关闭的对象而不抛出任何异常。 CloseUtil.closeQuietly(zos); // 删除压缩之前的文件夹 FileUtils.del(srcDir); } } /** * 递归压缩方法 * * @param sourceFile 源文件 * @param zos zip输出流 * @param name 压缩后的名称 * @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败) */ private static void compress(File sourceFile, ZipOutputStream zos, String name, boolean KeepDirStructure) throws Exception { byte[] buf = new byte[BUFFER_SIZE]; if (sourceFile.isFile()) { // 向zip输出流中添加一个zip实体,构造器中name为zip实体的文件的名字 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) { // 需要保留原来的文件结构时,需要对空文件夹进行处理 if (KeepDirStructure) { // 空文件夹的处理 zos.putNextEntry(new ZipEntry(name + "/")); // 没有文件,不需要文件的copy zos.closeEntry(); } } else { for (File file : listFiles) { // 判断是否需要保留原来的文件结构 if (KeepDirStructure) { // 注意:file.getName()前面需要带上父文件夹的名字加一斜杠, // 不然最后压缩包中就不能保留原来的文件结构,即:所有文件都跑到压缩包根目录下了 compress(file, zos, name + "/" + file.getName(), true); } else { compress(file, zos, file.getName(), false); } } } } } /** * 下载ZIP压缩包 * * @param response 响应 */ public static void downloadZip(HttpServletResponse response, String zipDir) { //下载文件 try { File zipFile = new File(zipDir); response.setCharacterEncoding("utf-8"); response.setContentType("application/zip"); response.setHeader("Content-Disposition", "attachment;FileName=" + URLEncoder.encode(zipFile.getName(), "UTF-8")); response.addHeader("Content-Length", "" + zipFile.length()); ServletOutputStream out = response.getOutputStream(); int len; FileInputStream fis = new FileInputStream(zipFile); BufferedInputStream bis = new BufferedInputStream(fis); int count = fis.available(); byte[] buffer = new byte[count]; while ((len = bis.read(buffer)) > SystemConstants.NUMBER_ZERO) { out.write(buffer, SystemConstants.NUMBER_ZERO, len); out.flush(); } out.close(); fis.close(); bis.close(); response.flushBuffer(); } catch (IOException e) { logger.error(e.getMessage(), e); throw new ServiceException("文件读取失败!"); } } /** * 删除zip * * @param filePath filePath */ public static void deleteZip(String filePath) { File file = new File(filePath); // zip文件 判断 是否存在 if (file.getName().endsWith(".zip")) { FileUtils.del(file); } }