packageutils;importjava.io.BufferedInputStream;importjava.io.BufferedOutputStream;importjava.io.File;importjava.io.FileInputStream;importjava.io.FileOutputStream;importjava.io.IOException;importjava.io.InputStream;importjava.io.OutputStream;importjava.nio.file.FileVisitResult;importjava.nio.file.Files;importjava.nio.file.Path;importjava.nio.file.Paths;importjava.nio.file.SimpleFileVisitor;importjava.nio.file.attribute.BasicFileAttributes;importjava.util.Objects;importorg.apache.commons.compress.archivers.ArchiveEntry;importorg.apache.commons.compress.archivers.ArchiveInputStream;importorg.apache.commons.compress.archivers.ArchiveOutputStream;importorg.apache.commons.compress.archivers.zip.ZipArchiveEntry;importorg.apache.commons.compress.archivers.zip.ZipArchiveInputStream;importorg.apache.commons.compress.archivers.zip.ZipArchiveOutputStream;importorg.apache.commons.compress.utils.IOUtils;/*** 压缩文件夹工具类
**/
public classCompressUtils {/*** 压缩文件夹到指定zip文件
*
*@paramsrcDir 源文件夹
*@paramtargetFile 目标知道zip文件
*@throwsIOException IO异常,抛出给调用者处理*/
public static void zip(String srcDir, String targetFile) throwsIOException {try(
OutputStream outputStream= newFileOutputStream(targetFile);
) {
zip(srcDir, outputStream);
}
}/*** 压缩文件夹到指定输出流中,可以是本地文件输出流,也可以是web响应下载流
*
*@paramsrcDir 源文件夹
*@paramoutputStream 压缩后文件的输出流
*@throwsIOException IO异常,抛出给调用者处理*/
public static void zip(String srcDir, OutputStream outputStream) throwsIOException {try(
BufferedOutputStream bufferedOutputStream= newBufferedOutputStream(outputStream);
ArchiveOutputStream out= newZipArchiveOutputStream(bufferedOutputStream);
) {
Path start=Paths.get(srcDir);
Files.walkFileTree(start,new SimpleFileVisitor() {
@Overridepublic FileVisitResult preVisitDirectory(Path dir, BasicFileAttributes attrs) throwsIOException {
ArchiveEntry entry= newZipArchiveEntry(dir.toFile(), start.relativize(dir).toString());
out.putArchiveEntry(entry);
out.closeArchiveEntry();return super.preVisitDirectory(dir, attrs);
}
@Overridepublic FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throwsIOException {try(
InputStream input= newFileInputStream(file.toFile())
) {
ArchiveEntry entry= newZipArchiveEntry(file.toFile(), start.relativize(file).toString());
out.putArchiveEntry(entry);
IOUtils.copy(input, out);
out.closeArchiveEntry();
}return super.visitFile(file, attrs);
}
});
}
}/*** 解压zip文件到指定文件夹
*
*@paramzipFileName 源zip文件路径
*@paramdestDir 解压后输出路径
*@throwsIOException IO异常,抛出给调用者处理*/
public static void unzip(String zipFileName, String destDir) throwsIOException {try(
InputStream inputStream= newFileInputStream(zipFileName);
) {
unzip(inputStream, destDir);
}
}/*** 从输入流中获取zip文件,并解压到指定文件夹
*
*@paraminputStream zip文件输入流,可以是本地文件输入流,也可以是web请求上传流
*@paramdestDir 解压后输出路径
*@throwsIOException IO异常,抛出给调用者处理*/
public static void unzip(InputStream inputStream, String destDir) throwsIOException {try(
BufferedInputStream bufferedInputStream= newBufferedInputStream(inputStream);
ArchiveInputStream in= newZipArchiveInputStream(bufferedInputStream);
) {
ArchiveEntry entry= null;while (Objects.nonNull(entry =in.getNextEntry())) {if(in.canReadEntryData(entry)) {
File file=Paths.get(destDir, entry.getName()).toFile();if(entry.isDirectory()) {if (!file.exists()) {
file.mkdirs();
}
}else{try(
OutputStream out= newFileOutputStream(file);
) {
IOUtils.copy(in, out);
}
}
}else{
System.out.println(entry.getName());
}
}
}
}
}