文件压缩工具包

package com.yjcloud.form.pdf;

import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * 文件压缩工具包
 * @author 
 * @since 2021-12-16
 */
public class ZipUtil {


    /**
     * 文件流压缩
     * @author 
     * @since 
     * @param baisByte 需要压缩的字节输出流(ByteArrayOutputStream)的字节数组
     * @param fileName 需要压缩的文件名
     * @return  压缩后字节数组输出流转为的字符串
     * @throws IOException
     */
    public static String zipByteArrayOutputStream(byte[] baisByte,String fileName) throws IOException {

        //1.将需要压缩的字节输出流，转为字节数组输入流，
        ByteArrayInputStream bais = new ByteArrayInputStream(baisByte);

        //2.创建字节数组输出流，用于返回压缩后的输出流字节数组
        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        //3.创建压缩输出流
        ZipOutputStream zipOut = new ZipOutputStream(baos);

        //zipOut.setEncoding("GBK");//设置编码格式，否则中文文件名乱码

        //zipOut.setMethod(ZipOutputStream.DEFLATED);//进行压缩存储

        //zipOut.setLevel(Deflater.BEST_COMPRESSION);//压缩级别值为0-9共10个级别(值越大，表示压缩越利害)

        //4.设置ZipEntry对象，并对需要压缩的文件命名
        zipOut.putNextEntry(new ZipEntry(fileName)); ;


        //5.读取要压缩的字节输出流，进行压缩
        int temp = 0 ;
        while((temp=bais.read())!=-1){
            zipOut.write(temp) ;    // 压缩输出
        }

        // 关闭流
        bais.close();
        zipOut.close();

        //字节数组输出流转为字符串并设置编码
        String result =new String(baos.toByteArray(), "ISO-8859-1");

        baos.close();// 关闭流

        return result;
    }







    /**
     * 批量文件流压缩
     * @author 
     * @since 
     * @param baisBytes  List<Map<被压缩压缩文件名, 被压缩的文件流>>
     * @return 压缩后字节数组输出流转为的字符串
     * @throws IOException
     */
    public static byte[] batchZipByteArrayOutputStream(List<Map<String, byte[]>> baisBytes) throws IOException {

        //1.创建字节数组输出流，用于返回压缩后的输出流字节数组
        ByteArrayOutputStream baos = new ByteArrayOutputStream();

        //2.创建压缩输出流
        ZipOutputStream zipOut = new ZipOutputStream(baos);

        //3.遍历要批量压缩的集合文件流
        ByteArrayInputStream bais =null;
        Map<String, byte[]> tempMap =null;
        String fileName=null;
        int temp = 0 ;
        for (int i = 0; i < baisBytes.size(); i++) {

            tempMap=baisBytes.get(i);

            fileName=tempMap.keySet().iterator().next();

            //3.1将需要压缩的字节输出流，转为字节数组输入流，
            bais = new ByteArrayInputStream(tempMap.get(fileName));

            //zipOut.setMethod(ZipOutputStream.DEFLATED);//进行压缩存储

            //zipOut.setLevel(Deflater.BEST_COMPRESSION);//压缩级别值为0-9共10个级别(值越大，表示压缩越利害)

            //3.2设置ZipEntry对象，并对需要压缩的文件命名
            zipOut.putNextEntry(new ZipEntry(fileName));


            //3.3读取要压缩的字节输出流，进行压缩
            temp = 0 ;
            while((temp=bais.read())!=-1){
                zipOut.write(temp) ;    // 压缩输出
            }

            // 3.4关闭流
            bais.close();
        }
        zipOut.close();
        baos.close();// 关闭流
        return baos.toByteArray();
    }

}