批量下载文件(Post请求)

前端请求:

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

Controller:

@PostMapping
 public void downloadFiles(HttpServletRequest request, HttpServletResponse response, @RequestBody Map<String, Object> map){
        List<String> ids = (List<String>) map.get("ids");
        System.out.println(ids);
}

 

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) {
        logger.error(e.getMessage());
    }
    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) {
        logger.error(e.getMessage());
    }


    //循环将文件写入压缩流
    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是防止要下载的文件有重名的导致下载失败
            if (zipos != null) {
                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();
                //将该文件的downloadCount加1
                increaseOneDownloadCount(fileInfo.getId());
            }
        } catch (IOException e) {
            logger.error(e.getMessage());
        }
    }
    //关闭流
    if (os != null) {
        try {
            os.flush();
            os.close();
        } catch (IOException e) {
            logger.error(e.getMessage());
        }
    }
    if (zipos != null) {
        try {
            zipos.close();
        } catch (IOException e) {
            logger.error(e.getMessage());
        }
    }
}