文件压缩打包工具类

文件压缩打包工具类

import cn.hutool.core.io.FileUtil;
import cn.hutool.core.io.IoUtil;
import org.springframework.stereotype.Component;
import org.springframework.util.FileCopyUtils;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.net.URLEncoder;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Map;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * <p>
 * 文件压缩打包工具类
 * </p>
 *
 * @author lailai
 * @since 2021-4-15
 */
@Component
public class ZipUtil {

    private static final String TEM_PATH = System.getProperty("java.io.tmpdir");

    /**
     * 批量文件流压缩
     *
     * @param baisBytes   List<Map<被压缩压缩文件名, 被压缩的文件流>>
     * @param zipFileName 压缩包文件名
     * @param response    resp
     * @throws IOException e
     */
    public static void batchZipDownLoad(List<Map<String, byte[]>> baisBytes,String zipFileName,HttpServletResponse response) throws IOException {
        // 压缩文件
        File tempFile = FileUtil.createTempFile(zipFileName, ".zip", new File(TEM_PATH), true);
        //1.创建字节数组输出流
        FileOutputStream baos = new FileOutputStream(tempFile);
        //2.创建压缩输出流
        ZipOutputStream zipOut = new ZipOutputStream(baos);
        //zipOut.setEncoding("GBK");//设置编码格式，否则中文文件名乱码
        //3.遍历要批量压缩的集合文件流
        ByteArrayInputStream bais = null;
        Map<String, byte[]> tempMap = null;
        String fileName = null;
        int temp = 0;
        for (Map<String, byte[]> baisByte : baisBytes) {
            tempMap = baisByte;
            fileName = tempMap.keySet().iterator().next();
            //3.1将需要压缩的字节输出流，转为字节数组输入流，
            bais = new ByteArrayInputStream(tempMap.get(fileName));
            //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();// 关闭流
        try {
            // 以流的形式下载文件。
            InputStream fis = new BufferedInputStream(new FileInputStream(tempFile.getPath()));
            // 清空response
            response.reset();
            response.setContentType("application/octet-stream");
            // 如果输出的是中文名的文件，在此处就要用URLEncoder.encode方法进行处理
            URLEncoder.encode(tempFile.getName(), "UTF-8");
            response.setHeader("Content-Disposition",
                "attachment;filename=" + new String(zipFileName.getBytes(StandardCharsets.UTF_8), StandardCharsets.ISO_8859_1) + ".zip");
            //输出文件流
            FileCopyUtils.copy(fis, response.getOutputStream());
            fis.close();
            IoUtil.close(response.getOutputStream());
        } catch (Exception ex) {
            ex.printStackTrace();
        } finally {
            try {
                File f = new File(tempFile.getPath());
                FileUtil.del(f);
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}