• Home
• io.github.whaleal
• icefrog-extra
• 1.0.0
• source code
• CompressUtil.java

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

com.whaleal.icefrog.extra.compress.CompressUtil Maven / Gradle / Ivy

package com.whaleal.icefrog.extra.compress;

import com.whaleal.icefrog.core.io.IoUtil;
import com.whaleal.icefrog.core.util.StrUtil;
import com.whaleal.icefrog.extra.compress.archiver.Archiver;
import com.whaleal.icefrog.extra.compress.archiver.SevenZArchiver;
import com.whaleal.icefrog.extra.compress.archiver.StreamArchiver;
import com.whaleal.icefrog.extra.compress.extractor.Extractor;
import com.whaleal.icefrog.extra.compress.extractor.SenvenZExtractor;
import com.whaleal.icefrog.extra.compress.extractor.StreamExtractor;
import org.apache.commons.compress.archivers.ArchiveStreamFactory;
import org.apache.commons.compress.archivers.StreamingNotSupportedException;
import org.apache.commons.compress.compressors.CompressorException;
import org.apache.commons.compress.compressors.CompressorInputStream;
import org.apache.commons.compress.compressors.CompressorOutputStream;
import org.apache.commons.compress.compressors.CompressorStreamFactory;

import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
import java.nio.charset.Charset;

/**
 * 压缩工具类

 * 基于icefrogs-compress的压缩解压封装
 *
 * @author Looly
 * @author wh
 * @since 1.0.0
 */
public class CompressUtil {

	/**
	 * 获取压缩输出流，用于压缩指定内容，支持的格式例如：
	 * 

	 *     
{@value CompressorStreamFactory#GZIP}
	 *     
{@value CompressorStreamFactory#BZIP2}
	 *     
{@value CompressorStreamFactory#XZ}
	 *     
{@value CompressorStreamFactory#PACK200}
	 *     
{@value CompressorStreamFactory#SNAPPY_FRAMED}
	 *     
{@value CompressorStreamFactory#LZ4_BLOCK}
	 *     
{@value CompressorStreamFactory#LZ4_FRAMED}
	 *     
{@value CompressorStreamFactory#ZSTANDARD}
	 *     
{@value CompressorStreamFactory#DEFLATE}
	 * 
	 *
	 * @param compressorName 压缩名称，见：{@link CompressorStreamFactory}
	 * @param out            输出流，可以输出到内存、网络或文件
	 * @return {@link CompressorOutputStream}
	 */
	public static CompressorOutputStream getOut(String compressorName, OutputStream out) {
		try {
			return new CompressorStreamFactory().createCompressorOutputStream(compressorName, out);
		} catch (CompressorException e) {
			throw new CompressException(e);
		}
	}

	/**
	 * 获取压缩输入流，用于解压缩指定内容，支持的格式例如：
	 * 

	 *     
{@value CompressorStreamFactory#GZIP}
	 *     
{@value CompressorStreamFactory#BZIP2}
	 *     
{@value CompressorStreamFactory#XZ}
	 *     
{@value CompressorStreamFactory#PACK200}
	 *     
{@value CompressorStreamFactory#SNAPPY_FRAMED}
	 *     
{@value CompressorStreamFactory#LZ4_BLOCK}
	 *     
{@value CompressorStreamFactory#LZ4_FRAMED}
	 *     
{@value CompressorStreamFactory#ZSTANDARD}
	 *     
{@value CompressorStreamFactory#DEFLATE}
	 * 
	 *
	 * @param compressorName 压缩名称，见：{@link CompressorStreamFactory}，null表示自动检测
	 * @param in            输出流，可以输出到内存、网络或文件
	 * @return {@link CompressorOutputStream}
	 */
	public static CompressorInputStream getIn(String compressorName, InputStream in) {
		in = IoUtil.toMarkSupportStream(in);
		try {
			if(StrUtil.isBlank(compressorName)){
				compressorName = CompressorStreamFactory.detect(in);
			}
			return new CompressorStreamFactory().createCompressorInputStream(compressorName, in);
		} catch (CompressorException e) {
			throw new CompressException(e);
		}
	}

	/**
	 * 创建归档器，支持：
	 * 

	 *     
{@link ArchiveStreamFactory#AR}
	 *     
{@link ArchiveStreamFactory#CPIO}
	 *     
{@link ArchiveStreamFactory#JAR}
	 *     
{@link ArchiveStreamFactory#TAR}
	 *     
{@link ArchiveStreamFactory#ZIP}
	 *     
{@link ArchiveStreamFactory#SEVEN_Z}
	 * 
	 *
	 * @param charset      编码
	 * @param archiverName 归档类型名称，见{@link ArchiveStreamFactory}
	 * @param file         归档输出的文件
	 * @return Archiver
	 */
	public static Archiver createArchiver(Charset charset, String archiverName, File file) {
		if (ArchiveStreamFactory.SEVEN_Z.equalsIgnoreCase(archiverName)) {
			return new SevenZArchiver(file);
		}
		return StreamArchiver.create(charset, archiverName, file);
	}

