Java实现文件批量下载，并按照类别放到不同文件夹，并生成压缩包

实现批量下载，文件按照类别放到不同文件夹，并生成压缩包

实现思路

1. 获取所有文件信息（类别、名称、文件路径等）
2. 对获取的文件List进行分组，根据某个维度进行分组，可以使用 Collectors.groupingBy(XXX::getXx) 的方式分组得到Map<String, List>
3. 创建 ZipOutputStream 对象
4. 在ZipOutputStream 中，根据步骤2中的Map<String, List>将List中的文件通过文件流将文件输入到zip中
5. 得到对应的zip压缩文件，再将压缩文件zip发送给前端
6. 关闭流

代码实现

public void batchDownload(@RequestBody List<UploadInfoDto> uploadInfoListDto, HttpServletResponse response) throws Exception {
    List<UploadInfo> uploadInfos = new ArrayList<>();
    //获取对应的文件信息，存放List中
    for (UploadInfoDto dto : uploadInfoListDto) {
        List<UploadInfo> currentUploadInfos = uploadService.getUploadFile(dto);
        uploadInfos.addAll(currentUploadInfos);
    }

    String zipName = "annex.zip";
    response.setContentType("APPLICATION/OCTET-STREAM");
    response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(zipName, "UTF-8"));

    logger.info("开始下载");
    Map<String, List<UploadInfo>> UploadInfoGroup = uploadInfos.stream().collect(Collectors.groupingBy(UploadInfo::getClaimNo));
    // 压缩文件路径
    String zipFilePath = "billFileAnnex.zip";
    OutputStream os = response.getOutputStream();
    try {
        // 创建 ZipOutputStream 对象
        FileOutputStream fos = new FileOutputStream(zipFilePath);
        ZipOutputStream zos = new ZipOutputStream(fos);
        // 循环遍历每个种类文件
        for (String category : UploadInfoGroup.keySet()) {
            // 创建一个新的 ZipEntry 对象，表示一个压缩文件
            ZipEntry entry = new ZipEntry(category + "/");//根据不同类型放入不同文件夹
            zos.putNextEntry(entry);
            for (UploadInfo uploadInfo : UploadInfoGroup.get(claimNo)) {
                Map<String, Object> metaMap = new HashMap<>();
                // 将各个种类文件夹下的文件添加到压缩文件中
                InputStream is = OSSUtil.downloadObject(uploadInfo.getFilePath());//获取oss上的文件流，下载到zip中
                addFileToZip(zos, uploadInfo, is);
            }
        }
        zos.flush();//zos必须在os之前，否则前端下载zip报错 zip: 不可预料的压缩文件末端
        zos.close();
        BufferedInputStream bis = new BufferedInputStream(Files.newInputStream(Paths.get(zipFilePath)));
        byte[] buffer = new byte[1024];
        int length;
        while ((length = bis.read(buffer)) > 0) {
            os.write(buffer, 0, length);
        }
        bis.close();
        os.flush();
        os.close();
        logger.info("压缩完成！");
        //压缩文件发送给前端后，删除本地生成的临时文件
        File zipFile = new File(zipFilePath);
        if (zipFile.delete()) {
            logger.info("delete file success");
        }
    } catch (IOException e) {
        logger.error("批量文件下载异常");
        e.printStackTrace();
    }
}

//将文件分组存入不同文件夹
private static void addFileToZip(ZipOutputStream zos, UploadInfo uploadInfo, InputStream inputStream) throws IOException {
    // 创建一个新的 ZipEntry 对象，表示一个压缩文件中的文件
    ZipEntry entry = new ZipEntry(uploadInfo.getCategory() + "/" + uploadInfo.getFileName());
    zos.putNextEntry(entry);

    // 写入文件内容到 ZipOutputStream 中
    byte[] buffer = new byte[1024];
    int length;
    while ((length = inputStream.read(buffer)) > 0) {
        zos.write(buffer, 0, length);
    }
    inputStream.close();
}

效果截图