边压缩边下载多个文件为一个zip

最新推荐文章于 2021-11-25 19:13:42 发布

bigliu123

最新推荐文章于 2021-11-25 19:13:42 发布

阅读量124

点赞数

分类专栏： Java

本文链接：https://blog.csdn.net/bigliu123/article/details/86135130

版权

Java 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

前端请求参数

{"ids":["1080431045908779009","1080434473292681218"]}

Controller

/**
 * 批量下载文件
 */
@GetMapping("/downloadFiles")
public void downloadFiles(HttpServletRequest request, HttpServletResponse response,
                          @RequestParam("ids") List<String> ids,
                          @RequestParam(value = "fileName", required = false) String fileName){
    fileName = StringUtils.isBlank(fileName) ? "files" : fileName;
    fileInfoService.downloadFiles(request, response, ids, fileName);
}

Service

/**
 * 批量下载文件
 *
 * @param request  http请求
 * @param response http响应
 * @param ids      文件id数组
 * @param fileName 打包文件名称
 */
@Override
public void downloadFiles(HttpServletRequest request, HttpServletResponse response, List<String> ids, String fileName) {
    //响应头的设置
    response.reset();
    response.setCharacterEncoding("utf-8");
    response.setContentType("multipart/form-data");

    //设置压缩包的名字
    //解决不同浏览器压缩包名字含有中文时乱码的问题
    String downloadName = fileName + ".zip";
    String agent = request.getHeader("USER-AGENT");
    try {
        if (agent.contains("MSIE") || agent.contains("Trident")) {
            downloadName = java.net.URLEncoder.encode(downloadName, "UTF-8");
        } else {
            downloadName = new String(downloadName.getBytes("UTF-8"), "ISO-8859-1");
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
    response.setHeader("Content-Disposition", "attachment;filename=\"" + downloadName + "\"");
    //设置压缩流：直接写入response，实现边压缩边下载
    ZipOutputStream zipos = null;
    try {
        zipos = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()));
        //设置压缩方法
        zipos.setMethod(ZipOutputStream.DEFLATED);
    } catch (Exception e) {
        e.printStackTrace();
    }


    //循环将文件写入压缩流
    DataOutputStream os = null;
    for (int i = 0; i < ids.size(); i++) {
        FileInfo fileInfo = getById(ids.get(i));
        //拼接文件路径
        String filePath = baseFilePath + fileInfo.getFolder() + "\\" + fileInfo.getSaveName();
        File file = new File(filePath);
        if (!file.exists()) {
            continue;
        }
        try {
            //添加ZipEntry，并ZipEntry中写入文件流
            //这里，加上i是防止要下载的文件有重名的导致下载失败
            zipos.putNextEntry(new ZipEntry(fileInfo.getName()));
            os = new DataOutputStream(zipos);
            InputStream is = new FileInputStream(file);
            byte[] b = new byte[1000];
            int length = 0;
            while ((length = is.read(b)) != -1) {
                os.write(b, 0, length);
            }
            is.close();
            zipos.closeEntry();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    //关闭流
    if (os != null) {
        try {
            os.flush();
            os.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
    if (zipos != null) {
        try {
            zipos.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}