批量下载文件(Get请求)

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

bigliu123

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

阅读量326

点赞数

分类专栏： Java

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

版权

Java 专栏收录该内容

16 篇文章 0 订阅

订阅专栏

前端请求路径:

http://172.16.216.63:8091/cadInteraction/fileInfo/downloadFiles?ids=1080431045908779009,1080434473292681218,1080434489415585794

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) {
        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());
        }
    }
}