	/**
	 * 创建归档器，支持：
	 * 

	 *     
{@link ArchiveStreamFactory#AR}
	 *     
{@link ArchiveStreamFactory#CPIO}
	 *     
{@link ArchiveStreamFactory#JAR}
	 *     
{@link ArchiveStreamFactory#TAR}
	 *     
{@link ArchiveStreamFactory#ZIP}
	 *     
{@link ArchiveStreamFactory#SEVEN_Z}
	 * 
	 *
	 * @param charset      编码
	 * @param archiverName 归档类型名称，见{@link ArchiveStreamFactory}
	 * @param out          归档输出的流
	 * @return Archiver
	 */
	public static Archiver createArchiver(Charset charset, String archiverName, OutputStream out) {
		if (ArchiveStreamFactory.SEVEN_Z.equalsIgnoreCase(archiverName)) {
			return new SevenZArchiver(out);
		}
		return StreamArchiver.create(charset, archiverName, out);
	}

	/**
	 * 创建归档解包器，支持：
	 * 

	 *     
{@link ArchiveStreamFactory#AR}
	 *     
{@link ArchiveStreamFactory#CPIO}
	 *     
{@link ArchiveStreamFactory#JAR}
	 *     
{@link ArchiveStreamFactory#TAR}
	 *     
{@link ArchiveStreamFactory#ZIP}
	 *     
{@link ArchiveStreamFactory#SEVEN_Z}
	 * 
	 *
	 * @param charset 编码，7z格式此参数无效
	 * @param file    归档文件
	 * @return {@link Extractor}
	 */
	public static Extractor createExtractor(Charset charset, File file) {
		return createExtractor(charset, null, file);
	}

	/**
	 * 创建归档解包器，支持：
	 * 

	 *     
{@link ArchiveStreamFactory#AR}
	 *     
{@link ArchiveStreamFactory#CPIO}
	 *     
{@link ArchiveStreamFactory#JAR}
	 *     
{@link ArchiveStreamFactory#TAR}
	 *     
{@link ArchiveStreamFactory#ZIP}
	 *     
{@link ArchiveStreamFactory#SEVEN_Z}
	 * 
	 *
	 * @param charset      编码，7z格式此参数无效
	 * @param archiverName 归档类型名称，见{@link ArchiveStreamFactory}，null表示自动识别
	 * @param file         归档文件
	 * @return {@link Extractor}
	 */
	public static Extractor createExtractor(Charset charset, String archiverName, File file) {
		if (ArchiveStreamFactory.SEVEN_Z.equalsIgnoreCase(archiverName)) {
			return new SenvenZExtractor(file);
		}
		try {
			return new StreamExtractor(charset, archiverName, file);
		} catch (CompressException e) {
			final Throwable cause = e.getCause();
			if (cause instanceof StreamingNotSupportedException && cause.getMessage().contains("7z")) {
				return new SenvenZExtractor(file);
			}
			throw e;
		}
	}

	/**
	 * 创建归档解包器，支持：
	 * 

	 *     
{@link ArchiveStreamFactory#AR}
	 *     
{@link ArchiveStreamFactory#CPIO}
	 *     
{@link ArchiveStreamFactory#JAR}
	 *     
{@link ArchiveStreamFactory#TAR}
	 *     
{@link ArchiveStreamFactory#ZIP}
	 *     
{@link ArchiveStreamFactory#SEVEN_Z}
	 * 
	 *
	 * @param charset 编码，7z格式此参数无效
	 * @param in      归档输入的流
	 * @return {@link Extractor}
	 */
	public static Extractor createExtractor(Charset charset, InputStream in) {
		return createExtractor(charset, null, in);
	}

	/**
	 * 创建归档解包器，支持：
	 * 

	 *     
{@link ArchiveStreamFactory#AR}
	 *     
{@link ArchiveStreamFactory#CPIO}
	 *     
{@link ArchiveStreamFactory#JAR}
	 *     
{@link ArchiveStreamFactory#TAR}
	 *     
{@link ArchiveStreamFactory#ZIP}
	 *     
{@link ArchiveStreamFactory#SEVEN_Z}
	 * 
	 *
	 * @param charset      编码，7z格式此参数无效
	 * @param archiverName 归档类型名称，见{@link ArchiveStreamFactory}，null表示自动识别
	 * @param in           归档输入的流
	 * @return {@link Extractor}
	 */
	public static Extractor createExtractor(Charset charset, String archiverName, InputStream in) {
		if (ArchiveStreamFactory.SEVEN_Z.equalsIgnoreCase(archiverName)) {
			return new SenvenZExtractor(in);
		}
		try {
			return new StreamExtractor(charset, archiverName, in);
		} catch (CompressException e) {
			final Throwable cause = e.getCause();
			if (cause instanceof StreamingNotSupportedException && cause.getMessage().contains("7z")) {
				return new SenvenZExtractor(in);
			}
			throw e;
		}
	}
}