Java之多文件夹和文件打包压缩包下载

需求：主文件夹内有多个子级文件夹，子级文件夹内有txt、图片和pdf等类型文件，需要将主文件夹打包压缩包下载。目录不一样也通用。
实现方式改过一版，这里只展示改过之后的方式。
改之前思路：查询出所需下载的文件，将文件封装进文件夹，将文件夹压缩，返回文件供前端下载。但是这种方法每次下载的时候就要生成新的文件夹和压缩包，接口的响应时间较长。
改之后思路：提前生成文件夹。在用户上传完文件之后，提前将文件夹层级生成好，导出时只需要将指定文件夹提取出来添加压缩包输出流即可。
代码：只展示service层代码

    /**
     * 导出建议列表
     * ManuscriptListVO：这个实体包含了我需要的文件夹名称和文件地址 文件是存在oss里的
     * RecordCategory：这个实体是存放文件层级的 下面会展示
     *
     * @param httpServletResponse http servlet响应
     * @paramstatus         审核状态
     * @param httpServletRequest  http servlet请求
     */
    @Override
    public void compressedPackageExport(HttpServletResponse httpServletResponse, String status, HttpServletRequest httpServletRequest) {
        try {
            // 获取临时目录
            String tempDir = System.getProperty("java.io.tmpdir") + "temp" + File.separator;
            File tempDirFile = new File(tempDir);
            if (!tempDirFile.exists()) tempDirFile.mkdirs();
            // 压缩包名称
            String zipName = "压缩包导出" + new SimpleDateFormat("yyyyMMddHHmmss").format(new Date()) + ".zip";
            String homeFolderPath = tempDir + zipName;
            ZipOutputStream zos = new ZipOutputStream(new FileOutputStream(homeFolderPath));
            // 查询建言
            List<ManuscriptListVO> manuscriptListVOList = manuscriptMapper.getDataByStatus(status);
            if (CollectionUtils.isEmpty(manuscriptListVOList)) return;
            // 处理数据
            manuscriptListVOList.forEach(manuscript -> {
                // 获取该文件层级
                RecordCategory recordCategory = getRecordCategory(manuscript);
                addFileAndFolder(recordCategory, zos);
            });
            zos.flush();
            zos.close();
            /*
            导出zip文件
             */
            File tempFile = new File(homeFolderPath);
            httpServletResponse.setHeader("content-length", tempFile.length() + "");
            FileUtils.setAttachmentResponseHeader(httpServletResponse, zipName);
            FileUtils.writeBytes(homeFolderPath, httpServletResponse.getOutputStream());
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 封装文件层级
     * HOME_FOLDER_RUL 是配置文件中的预下载文件夹地址 见下图
     *
     * @param manuscript 建言
     * @return RecordCategory
     */
    private RecordCategory getRecordCategory(DataListVO data) {
        // 文件夹名称：姓名 手机号 防重复id
        String mainFolderName = data.getName() + " " + data.getPhone() + " " + data.getId();
        String mainFolder = HOME_FOLDER_RUL + "/" + mainFolderName;
        RecordCategory result = new RecordCategory().setFileName(mainFolderName).setFilePath(mainFolder);
        // 封装子级文件
        List<RecordCategory> recordCategoryList = new ArrayList<>();
        List<String> fileList = JSON.parseArray(data.getDownfile(), String.class);
        if (!CollectionUtils.isEmpty(fileList))
            for (String file : fileList) {
                String fileName = file.substring(file.lastIndexOf("/") + 1);
                recordCategoryList.add(new RecordCategory().setFileName(fileName).setFilePath(mainFolder + "/" + fileName));
            }
        // 封装txt
        recordCategoryList.add(new RecordCategory().setFileName("压缩包.txt").setFilePath(mainFolder + "/" + "压缩包.txt"));
        // 封装文件list
        return result.setSubordinateDirectory(recordCategoryList);
    }

    /**
     * 添加文件夹和文件
     * 注意：如果想要保持完整的目录 需要在文件夹名称后边加上`/` 否则下载的文件夹和文件都会跑到一个目录里
     *
     * @param recordCategory 文件路径
     * @param zos ZipOutputStream
     */
    private void addFileAndFolder(RecordCategory recordCategory, ZipOutputStream zos) {
        // 添加文件夹
        try {
            zos.putNextEntry(new ZipEntry(recordCategory.getFileName() + "/"));
            zos.closeEntry();
        } catch (IOException e) {
            e.printStackTrace();
        }
        // 遍历添加附件
        if (!CollectionUtils.isEmpty(recordCategory.getSubordinateDirectory()))
            for (RecordCategory data : recordCategory.getSubordinateDirectory()) {
                try (FileInputStream fis = new FileInputStream(new File(data.getFilePath()))) {
                    // 压缩文件中写入名称
                    zos.putNextEntry(new ZipEntry(recordCategory.getFileName() + "/" + data.getFileName()));
                    // 向压缩文件中输出数据
                    int len;
                    byte[] bt = new byte[5 * 1024];
                    while ((len = fis.read(bt)) != -1) {
                        // 压缩文件中写入真正的文件流
                        zos.write(bt, 0, len);
                    }
                    zos.closeEntry();
                    IOUtils.close(fis);
                } catch (Exception e) {
                    e.printStackTrace();
                    throw new ServiceException("资源异常");
                }
            }
    }

RecordCategory

import lombok.Data;
import lombok.experimental.Accessors;

import java.util.List;

@Accessors(chain = true)
@Data
public class RecordCategory {
    /**
     * 文件夹路径
     */
    String filePath;

    /**
     * 文件夹名称
     */
    String fileName;

    /**
     * 下级目录
     */
    List<RecordCategory> subordinateDirectory;
}

都是常规功能，任何目录层级都可以用