本文来源:
http://code.google.com/p/javawing/source/browse/trunk/Utils/src/org/javawing/util/ZipUtil.java?r=63
/**
* $Log: ZipCompress.java,v $
*/
package org.javawing.util;
import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Enumeration;
import java.util.zip.CRC32;
import java.util.zip.CheckedOutputStream;
import java.util.zip.Deflater;
import java.util.zip.ZipException;
import org.apache.log4j.Logger;
import org.apache.tools.zip.ZipEntry;
import org.apache.tools.zip.ZipFile;
import org.apache.tools.zip.ZipOutputStream;
/**
*
@brief
* <p><b>Zip file tool class</b></p>
*
* compress and decompress file
*
* <p><center>COPYRIGHT (C) 2009,RiseTek Systems Inc.All Rights Reserved.</center></p>
* @author TangJian
* @version 1.0
* @see
* @since 2009-12-10
*/
public class ZipUtil {
private static boolean isCreateSrcDir = false;//是否创建源目录
private static Logger log = Logger.getLogger(ZipUtil.class.getName());
/**
* @param args
* @throws IOException
*/
public static void main(String[] args) throws IOException {
String src = "e:/temp/src";//指定压缩源,可以是目录或文件
String decompressDir = "e:/temp/decompress";//解压路径
String archive = "e:/temp/test.wid";//压缩包路径
String comment = "Java Zip [email protected]";//压缩包注释
//----压缩文件或目录
writeByApacheZipOutputStream(src, archive, comment);
//----使用apace ZipFile读取压缩文件
readByApacheZipFile(archive, decompressDir);
// File file = new File(archive);
// readByApacheZipFile(file, decompressDir);
}
public static void writeByApacheZipOutputStream(String src, String archive,
String comment) throws FileNotFoundException, IOException {
//----压缩文件:
FileOutputStream f = new FileOutputStream(archive);
//使用指定校验和创建输出流
CheckedOutputStream csum = new CheckedOutputStream(f, new CRC32());
ZipOutputStream zos = new ZipOutputStream(csum);
//支持中文
zos.setEncoding("GBK");
BufferedOutputStream out = new BufferedOutputStream(zos);
//设置压缩包注释
zos.setComment(comment);
//启用压缩
zos.setMethod(ZipOutputStream.DEFLATED);
//压缩级别为最强压缩,但时间要花得多一点
zos.setLevel(Deflater.BEST_COMPRESSION);
File srcFile = new File(src);
if (!srcFile.exists() || (srcFile.isDirectory() && srcFile.list().length == 0)) {
throw new FileNotFoundException(
"File must exist and ZIP file must have at least one entry.");
}
//获取压缩源所在父目录
src = src.replaceAll("\\\\", "/");
String prefixDir = null;
if (srcFile.isFile()) {
prefixDir = src.substring(0, src.lastIndexOf("/") + 1);
} else {
prefixDir = (src.replaceAll("/$", "") + "/");
}
//如果不是根目录
if (prefixDir.indexOf("/") != (prefixDir.length() - 1) && isCreateSrcDir) {
prefixDir = prefixDir.replaceAll("[^/]+/$", "");
}
//开始压缩
writeRecursive(zos, out, srcFile, prefixDir);
out.close();
// 注:校验和要在流关闭后才准备,一定要放在流被关闭后使用
log.info("Checksum: " + csum.getChecksum().getValue());
}
public static void readByApacheZipFile(File file, String decompressDir)
throws IOException, FileNotFoundException, ZipException {
ZipFile zf = new ZipFile(file, "GBK");// 支持中文
readByApacheZipFile(decompressDir, zf);
}
/**
* 使用 org.apache.tools.zip.ZipFile 解压文件,它与 java 类库中的
* java.util.zip.ZipFile 使用方式是一致的,只不过多了设置编码方式的
* 接口。
*
* 注,apache 没有提供 ZipInputStream 类,所以只能使用它提供的ZipFile
* 来读取压缩文件。
* @param archive 压缩包路径
* @param decompressDir 解压路径
* @throws IOException
* @throws FileNotFoundException
* @throws ZipException
*/
public static void readByApacheZipFile(String archive, String decompressDir)
throws IOException, FileNotFoundException, ZipException {
ZipFile zf = new ZipFile(archive, "GBK");//支持中文
readByApacheZipFile(decompressDir, zf);
}
/**
* @brief
* main extract method
*
* @param decompressDir
* @param zf
* @throws FileNotFoundException
* @throws IOException
* @throws ZipException
*/
private static void readByApacheZipFile(String decompressDir, ZipFile zf)
throws FileNotFoundException, IOException, ZipException {
BufferedInputStream bi;
Enumeration<?> e = zf.getEntries();
while (e.hasMoreElements()) {
ZipEntry ze2 = (ZipEntry) e.nextElement();
String entryName = ze2.getName();
String path = decompressDir + "/" + entryName;
if (ze2.isDirectory()) {
log.info("正在创建解压目录 - " + entryName);
File decompressDirFile = new File(path);
if (!decompressDirFile.exists()) {
decompressDirFile.mkdirs();
}
} else {
log.info("正在创建解压文件 - " + entryName);
String fileDir = path.substring(0, path.lastIndexOf("/"));
File fileDirFile = new File(fileDir);
if (!fileDirFile.exists()) {
fileDirFile.mkdirs();
}
BufferedOutputStream bos = new BufferedOutputStream(
new FileOutputStream(decompressDir + "/" + entryName));
bi = new BufferedInputStream(zf.getInputStream(ze2));
byte[] readContent = new byte[1024];
int readCount = bi.read(readContent);
while (readCount != -1) {
//解压对文件二进制加密
bos.write(readContent, 0, readCount);
readCount = bi.read(readContent);
}
bos.close();
}
}
zf.close();
}
/**
* 递归压缩
*
* 使用 org.apache.tools.zip.ZipOutputStream 类进行压缩,它的好处就是支持中文路径,
* 而Java类库中的 java.util.zip.ZipOutputStream 压缩中文文件名时压缩包会出现乱码。
* 使用 apache 中的这个类与 java 类库中的用法是一新的,只是能设置编码方式了。
*
* @param zos
* @param bo
* @param srcFile
* @param prefixDir
* @throws IOException
* @throws FileNotFoundException
*/
private static void writeRecursive(ZipOutputStream zos, BufferedOutputStream bo,
File srcFile, String prefixDir) throws IOException, FileNotFoundException {
ZipEntry zipEntry;
String filePath = srcFile.getAbsolutePath().replaceAll("\\\\", "/").replaceAll(
"//", "/");
if (srcFile.isDirectory()) {
filePath = filePath.replaceAll("/$", "") + "/";
}
String entryName = filePath.replace(prefixDir, "").replaceAll("/$", "");
if (srcFile.isDirectory()) {
if (!"".equals(entryName)) {
log.info("正在创建目录 - " + srcFile.getAbsolutePath()
+ " entryName=" + entryName);
//如果是目录,则需要在写目录后面加上 /
zipEntry = new ZipEntry(entryName + "/");
zos.putNextEntry(zipEntry);
}
File srcFiles[] = srcFile.listFiles();
for (int i = 0; i < srcFiles.length; i++) {
writeRecursive(zos, bo, srcFiles[i], prefixDir);
}
} else {
log.info("正在写文件 - " + srcFile.getAbsolutePath() + " entryName="
+ entryName);
BufferedInputStream bi = new BufferedInputStream(new FileInputStream(srcFile));
//开始写入新的ZIP文件条目并将流定位到条目数据的开始处
zipEntry = new ZipEntry(entryName);
zos.putNextEntry(zipEntry);
byte[] buffer = new byte[1024];
int readCount = bi.read(buffer);
while (readCount != -1) {
bo.write(buffer, 0, readCount);
readCount = bi.read(buffer);
}
//注,在使用缓冲流写压缩文件时,一个条件完后一定要刷新一把,不
//然可能有的内容就会存入到后面条目中去了
bo.flush();
//文件读完后关闭
bi.close();
}
}
}