package com.customcode.util;
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.zip.ZipEntry;
import java.util.zip.ZipInputStream;
import java.util.zip.ZipOutputStream;
/**
* Zip压缩/解压缩工具类
* 实现对目标路径及其子路径下的所有文件及空目录的压缩
* 参考网上若干种实现,并修改其bug
* @Author:YUYB
**/
public class ZipUtil {
/** 缓冲器大小 */
private static final int BUFFER = 512;
/**
* 取的给定源目录下的所有文件及空的子目录
* 递归实现
*
* @param srcFile
*
* @return
*/
private static List<File> getAllFiles(File srcFile) {
List<File> fileList = new ArrayList<File>();
File[] tmp = srcFile.listFiles();
for (int i = 0; i < tmp.length; i++) {
if (tmp[i].isFile()) {
fileList.add(tmp[i]);
System.out.println("add file: "+tmp[i].getName());
}
if (tmp[i].isDirectory()) {
if (tmp[i].listFiles().length!=0){//若不是空目录,则递归添加其下的目录和文件
fileList.addAll(getAllFiles(tmp[i]));
}
else{//若是空目录,则添加这个目录到fileList
fileList.add(tmp[i]);
System.out.println("add empty dir: "+tmp[i].getName());
}
}
} // end for
return fileList;
}
/**
* 取相对路径
* 依据文件名和压缩源路径得到文件在压缩源路径下的相对路径
*
* @param dirPath 压缩源路径
* @param file
*
* @return 相对路径
*/
private static String getRelativePath(String dirPath, File file) {
File dir = new File(dirPath);
String relativePath = file.getName();
while (true) {
file = file.getParentFile();
if (file == null) {
break;
}
if (file.equals(dir)) {
break;
} else {
relativePath = file.getName() + "/" + relativePath;
}
} // end while
return relativePath;
}
/**
* 创建文件
* 根据压缩包内文件名和解压缩目的路径,创建解压缩目标文件,
* 生成中间目录
* @param dstPath 解压缩目的路径
* @param fileName 压缩包内文件名
*
* @return 解压缩目标文件
*
* @throws IOException
*/
private static File createFile(String dstPath, String fileName) throws IOException {
String[] dirs = fileName.split("/");//将文件名的各级目录分解
File file = new File(dstPath);
if (dirs.length > 1) {//文件有上级目录
for (int i = 0; i < dirs.length - 1; i++) {
file = new File(file, dirs[i]);//依次创建文件对象知道文件的上一级目录
}
if (!file.exists()) {
file.mkdirs();//文件对应目录若不存在,则创建
System.out.println("mkdirs: " + file.getCanonicalPath());
}
file = new File(file, dirs[dirs.length - 1]);//创建文件
return file;
} else {
if (!file.exists()) {
file.mkdirs();//若目标路径的目录不存在,则创建
System.out.println("mkdirs: " + file.getCanonicalPath());
}
file = new File(file, dirs[0]);//创建文件
return file;
}
}
/**
* 解压缩方法
*
*
* @param zipFileName 压缩文件名
* @param dstPath 解压目标路径
*
* @return
*/
public static boolean unzip(String zipFileName, String dstPath) {
System.out.println("zip uncompressing...");
System.out.println("zip zipFileName 压缩文件名..."+zipFileName);
System.out.println("zip dstPath 解压目标路径..."+dstPath);
try {
ZipInputStream zipInputStream = new ZipInputStream(new FileInputStream(zipFileName));
ZipEntry zipEntry = null;
byte[] buffer = new byte[BUFFER];//缓冲器
int readLength = 0;//每次读出来的长度
while ((zipEntry = zipInputStream.getNextEntry()) != null) {
if (zipEntry.isDirectory()) {//若是zip条目目录,则需创建这个目录
File dir = new File(dstPath + "/" + zipEntry.getName());
if (!dir.exists()) {
dir.mkdirs();
System.out.println("mkdirs: " + dir.getCanonicalPath());
continue;//跳出
}
}
File file = createFile(dstPath, zipEntry.getName());//若是文件,则需创建该文件
System.out.println("file created: " + file.getCanonicalPath());
OutputStream outputStream = new FileOutputStream(file);
while ((readLength = zipInputStream.read(buffer, 0, BUFFER)) != -1) {
outputStream.write(buffer, 0, readLength);
}
outputStream.close();
System.out.println("file uncompressed: " + file.getCanonicalPath());
} // end while
} catch (FileNotFoundException e) {
System.out.println(e.getMessage());
e.printStackTrace();
System.out.println("unzip fail!");
return false;
} catch (IOException e) {
System.out.println(e.getMessage());
e.printStackTrace();
System.out.println("unzip fail!");
return false;
}
System.out.println("unzip success!");
return true;
}
/**
* 压缩方法
* (可以压缩空的子目录)
* @param srcPath 压缩源路径
* @param zipFileName 目标压缩文件
*
* @return
*/
public static boolean zip(String srcPath, String zipFileName) {
System.out.println("zip compressing...");
File srcFile = new File(srcPath);
List<File> fileList = getAllFiles(srcFile);//所有要压缩的文件
byte[] buffer = new byte[BUFFER];//缓冲器
ZipEntry zipEntry = null;
int readLength = 0;//每次读出来的长度
try {
ZipOutputStream zipOutputStream = new ZipOutputStream(new FileOutputStream(zipFileName));
for (File file : fileList) {
if (file.isFile()){//若是文件,则压缩这个文件
zipEntry = new ZipEntry(getRelativePath(srcPath, file));
zipEntry.setSize(file.length());
zipEntry.setTime(file.lastModified());
zipOutputStream.putNextEntry(zipEntry);
InputStream inputStream = new BufferedInputStream(new FileInputStream(file));
while ((readLength = inputStream.read(buffer, 0, BUFFER)) != -1) {
zipOutputStream.write(buffer, 0, readLength);
}
inputStream.close();
System.out.println("file compressed: " + file.getCanonicalPath());
}else {//若是目录(即空目录)则将这个目录写入zip条目
zipEntry = new ZipEntry(getRelativePath(srcPath, file)+"/");
zipOutputStream.putNextEntry(zipEntry);
System.out.println("dir compressed: " + file.getCanonicalPath()+"/");
}
} // end for
zipOutputStream.close();
} catch (FileNotFoundException e) {
System.out.println(e.getMessage());
e.printStackTrace();
System.out.println("zip fail!");
return false;
} catch (IOException e) {
System.out.println(e.getMessage());
e.printStackTrace();
System.out.println("zip fail!");
return false;
}
System.out.println("zip success!");
return true;
}
public static String unzips(String filepath) throws Exception {
File file = new File(filepath);
ZipInputStream zis = new ZipInputStream(new BufferedInputStream(new FileInputStream(file)));
File fout = null;
String parent = file.getParent();
String ofilename = "";
ZipEntry entry;
while((entry = zis.getNextEntry())!=null && !entry.isDirectory()){
if(entry.getName().contains(":")){
fout = new File(parent, entry.getName().substring(entry.getName().lastIndexOf("\\"), entry.getName().length()));
ofilename = parent +"\\"+ entry.getName().substring(entry.getName().lastIndexOf("\\"), entry.getName().length());
}else{
fout = new File(parent, entry.getName());
ofilename = parent +"\\"+ entry.getName();
}
System.out.println("ofilename ofilename!============"+ofilename);
if(!fout.exists()){
(new File(fout.getParent())).mkdirs();
}
BufferedOutputStream bos = new BufferedOutputStream(new FileOutputStream(fout));
int b = -1;
byte[] buffer = new byte[1024];
while((b = zis.read(buffer))!= -1){
bos.write(buffer, 0, b);
}
bos.close();
}
zis.close();
return ofilename;
}
// public static void unrar(String sourceRar,String destDir) throws Exception{
// Archive a = null;
// FileOutputStream fos = null;
// try{
// a = new Archive(new File(sourceRar));
// FileHeader fh = a.nextFileHeader();
// while(fh!=null){
// if(!fh.isDirectory()){
// //1 根据不同的操作系统拿到相应的 destDirName 和 destFileName
// String compressFileName = fh.getFileNameString().trim();
// String destFileName = "";
// String destDirName = "";
// //非windows系统
// if(File.separator.equals("/")){
// destFileName = destDir + compressFileName.replaceAll("\\\\", "/");
// destDirName = destFileName.substring(0, destFileName.lastIndexOf("/"));
// //windows系统
// }else{
// destFileName = destDir + compressFileName.replaceAll("/", "\\\\");
// destDirName = destFileName.substring(0, destFileName.lastIndexOf("\\"));
// }
// //2创建文件夹
// File dir = new File(destDirName);
// if(!dir.exists()||!dir.isDirectory()){
// dir.mkdirs();
// }
// //3解压缩文件
// fos = new FileOutputStream(new File(destFileName));
// a.extractFile(fh, fos);
// fos.close();
// fos = null;
// }
// fh = a.nextFileHeader();
// }
// a.close();
// a = null;
// }catch(Exception e){
// throw e;
// }finally{
// if(fos!=null){
// try{fos.close();fos=null;}catch(Exception e){e.printStackTrace();}
// }
// if(a!=null){
// try{a.close();a=null;}catch(Exception e){e.printStackTrace();}
// }
// }
// }
}
Zip压缩/解压缩工具类
最新推荐文章于 2022-08-30 19:52:06 发布