Spring Boot整合批量文件下载功能

痛定思痛。 2022-09-04 07:52 361阅读 0赞

> **导语**  
>   在测试功能中，出现了再Spring Boot中打包压缩文件的需求，下面来实现一个这个操作

/** * 批量下载文件 * * @param request * @param response */
        @GetMapping("/qcodeExcept")
        public void qcodeExcept(HttpServletRequest request, HttpServletResponse response) { 
    
            //响应头的设置
            response.reset();
            response.setCharacterEncoding("utf-8");
            response.setContentType("multipart/form-data");
            //设置压缩包的名字
            String dates = "测试";
            String billname = "附件包-" + dates;
            String downloadName = billname + ".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 + "\"");
    
            //设置压缩流：直接写入response，实现边压缩边下载
            ZipOutputStream zipos = null;
            try { 
                zipos = new ZipOutputStream(new BufferedOutputStream(response.getOutputStream()));
                zipos.setMethod(ZipOutputStream.DEFLATED); //设置压缩方法
            } catch (Exception e) { 
                e.printStackTrace();
            }
    
            //循环将文件写入压缩流
            DataOutputStream os = null;
    
    
            //查询数据库获取文件信息
            JcHouseBaseInfo jcHouseBaseInfo = getJcHouseBaseInfo();
            List<JcHouseBaseInfo> list = jcHouseBaseInfoService.selectJcHouseBaseInfoList(jcHouseBaseInfo);
            //遍历文件信息（主要获取文件名/文件路径等）
            for (JcHouseBaseInfo authAttachmentDetail : list) { 
                //文件路径
                String filePath = RuoYiConfig.getQrcodeUrl() + File.separator + authAttachmentDetail.getQcodePath();
                System.out.println("filePath===" + filePath);
                File file = new File(filePath);
                if (!file.exists()) { 
                    throw new BusinessException("文件已不存在");
                } else { 
                    try { 
                        //添加ZipEntry，并ZipEntry中写入文件流
                        String fileName = authAttachmentDetail.getTeam() + "-" + authAttachmentDetail.getQcodePath();//.substring(0,authAttachmentDetail.getFileName().indexOf("."));
                        zipos.putNextEntry(new ZipEntry(fileName));
                        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 (IOException e) { 
                        e.printStackTrace();
                    }
                }
            }
            //关闭流
            try { 
                os.flush();
                os.close();
                zipos.close();
            } catch (IOException e) { 
                e.printStackTrace();
            }
        }