package com.kehwa.me17.util;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Enumeration;
/*
import java.util.zip.ZipEntry;
import java.util.zip.ZipFile;
import java.util.zip.ZipOutputStream;*/
import org.apache.tools.zip.ZipOutputStream;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* 用ZIP压缩和解压缩文件或目录
*/
public class CompressUtil {
private final static Log log = LogFactory.getLog(CompressUtil.class);
/**
* 压缩文件或者目录
* @param baseDirName 压缩的根目录
* @param fileName 根目录下待压缩的文件或文件夹名,
* 星号*表示压缩根目录下的全部文件。
* @param targetFileName 目标ZIP文件
*/
public static void zipFile(String baseDirName,String targetFileName) {
//检测根目录是否存在
if (baseDirName == null){
log.info("压缩失败,根目录不存在:" + baseDirName);
return;
}
File file = new File(baseDirName);
if (!file.exists()){
log.info("压缩失败,文件不存在:" + baseDirName);
return;
}
String baseDirPath = file.getAbsolutePath();
//目标文件
File targetFile = new File(targetFileName);
try{
//创建一个zip输出流来压缩数据并写入到zip文件
ZipOutputStream out =new ZipOutputStream(
new FileOutputStream(targetFile));
if (file.isFile()){
CompressUtil.fileToZip(baseDirPath, file, out);
} else {
CompressUtil.dirToZip(baseDirPath, file, out);
}
out.close();
log.info("压缩文件成功,目标文件名:" + targetFileName);
} catch (IOException e){
log.info("压缩失败:" + e);
e.printStackTrace();
}
}
/**
* 解压缩ZIP文件,将ZIP文件里的内容解压到targetDIR目录下
* @param zipName 待解压缩的ZIP文件名
* @param targetBaseDirName 目标目录
*/
public static void upzipFile(String zipFileName, String targetBaseDirName){
if (!targetBaseDirName.endsWith(File.separator)){
targetBaseDirName += File.separator;
}
try {
//根据ZIP文件创建ZipFile对象
ZipFile zipFile = new ZipFile(zipFileName);
ZipEntry entry = null;
String entryName = null;
String targetFileName = null;
byte[] buffer = new byte[4096];
int bytes_read;
//获取ZIP文件里所有的entry
Enumeration entrys = zipFile.getEntries();
//遍历所有entry
while (entrys.hasMoreElements()) {
entry = (ZipEntry)entrys.nextElement();
//获得entry的名字
entryName = entry.getName();
System.out.println(entryName);
targetFileName = targetBaseDirName + entryName;
if (entry.isDirectory()){
// 如果entry是一个目录,则创建目录
new File(targetFileName).mkdirs();
continue;
} else {
// 如果entry是一个文件,则创建父目录
new File(targetFileName).getParentFile().mkdirs();
}
File targetFile = new File(targetFileName);
FileOutputStream os = new FileOutputStream(targetFile);
InputStream is = zipFile.getInputStream(entry);
while ((bytes_read = is.read(buffer)) != -1){
os.write(buffer, 0, bytes_read);
}
//关闭流
os.close( );
is.close( );
}
log.info("解压缩文件成功!");
} catch (IOException err) {
System.err.println("解压缩文件失败: " + err);
}
}
/**
* 将目录压缩到ZIP输出流。
*/
private static void dirToZip(String baseDirPath, File dir,
ZipOutputStream out){
if (dir.isDirectory()){
//列出dir目录下所有文件
File[] files = dir.listFiles();
// 如果是空文件夹
if (files.length == 0){
ZipEntry entry = new ZipEntry(getEntryName(baseDirPath, dir));
// 存储目录信息
try {
out.putNextEntry(entry);
out.closeEntry();
} catch (IOException e) {
e.printStackTrace();
}
return;
}
for (int i=0; i<files.length; i++){
if (files[i].isFile()){
//如果是文件,调用fileToZip方法
CompressUtil.fileToZip(baseDirPath, files[i], out);
} else {
//如果是目录,递归调用
CompressUtil.dirToZip(baseDirPath, files[i], out);
}
}
}
}
/**
* 将文件压缩到ZIP输出流
*/
private static void fileToZip(String baseDirPath, File file,
ZipOutputStream out){
FileInputStream in = null;
ZipEntry entry = null;
// 创建复制缓冲区
byte[] buffer = new byte[4096];
int bytes_read;
if (file.isFile()){
try {
// 创建一个文件输入流
in = new FileInputStream(file);
// 做一个ZipEntry
entry = new ZipEntry(getEntryName(baseDirPath, file));
// 存储项信息到压缩文件
out.putNextEntry(entry);
// 复制字节到压缩文件
while((bytes_read = in.read(buffer)) != -1){
out.write(buffer, 0, bytes_read);
}
out.closeEntry();
in.close();
log.info("添加文件"
+ file.getAbsolutePath() + "被到ZIP文件中!");
} catch (IOException e){
e.printStackTrace();
}
}
}
/**
* 获取待压缩文件在ZIP文件中entry的名字。即相对于跟目录的相对路径名
* @param baseDirPath
* @param file
* @return
*/
private static String getEntryName(String baseDirPath, File file){
if (!baseDirPath.endsWith(File.separator)){
baseDirPath += File.separator;
}
String filePath = file.getAbsolutePath();
// 对于目录,必须在entry名字后面加上"/",表示它将以目录项存储。
if (file.isDirectory()){
filePath += "/";
}
int index = filePath.indexOf(baseDirPath);
return filePath.substring(index + baseDirPath.length());
}
public static void main(String[] args) {
//压缩C盘下的temp目录,压缩后的文件是C:/temp.zip
/*String baseDirName = "C:/";
String fileName = "temp/";
String zipFileName = "C:/temp.zip";*/
String baseDirName = "E:/plat";
String zipFileName = "E:/aa.zip";
//CompressUtil.upzipFile("e:/ME3.zip", "E:/");
CompressUtil.zipFile("e:/123", "e:/123.zip");
//将刚创建的ZIP文件解压缩到D盘的temp目录下
//System.out.println();
/*String zipFileName = "E:/plat.zip";
String fileName = "E:/plat1";
upzipFile(zipFileName,fileName);*/
//CompressUtil.zipFile(baseDirName, fileName, zipFileName);
}
}
补充说明一点,使用java.util.zip包时,是以UTF-8编码格式读取的文件名,因此在中文windows操作系统(Windows操作系统默认字符集为:GBK)中使用时会导致文件名解析错误,因此需要使用org.apache.tools.zip.ZipEntry和 org.apache.tools.zip.ZipOutputStream类来解决这一问题。