/**
*
* @author LIUWEN276
* @Description: 压缩文件的通用工具类-采用org.apache.tools.zip.ZipOutputStream实现,较复杂。
*/
public class ZipCompressor {
private static Logger logger = LoggerFactory.getLogger(PvsnLogType.PVSN_COMMON.getLogName());
static final int BUFFER = 4096;
/**
* 将多个文件或目录压缩至指定的.zip文件中
* @param srcFileFullName 被压缩文件的文件全路径(包括文件名 目录或文件)
* @param threeDesKey 3DES 加密密钥
* @param targetZipFileFullName 目标压缩文件全路径(包括文件名)
*/
public static void encryptCompressExe(List<String> srcFileFullNameList,String threeDesKey,String targetZipFileFullName) {
List<File> files = new ArrayList<File>();
if(CollectionUtils.isEmpty(srcFileFullNameList) || StringUtils.isEmpty(threeDesKey) || StringUtils.isEmpty(targetZipFileFullName)){
logger.error("源文件路径,3DES密钥,目标zip文件路径不能为空{}", srcFileFullNameList,threeDesKey,targetZipFileFullName);
return;
}
for(String fileFullName : srcFileFullNameList){
File file = new File(fileFullName);
if (!file.exists()) {
throw new RuntimeException(fileFullName + "不存在!");
}
files.add(file);
}
ZipOutputStream out = null;
try {
// 设置流以加密模式输出
OutputStream encryptOutPutStream = KTDes3Tool.encryptMode(threeDesKey, new FileOutputStream(targetZipFileFullName));
// 设置流以压缩模式输出
out = new ZipOutputStream(encryptOutPutStream);
out.setEncoding(PvsnConstants.ENCODER_UTF8);
for(File singleFile : files){
compressByType(singleFile, out, "");
}
} catch (Exception e) {
logger.error("执行压缩操作时发生异常:{}" + e);
throw new RuntimeException(e);
}finally{
if(null != out){
try {
out.close();
} catch (IOException e) {
logger.error("关闭文件输出流时发生异常:{}" + e);
}
}
}
}
/**
* 将单个文件或某个目录压缩至指定的.zip文件中
* @param srcFileFullName 被压缩文件的文件全路径(包括文件名 目录或文件)
* @param threeDesKey 3DES 加密密钥
* @param targetZipFileFullName 目标压缩文件全路径(包括文件名)
*/
public static void encryptCompressExe(String srcFileFullName,String threeDesKey,String targetZipFileFullName) {
File file = new File(srcFileFullName);
if (!file.exists()) {
throw new RuntimeException(srcFileFullName + "不存在!");
}
ZipOutputStream out = null;
try {
// 设置流以加密模式输出
OutputStream encryptOutPutStream = KTDes3Tool.encryptMode(threeDesKey, new FileOutputStream(targetZipFileFullName));
// 设置流以压缩模式输出
out = new ZipOutputStream(encryptOutPutStream);
out.setEncoding(PvsnConstants.ENCODER_UTF8);
compressByType(file, out, "");
} catch (Exception e) {
logger.error("执行压缩操作时发生异常:{}" + e);
throw new RuntimeException(e);
}finally{
if(null != out){
try {
out.close();
} catch (IOException e) {
logger.error("关闭文件输出流时发生异常:{}" + e);
}
}
}
}
/**
* 判断是目录还是文件,根据类型(文件/文件夹)执行不同的压缩方法
*
* @param file
* @param out
* @param basedir
*/
private static void compressByType(File file, ZipOutputStream out, String basedir) {
/* 判断是目录还是文件 */
if (file.isDirectory()) {
logger.info("压缩:" + basedir + file.getName());
compressDirectory(file, out, basedir);
} else {
logger.info("压缩:" + basedir + file.getName());
compressFile(file, out, basedir);
}
}
/**
* 压缩一个目录
*
* @param dir
* @param out
* @param basedir
*/
private static void compressDirectory(File dir, ZipOutputStream out, String basedir) {
if (!dir.exists()) {
return;
}
File[] files = dir.listFiles();
for (int i = 0; i < files.length; i++) {
/* 递归 */
compressByType(files[i], out, basedir + dir.getName() + "/");
}
}
/**
* 压缩一个文件
*
* @param file
* @param out
* @param basedir
*/
private static void compressFile(File file, ZipOutputStream out, String basedir) {
if (!file.exists()) {
return;
}
try {
BufferedInputStream bis = new BufferedInputStream(new FileInputStream(file));
ZipEntry entry = new ZipEntry(basedir + file.getName());
out.putNextEntry(entry);
int count;
byte data[] = new byte[BUFFER];
while ((count = bis.read(data, 0, BUFFER)) != -1) {
out.write(data, 0, count);
}
bis.close();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}