tar.gz压缩与解压

思路

• 压缩

1. 将多个或单个文件打包为tar文件
2. 将Tar文件压缩为gzip文件, 输出到指定的路径

IO流	类型	输入/输出
FileInputStream	节点流	输入
BufferedInputStream	处理流	输入
FileOutputStream	节点流	输出
BufferedOutputStream	处理流	输出
TarOutputStream	处理流	输出
GZIPOutputStream	处理流	输出

• 解压

1. 读取文件转换为GZIPInputStream流对象
2. 将GZIPInputStream流对象转换为TarInputStream流对象, 输出到指定路径

IO流	类型	输入/输出
FileInputStream	节点流	输入
BufferedInputStream	处理流	输入
GZIPInputStream	处理流	输入
TarInputStream	处理流	输入
FileOutputStream	节点流	输出

实现

package com.ctid.mainlib.system.util.datehandle;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.List;
import java.util.zip.GZIPInputStream;
import java.util.zip.GZIPOutputStream;

import org.apache.activemq.console.command.store.tar.TarEntry;
import org.apache.activemq.console.command.store.tar.TarInputStream;
import org.apache.activemq.console.command.store.tar.TarOutputStream;
import org.apache.commons.io.IOUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import com.ctid.core.exception.ServiceException;
import com.ctid.mainlib.system.aspect.PrintRunTime;
import com.ctid.mainlib.system.beans.Verify;

/**
 * @功能描述 TarGZIP解压缩工具类
 * @author 索睿君
 * @date 2019年7月1日17:02:31
 */
public class TarGZIP {
	private static final Log Log = LogFactory.getLog(TarGZIP.class);
	/**
	 * @功能描述 解压tar.gz文件
	 * @param targzFile tar.gz压缩文件
	 * @param outPath 存放解压后文件的目录
	 * @return 返回结果
	 * @throws ServiceException 
	 */
	public static void unpack(File targzFile, String outPath) throws ServiceException {
		//1. 验证参数, 初始化输出路径
		if(targzFile == null || !targzFile.isFile() || !Verify.isEmpty(outPath)){
			throw new ServiceException("文件解压缩执行异常, 非法参数!");
		}
		long startTime = System.currentTimeMillis();
		//2. 读取tar.gz文件转换为tar文件
		try (FileInputStream fileInputStream = new FileInputStream(targzFile);
				BufferedInputStream bins = new BufferedInputStream(fileInputStream);
				GZIPInputStream gzipInputStream = new GZIPInputStream(bins);
				TarInputStream tarIn = new TarInputStream(gzipInputStream, 1024 * 2)) {
			//3. 迭代tar文件集合, 解压文件
			for (TarEntry entry = tarIn.getNextEntry(); entry != null; entry = tarIn.getNextEntry()){
				File targetFileName = new File(outPath + "/" + entry.getName());
				IOUtils.copy(tarIn, new FileOutputStream(targetFileName));
			}
			long endTime = System.currentTimeMillis();
			Log.info("文件["+targzFile+"]解压执行完毕, 耗时:" + (endTime - startTime) + "ms");
		} catch (Exception e) {
			throw new ServiceException("[" + targzFile + "] 解压执行异常, " + e.toString());
		}
	}

	/**
	 * @功能描述 压缩tar.gz 文件
	 * @param resourceList 源文件集合
	 * @param outPath 目标文件
	 * @return 返回压缩结果
	 * @throws Exception
	 */
	@PrintRunTime(function="压缩tar.gz文件")
	public static void packet(List<File> resourceList, String outPath) throws Exception {
			//1. 参数验证, 初始化输出路径
			if(resourceList == null || resourceList.size() < 1 || !Verify.isEmpty(outPath)){
				throw new ServiceException("文件压缩执行异常, 非法参数!");
			}
			long startTime = System.currentTimeMillis();
			// 2. 迭代源文件集合, 将文件打包为Tar
			try (FileOutputStream fileOutputStream = new FileOutputStream(outPath+".tmp");
					BufferedOutputStream bufferedOutput = new BufferedOutputStream(fileOutputStream);
					TarOutputStream tarOutputStream = new TarOutputStream(bufferedOutput);) {
				for (File resourceFile : resourceList) {
					if(!resourceFile.isFile()){
						continue;
					}
					try(FileInputStream fileInputStream = new FileInputStream(resourceFile);
							BufferedInputStream bufferedInput = new BufferedInputStream(fileInputStream);){
						TarEntry entry = new TarEntry(new File(resourceFile.getName()));
						entry.setSize(resourceFile.length());
						tarOutputStream.putNextEntry(entry);
						IOUtils.copy(bufferedInput, tarOutputStream);
					} catch (Exception e) {
						throw new ServiceException("文件["+resourceFile+"]压缩执行异常, 嵌套异常: \n" + e.toString());
					}finally {
						tarOutputStream.closeEntry();
					}
				}
			} catch (Exception e) {
				Files.delete(Paths.get(outPath+".tmp"));
				throw new ServiceException("文件压缩至["+outPath+"]执行异常, 嵌套异常: \n" + e.toString());
			}
			//3. 读取打包好的Tar临时文件文件, 使用GZIP方式压缩
			try (FileInputStream fileInputStream = new FileInputStream(outPath+".tmp");
					BufferedInputStream bufferedInput = new BufferedInputStream(fileInputStream);
					FileOutputStream fileOutputStream = new FileOutputStream(outPath);
					GZIPOutputStream gzipOutputStream = new GZIPOutputStream(fileOutputStream);
					BufferedOutputStream bufferedOutput = new BufferedOutputStream(gzipOutputStream);
					) {
				byte[] cache = new byte[1024];
				for (int index = bufferedInput.read(cache); index != -1; index = bufferedInput.read(cache)) {
					bufferedOutput.write(cache,0,index);
				}
				long endTime = System.currentTimeMillis();
				Log.info("文件["+outPath+"]压缩执行完毕, 耗时:" + (endTime - startTime) + "ms");
			} catch (Exception e) {
				throw new ServiceException("文件压缩至["+outPath+"]执行异常, 嵌套异常: \n" + e.toString());
			}finally {
				Files.delete(Paths.get(outPath+".tmp"));
			}
		}
}

总结

1. 压缩文件那一部分可能写的不对, 之前打算直接将文件转换为TarOutputStream流, 再将 TarOutputStream流 转换为GZIPOutputStream流输出, 但是在TarOutputStream流转换为GZIPOutputStream流的时候会报错, 无奈只好将文件转为tar生成到本地, 再读取本地文件以GZIPOutputStream流输出
2. 如果有更好的方式来处理tar.gz文件, 千万要在下边留言o(￣︶￣)o