spring-boot文件打包ZIP下载

spring-boot 打包下载文件 demo

package com.xxxxx.core.utils;

import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.List;
import java.util.UUID;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * @Author HanKeQi
 * @Date 2020/7/21 2:28 下午
 * @Version 1.0
 **/
public class DownLoadZipUtil {

    /**
     * 打包下载
     * @param paths 目前处理是文件绝对路径， TODO 此处可以改成 List<File> 去处理 或者网络图片下载流
     */
    public static void zipFileDownload(List<String> paths){
        HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
        HttpServletResponse response = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getResponse();
        //循环将文件写入压缩流
        DataOutputStream os = null;
        //设置压缩流：直接写入response，实现边压缩边下载
        ZipOutputStream zipos = null;
        try {
            response.reset();
            response.setCharacterEncoding("utf-8");
            response.setContentType("multipart/form-data");
            //设置压缩包的名字
            //解决不同浏览器压缩包名字含有中文时乱码的问题
            String downloadName = UUID.randomUUID().toString() +".zip";
            String agent = request.getHeader("USER-AGENT");
            try {
                //针对IE或者以IE为内核的浏览器：
                if (agent.contains("MSIE") || agent.contains("Trident")) {
                    downloadName = java.net.URLEncoder.encode(downloadName, "UTF-8");
                } else {
                    //非IE浏览器的处理：
                    downloadName = new String(downloadName.getBytes("UTF-8"),"ISO-8859-1");
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
            response.setHeader("Content-Disposition", "attachment;fileName=\"" + downloadName + "\"");
            zipos = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()));
            zipos.setMethod(ZipOutputStream.DEFLATED); //设置压缩方法
            for (String path : paths){
                File file = new File(path);
                if(file.exists()){
                    try {
                        //添加ZipEntry，并ZipEntry中写入文件流
                        //这里，防止要下载的文件有重名的导致下载失败
                        zipos.putNextEntry(new ZipEntry("folder/"+ file.getName()));
                        os = new DataOutputStream(zipos);
                        InputStream is = new FileInputStream(file);
                        byte[] b = new byte[100];
                        int length = 0;
                        while((length = is.read(b))!= -1){
                            os.write(b, 0, length);
                        }
                        is.close();
                        zipos.closeEntry();
                    }catch (Exception e){
                        e.printStackTrace();
                    }
                }
            }
        }catch (Exception e){
			e.printStackTrace();
        }finally {
            try {
                if (os!= null){
                    os.flush();
                    os.close();
                }
                if (zipos != null){
                    zipos.close();
                }
            }catch (Exception e){
                e.printStackTrace();
            }
        }
    }

}