本文介绍Java压缩/解压.zip、.tar.gz、.tar.bz2的方式。
- 对于zip文件:使用
java.util.zip.ZipEntry
和java.util.zip.ZipFile
,通过设置Charset
为StandardCharsets.UTF_8
支持中文。 - 对于.tar.gz、tgz文件:可以看做先用tar打包,再使用gz进行压缩。使用
commons-compress
包的TarArchiveInputStream
和GzipCompressorInputStream
。 - 对于.tar.bz2文件:可以看做先用tar打包,再使用bz2进行压缩。使用
commons-compress
包的TarArchiveInputStream
和BZip2CompressorInputStream
。
在这里有个问题如果使用TarInputStream
搭配jdk的 GZIPInputStream
会产生乱码。而使用commons-compress
包的TarArchiveInputStream
和GzipCompressorInputStream
则可解决乱码问题。
代码如下:
public class ZipUtil {
private static final Logger LOG = LoggerFactory.getLogger(ZipUtil.class);
private static final int BUFFER_SIZE = 1024 * 100;
private ZipUtil() {
}
/**
* 私有函数将文件集合压缩成tar包后返回
*
* @param files 要压缩的文件集合
* @param target tar 输出流的目标文件
* @return File 指定返回的目标文件
*/
private static File pack(List<File> files, File target) throws IOException{
try (FileOutputStream fos = new FileOutputStream(target)) {
try (BufferedOutputStream bos = new BufferedOutputStream(fos, BUFFER_SIZE)) {
try (TarArchiveOutputStream taos = new TarArchiveOutputStream(bos)) {
//解决文件名过长问题
taos.setLongFileMode(TarArchiveOutputStream.LONGFILE_GNU);
for (File file : files) {
taos.putArchiveEntry(new TarArchiveEntry(file));
try (FileInputStream fis = new FileInputStream(file)) {
IOUtils.copy(fis, taos);
taos.closeArchiveEntry();
}
}
}
}
}
return target;
}
/**
* 将tar压缩成tar.gz文件
*
* @param list
* @param outPutPath
* @param fileName
*/
public static File compress(List<File> list, String outPutPath, String fileName) throws IOException {
File outPutFile = new File(outPutPath + File.separator + fileName + ".tar.gz");
File tempTar = new File("temp.tar");
try (FileInputStream fis = new FileInputStream(pack(list, tempTar))) {
try (BufferedInputStream bis = new BufferedInputStream(fis, BUFFER_SIZE)) {
try (FileOutputStream fos = new FileOutputStream(outPutFile)) {
try (GZIPOutputStream gzp = new GZIPOutputStream(fos)) {
int count;
byte[] data = new byte[BUFFER_SIZE];
while ((count = bis.read(data, 0, BUFFER_SIZE)) != -1) {
gzp.write(data, 0, count);
}
}
}
}
}
try {
Files.deleteIfExists(tempTar.toPath());
} catch (IOException e) {
e.printStackTrace();
}
return outPutFile;
}
public static boolean decompress(String filePath, String outputDir, boolean isDeleted) {
File file = new File(filePath);
if (!file.exists()) {
LOG.error("decompress file not exist.");
return false;
}
try {
if (filePath.endsWith(".zip")) {
unZip(file, outputDir);
}
if (filePath.endsWith(".tar.gz") || filePath.endsWith(".tgz")) {
decompressTarGz(file, outputDir);
}
if (filePath.endsWith(".tar.bz2")) {
decompressTarBz2(file, outputDir);
}
filterFile(new File(outputDir));
if (isDeleted) {
FileUtils.deleteQuietly(file);
}
return true;
} catch (IOException e) {
LOG.error("decompress occur error.");
}
return false;
}
/**
* 解压 .zip 文件
*
* @param file 要解压的zip文件对象
* @param outputDir 要解压到某个指定的目录下
* @throws IOException
*/
public static void unZip(File file, String outputDir) throws IOException {
try (ZipFile zipFile = new ZipFile(file, StandardCharsets.UTF_8)) {
//创建输出目录
createDirectory(outputDir, null);
Enumeration<?> enums = zipFile.entries();
while (enums.hasMoreElements()) {
ZipEntry entry = (ZipEntry) enums.nextElement();
if (entry.isDirectory()) {
//创建空目录
createDirectory(outputDir, entry.getName());
} else {
try (InputStream in = zipFile.getInputStream(entry)) {
try (OutputStream out = new FileOutputStream(
new File(outputDir + File.separator + entry.getName()))) {
writeFile(in, out);
}
}
}
}
}
}
public static void decompressTarGz(File file, String outputDir) throws IOException {
try (TarArchiveInputStream tarIn = new TarArchiveInputStream(
new GzipCompressorInputStream(
new BufferedInputStream(
new FileInputStream(file))))) {
//创建输出目录
createDirectory(outputDir, null);
TarArchiveEntry entry = null;
while ((entry = tarIn.getNextTarEntry()) != null) {
//是目录
if (entry.isDirectory()) {
//创建空目录
createDirectory(outputDir, entry.getName());
} else {
//是文件
try (OutputStream out = new FileOutputStream(
new File(outputDir + File.separator + entry.getName()))) {
writeFile(tarIn, out);
}
}
}
}
}
/**
* 解压缩tar.bz2文件
*
* @param file 压缩包文件
* @param outputDir 目标文件夹
*/
public static void decompressTarBz2(File file, String outputDir) throws IOException {
try (TarArchiveInputStream tarIn =
new TarArchiveInputStream(
new BZip2CompressorInputStream(
new FileInputStream(file)))) {
createDirectory(outputDir, null);
TarArchiveEntry entry;
while ((entry = tarIn.getNextTarEntry()) != null) {
if (entry.isDirectory()) {
createDirectory(outputDir, entry.getName());
} else {
try (OutputStream out = new FileOutputStream(
new File(outputDir + File.separator + entry.getName()))) {
writeFile(tarIn, out);
}
}
}
}
}
/**
* 写文件
*
* @param in
* @param out
* @throws IOException
*/
public static void writeFile(InputStream in, OutputStream out) throws IOException {
int length;
byte[] b = new byte[BUFFER_SIZE];
while ((length = in.read(b)) != -1) {
out.write(b, 0, length);
}
}
/**
* 创建目录
*
* @param outputDir
* @param subDir
*/
public static void createDirectory(String outputDir, String subDir) {
File file = new File(outputDir);
//子目录不为空
if (!(subDir == null || subDir.trim().equals(""))) {
file = new File(outputDir + File.separator + subDir);
}
if (!file.exists()) {
if (!file.getParentFile().exists()) {
file.getParentFile().mkdirs();
}
file.mkdirs();
}
}
/**
* 删除Mac压缩再解压产生的 __MACOSX 文件夹和 .开头的其他文件
*
* @param filteredFile
*/
public static void filterFile(File filteredFile) {
if (filteredFile != null) {
File[] files = filteredFile.listFiles();
for (File file : files) {
if (file.getName().startsWith(".") ||
(file.isDirectory() && file.getName().equals("__MACOSX"))) {
FileUtils.deleteQuietly(file);
}
}
}
}
}
参考文档: