package com.cigna.hmc.groupinsurance.utils;import java.io.IOException;import org.apache.commons.lang.StringUtils;import net.lingala.zip4j.exception.ZipException;
import net.lingala.zip4j.io.ZipOutputStream;
import net.lingala.zip4j.model.ZipParameters;
import net.lingala.zip4j.util.Zip4jConstants;/*** * @author josnow* @date 2017年5月24日 下午3:12:31* @version 1.0.0* @desc zip工具集成原CompressUtil方法，增加了对内存文件和流文件的压缩以避免产生临时文件*/
public class ZipUtils extends CompressUtil {/*** * @desc 将内存文件写入zip内。注意：最后必须调用closeZipOutputStream关闭输出流，或者手动关闭* @auth josnow* @date 2017年5月24日 下午5:23:02* @param fileName*            文件名* @param data*            文件数据* @param password*            密码*/public static void addFileToZip(String fileName, byte[] data, String password, ZipOutputStream zipOutputStream)throws ZipException, IOException {if (StringUtils.isEmpty(fileName) || data == null || data.length == 0 || zipOutputStream == null) {throw new ZipException(new StringBuilder("参数异常,fileName=").append(fileName).append(",data=").append(data).append(",zipOutputStream=").append(zipOutputStream).toString());}ZipParameters zipParameters = new ZipParameters();zipParameters.setCompressionMethod(Zip4jConstants.COMP_DEFLATE); // 压缩方式zipParameters.setCompressionLevel(Zip4jConstants.DEFLATE_LEVEL_NORMAL); // 压缩级别
zipParameters.setFileNameInZip(fileName);if (StringUtils.isNotBlank(password)) {zipParameters.setEncryptFiles(true);zipParameters.setEncryptionMethod(Zip4jConstants.ENC_METHOD_STANDARD);zipParameters.setPassword(password.toCharArray());}// 源文件是否为外部流，true表示使用内存文件而非本地存储文件zipParameters.setSourceExternalStream(true);zipOutputStream.putNextEntry(null, zipParameters);zipOutputStream.write(data);zipOutputStream.closeEntry();}/*** * @desc 将内存文件写入zip内。注意：最后必须调用closeZipOutputStream关闭输出流，或者手动关闭* @auth josnow* @date 2017年5月24日 下午5:46:02* @param fileName*            文件名* @param data*            文件数据*/public static void addFileToZip(String fileName, byte[] data, ZipOutputStream zipOutputStream)throws ZipException, IOException {addFileToZip(fileName, data, null, zipOutputStream);}/*** * @desc 将内存文件写入zip内。注意：最后必须调用closeZipOutputStream关闭输出流，或者手动关闭* @auth josnow* @date 2017年5月25日 上午11:08:56* @param zipParameters*            zip参数* @param data*            文件数据* @param zipOutputStream*            输出流*/public static void addFileToZip(ZipParameters zipParameters, byte[] data, ZipOutputStream zipOutputStream)throws ZipException, IOException {if (zipParameters == null || data == null || data.length == 0 || zipOutputStream == null) {throw new ZipException(new StringBuilder("参数异常,zipParameters=").append(zipParameters).append(",data=").append(data).append(",zipOutputStream=").append(zipOutputStream).toString());}zipOutputStream.putNextEntry(null, zipParameters);zipOutputStream.write(data);zipOutputStream.closeEntry();}/*** * @desc 关闭流* @auth josnow* @date 2017年5月25日 上午11:16:01* @param zipOutputStream*            输出流*/public static void closeZipOutputStream(ZipOutputStream zipOutputStream) throws IOException, ZipException {if (zipOutputStream == null) {return;}zipOutputStream.finish();zipOutputStream.close();}// public static void main(String[] args) throws Exception {// ByteArrayOutputStream byteArrayOutputStream = new// ByteArrayOutputStream(1024);// ZipOutputStream zipOutputStream = new// ZipOutputStream(byteArrayOutputStream);//// byte[] b = "德玛西亚哦哦奥法额外发撒旦；联发科就；".getBytes();//// addFileToZip("你好大猪头.txt", b, zipOutputStream);// addFileToZip("你就是大肥猪.txt", b, "123", zipOutputStream);//// closeZipOutputStream(zipOutputStream);//// byte[] zipData = byteArrayOutputStream.toByteArray();// System.out.println(new String(zipData));//// new FileOutputStream("D:\\nima.zip").write(zipData);// }

}

package com.cigna.hmc.groupinsurance.utils;import java.io.File;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;import org.apache.commons.lang.StringUtils;import net.lingala.zip4j.core.ZipFile;
import net.lingala.zip4j.exception.ZipException;
import net.lingala.zip4j.model.FileHeader;
import net.lingala.zip4j.model.ZipParameters;
import net.lingala.zip4j.util.Zip4jConstants;/*** ZIP压缩文件操作工具类 支持密码 依赖zip4j开源项目(http://www.lingala.net/zip4j/) 版本1.3.1* * @author ninemax*/
public class CompressUtil {/*** 使用给定密码解压指定的ZIP压缩文件到指定目录* <p>* 如果指定目录不存在,可以自动创建,不合法的路径将导致异常被抛出* * @param zip*            指定的ZIP压缩文件* @param dest*            解压目录* @param passwd*            ZIP文件的密码* @return 解压后文件数组* @throws ZipException*             压缩文件有损坏或者解压缩失败抛出*/public static File[] unzip(String zip, String dest, String passwd) throws ZipException {File zipFile = new File(zip);return unzip(zipFile, dest, passwd);}/*** 使用给定密码解压指定的ZIP压缩文件到当前目录* * @param zip*            指定的ZIP压缩文件* @param passwd*            ZIP文件的密码* @return 解压后文件数组* @throws ZipException*             压缩文件有损坏或者解压缩失败抛出*/public static File[] unzip(String zip, String passwd) throws ZipException {File zipFile = new File(zip);File parentDir = zipFile.getParentFile();return unzip(zipFile, parentDir.getAbsolutePath(), passwd);}/*** 使用给定密码解压指定的ZIP压缩文件到指定目录* <p>* 如果指定目录不存在,可以自动创建,不合法的路径将导致异常被抛出* * @param zip*            指定的ZIP压缩文件* @param dest*            解压目录* @param passwd*            ZIP文件的密码* @return 解压后文件数组* @throws ZipException*             压缩文件有损坏或者解压缩失败抛出*/public static File[] unzip(File zipFile, String dest, String passwd) throws ZipException {ZipFile zFile = new ZipFile(zipFile);zFile.setFileNameCharset("GBK");if (!zFile.isValidZipFile()) {throw new ZipException("压缩文件不合法,可能被损坏.");}File destDir = new File(dest);if (destDir.isDirectory() && !destDir.exists()) {destDir.mkdir();}if (zFile.isEncrypted()) {zFile.setPassword(passwd.toCharArray());}zFile.extractAll(dest);List<FileHeader> headerList = zFile.getFileHeaders();List<File> extractedFileList = new ArrayList<File>();for (FileHeader fileHeader : headerList) {if (!fileHeader.isDirectory()) {extractedFileList.add(new File(destDir, fileHeader.getFileName()));}}File[] extractedFiles = new File[extractedFileList.size()];extractedFileList.toArray(extractedFiles);return extractedFiles;}/*** 压缩指定文件到当前文件夹* * @param src*            要压缩的指定文件* @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.*/public static String zip(String src) {return zip(src, null);}/*** 使用给定密码压缩指定文件或文件夹到当前目录* * @param src*            要压缩的文件* @param passwd*            压缩使用的密码* @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.*/public static String zip(String src, String passwd) {return zip(src, null, passwd);}/*** 使用给定密码压缩指定文件或文件夹到当前目录* * @param src*            要压缩的文件* @param dest*            压缩文件存放路径* @param passwd*            压缩使用的密码* @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.*/public static String zip(String src, String dest, String passwd) {return zip(src, dest, true, passwd);}/*** 使用给定密码压缩指定文件或文件夹到指定位置.* <p>* dest可传最终压缩文件存放的绝对路径,也可以传存放目录,也可以传null或者"".<br />* 如果传null或者""则将压缩文件存放在当前目录,即跟源文件同目录,压缩文件名取源文件名,以.zip为后缀;<br />* 如果以路径分隔符(File.separator)结尾,则视为目录,压缩文件名取源文件名,以.zip为后缀,否则视为文件名.* * @param src*            要压缩的文件或文件夹路径* @param dest*            压缩文件存放路径* @param isCreateDir*            是否在压缩文件里创建目录,仅在压缩文件为目录时有效.<br />*            如果为false,将直接压缩目录下文件到压缩文件.* @param passwd*            压缩使用的密码* @return 最终的压缩文件存放的绝对路径,如果为null则说明压缩失败.*/public static String zip(String src, String dest, boolean isCreateDir, String passwd) {File srcFile = new File(src);dest = buildDestinationZipFilePath(srcFile, dest);ZipParameters parameters = new ZipParameters();parameters.setCompressionMethod(Zip4jConstants.COMP_DEFLATE); // 压缩方式parameters.setCompressionLevel(Zip4jConstants.DEFLATE_LEVEL_NORMAL); // 压缩级别if (!StringUtils.isEmpty(passwd)) {parameters.setEncryptFiles(true);parameters.setEncryptionMethod(Zip4jConstants.ENC_METHOD_STANDARD); // 加密方式
            parameters.setPassword(passwd.toCharArray());}try {ZipFile zipFile = new ZipFile(dest);if (srcFile.isDirectory()) {// 如果不创建目录的话,将直接把给定目录下的文件压缩到压缩文件,即没有目录结构if (!isCreateDir) {File[] subFiles = srcFile.listFiles();ArrayList<File> temp = new ArrayList<File>();Collections.addAll(temp, subFiles);zipFile.addFiles(temp, parameters);return dest;}zipFile.addFolder(srcFile, parameters);} else {zipFile.addFile(srcFile, parameters);}return dest;} catch (ZipException e) {e.printStackTrace();}return null;}/*** 构建压缩文件存放路径,如果不存在将会创建 传入的可能是文件名或者目录,也可能不传,此方法用以转换最终压缩文件的存放路径* * @param srcFile*            源文件* @param destParam*            压缩目标路径* @return 正确的压缩文件存放路径*/private static String buildDestinationZipFilePath(File srcFile, String destParam) {if (StringUtils.isEmpty(destParam)) {if (srcFile.isDirectory()) {destParam = srcFile.getParent() + File.separator + srcFile.getName() + ".zip";} else {String fileName = srcFile.getName().substring(0, srcFile.getName().lastIndexOf("."));destParam = srcFile.getParent() + File.separator + fileName + ".zip";}} else {createDestDirectoryIfNecessary(destParam); // 在指定路径不存在的情况下将其创建出来if (destParam.endsWith(File.separator)) {String fileName = "";if (srcFile.isDirectory()) {fileName = srcFile.getName();} else {fileName = srcFile.getName().substring(0, srcFile.getName().lastIndexOf("."));}destParam += fileName + ".zip";}}return destParam;}/*** 在必要的情况下创建压缩文件存放目录,比如指定的存放路径并没有被创建* * @param destParam*            指定的存放路径,有可能该路径并没有被创建*/private static void createDestDirectoryIfNecessary(String destParam) {File destDir = null;if (destParam.endsWith(File.separator)) {destDir = new File(destParam);} else {destDir = new File(destParam.substring(0, destParam.lastIndexOf(File.separator)));}if (!destDir.exists()) {destDir.mkdirs();}}}