批量下载
    适用场景

    1. @RestController
    2. @Api(tags = "ExportController")
    3. @RequestMapping("/export")
    4. @Validated
    5. @AllArgsConstructor
    6. @Slf4j
    7. public class ExportController {
    8.     private final ArchivalFileMapService archivalFileMapService;
    9.     private final ExportService exportService;
    10.     private final DataService dataService;
    12.     /**
    13.      * 按档号导出，以档号为文件夹名称
    14.      * <br/>
    15.      *
    16.      * @param archivalCode
    17.      * @param httpResponse
    18.      * @return com.hzzl.erms.archival.common.api.CommonResult<?>
    19.      */
    20.     @ApiOperation(value = "按档号导出文件")
    21.     @RequestMapping(value = "/archival", method = RequestMethod.GET)
    22.     @SneakyThrows
    23.     public void exportInfoAuto(@RequestParam @NotBlank String archivalCode, HttpServletResponse httpResponse) {
    24.         log.info("exportInfoAuto() called with parameters => 【archivalCode = {}】", archivalCode);
    26.         // 查询档号对应的文件列表
    27.         List<ArchivalFileMap> archivalFileMaps = archivalFileMapService.selectByArchivalCode(archivalCode);
    28.         Assert.notEmpty(archivalFileMaps);
    30.         //生成Excel文件 archivalTmp
    31.         String excelPath =
    32.                 StrUtil.builder().append(PathUtil.excelPath()).append(archivalCode).append(".xlsx").toString();
    33.         File excel = FileUtil.newFile(excelPath);
    35.         //待导出的条目数据
    36.         List<ExportEntity> list = new ArrayList<>();
    37.         list = generateMockData();
    38.         exportService.exportList(list, includedColumnFields(), excelPath);
    40.         // 用于存储所有需要下载的文件
    41.         List<SysFile> listFile =
    42.                 archivalFileMaps.stream().map(a -> {
    43.                     SysFile fileInfo = archivalFileMapService.getFileInfo(a.getUuid());
    44.                     Assert.notNull(fileInfo);
    45.                     return fileInfo;
    46.                 }).collect(Collectors.toList());
    48.         // 以当前日期作为 zip 压缩包的名称
    49.         String zipName = DateTimeFormatter.ofPattern("yyyy-MM-dd").format(LocalDateTime.now()) + ".zip";
    50.         // 清空输出流
    51.         httpResponse.reset();
    52.         httpResponse.setCharacterEncoding("UTF-8");
    53.         // 注意写法，避免中文乱码
    54.         httpResponse.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(zipName, "UTF-8"));
    55.         httpResponse.setContentType("application/x-msdownload");
    57.         @Cleanup
    58.         BufferedOutputStream bos = null;
    59.         bos = new BufferedOutputStream(httpResponse.getOutputStream());
    60.         @Cleanup
    61.         ZipOutputStream out = new ZipOutputStream(bos);
    62.         //把Excel放入压缩包
    63.         ZipEntry excelEntry = new ZipEntry(excel.getName());
    64.         out.putNextEntry(excelEntry);
    65.         out.write(FileUtil.readBytes(excel));
    67.         Assert.notEmpty(listFile);
    68.         for (SysFile sysFile : listFile) {
    69.             Assert.notNull(sysFile.getUrl());
    70.             // 对url进行预处理后，使用 Hutool 工具类下载
    71.             HttpResponse res = HttpRequest.get(UrlEncodeUtil.reformatUrl(sysFile.getUrl())).execute();
    72.             InputStream object = res.bodyStream();
    74.             byte[] buf = new byte[1024];
    75.             int length = 0;
    77.             String strDirectoryLevel =
    78.                     StrUtil.builder()
    79.                             .append(archivalCode)
    80.                             .append(PathUtil.separator())
    81.                             .append(sysFile.getName())
    82.                             .toString();
    84.             final ZipEntry zipEntry = new ZipEntry(strDirectoryLevel);
    85.             out.putNextEntry(zipEntry);
    87.             while ((length = object.read(buf)) > 0) {
    88.                 out.write(buf, 0, length);
    89.             }
    90.         }
    91.         out.close();
    92.         bos.close();
    94.         FileUtil.del(excel);
    95.     }
    96. }
    1. public class UrlEncodeUtil {
    2.     // 防止url中有中文出现异常
    3.     @SneakyThrows
    4.     public static String reformatUrl(String url) {
    5.         List<String> list = StrUtil.split(url, "/");
    6.         String tmp = list.get(list.size() - 1);
    7.         URLEncoder.encode(tmp, "UTF-8");
    8.         String urlTmp = StrUtil.subBefore(url, "/", true).toString() + "/" + tmp;
    9.         url = StrUtil.builder().append(urlTmp).toString();
    10.         return url;
    11.     }
    12. }

    重构一下

    1.         @ApiOperation(value = "按档号导出文件")
    2.     @RequestMapping(value = "/archival", method = RequestMethod.GET)
    3.     @SneakyThrows
    4.     public void exportInfoAuto(@RequestParam @NotBlank String archivalCode, HttpServletResponse httpResponse) {
    5.         log.info("exportInfoAuto() called with parameters => 【archivalCode = {}】", archivalCode);
    7.         // 查询档号对应的文件列表
    8.         List<ArchivalFileMap> archivalFileMaps = archivalFileMapService.selectByArchivalCode(archivalCode);
    9.         Assert.notEmpty(archivalFileMaps);
    11.         //生成Excel文件 archivalTmp
    12.         String excelPath =
    13.                 StrUtil.builder().append(PathUtil.excelPath()).append(archivalCode).append(".xlsx").toString();
    14.         File excel = FileUtil.newFile(excelPath);
    16.         //待导出的条目数据
    17.         List<ExportEntity> list = new ArrayList<>();
    18.         list = generateMockData();
    19.         exportService.exportList(list, includedColumnFields(), excelPath);
    21.         // 用于存储所有需要下载的文件
    22.         List<SysFile> listFile =
    23.                 archivalFileMaps.stream().map(a -> {
    24.                     SysFile fileInfo = archivalFileMapService.getFileInfo(a.getUuid());
    25.                     Assert.notNull(fileInfo);
    26.                     return fileInfo;
    27.                 }).collect(Collectors.toList());
    29.         // 以当前日期作为 zip 压缩包的名称
    30.         String zipName = DateTimeFormatter.ofPattern("yyyy-MM-dd").format(LocalDateTime.now()) + ".zip";
    31.         // 设置返回信息的元信息
    32.         setResponseHeaderMeta(httpResponse, zipName);
    34.         @Cleanup
    35.         BufferedOutputStream bos = null;
    36.         bos = new BufferedOutputStream(httpResponse.getOutputStream());
    37.         @Cleanup
    38.         ZipOutputStream out = new ZipOutputStream(bos);
    39.         //把Excel放入压缩包
    40.         ZipEntry excelEntry = new ZipEntry(excel.getName());
    41.         out.putNextEntry(excelEntry);
    42.         out.write(FileUtil.readBytes(excel));
    44.         Assert.notEmpty(listFile);
    45.         for (SysFile sysFile : listFile) {
    46.             Assert.notNull(sysFile.getUrl());
    47.             //构造目录层级
    48.             String strDirectoryLevel =
    49.                     StrUtil.builder()
    50.                             .append(archivalCode)
    51.                             .append(PathUtil.separator())
    52.                             .append(sysFile.getName())
    53.                             .toString();
    55.             final ZipEntry zipEntry = new ZipEntry(strDirectoryLevel);
    56.             out.putNextEntry(zipEntry);
    57.             // 对url进行预处理后，使用 Hutool 工具类下载
    58.             out.write(HttpUtil.downloadBytes(UrlEncodeUtil.reformatUrl(sysFile.getUrl())));
    59.         }
    60.         out.close();
    61.         bos.close();
    63.         FileUtil.del(excel);
    64.     }
    65.     @SneakyThrows
    66.     private void setResponseHeaderMeta(HttpServletResponse httpResponse, String fileName) {
    67.         // 清空输出流
    68.         httpResponse.reset();
    69.         httpResponse.setCharacterEncoding("UTF-8");
    70.         // 注意写法，避免中文乱码
    71.         httpResponse.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName, "UTF-8"));
    72.         httpResponse.setContentType("application/x-msdownload");
    73.     }