在线问题反馈模块实战(二十):实现文件批量导出到zip压缩包中功能

    /**
     * 所有问题反馈截图导出成zip(压缩包)
     */
    @GetMapping("/export-questions-images")
    @ApiOperation(value = "所有问题反馈截图导出成zip(压缩包)", notes = "所有问题反馈截图导出成zip(压缩包)")
    public void exportQuestionsImages(HttpServletResponse response) {
        commonService.exportQuestions(response);
    }

    /**
     * 所有问题反馈截图导出成zip(压缩包)
     */
    void exportQuestions(HttpServletResponse response);

/**
     * 所有问题反馈截图导出成zip(压缩包)
     */
    @Override
    public void exportQuestions(HttpServletResponse response){

        //1、判断父级目录是否存在
        File rootPath = new File(questionRootPath);
        if (!rootPath.exists()) {
            //创建父目录文件夹
            rootPath.mkdirs();
        }

        //2、查询问题反馈表中有上传过截图的数据
        List<UserQuestionsEntity> questions = userQuestionsService.getQuestions();

        //3、一次压缩
        //先将所有人的截图放到同一文件夹中并压缩
        questions.forEach(p -> {

            //子文件夹命名
            String userId = p.getCreatorAccountId();
            //获取截图父级目录
            String parentPath = questionImg + userId;
            //将多路径转成图片数组
            String[] paths = p.getFilePaths().split(",");
            // 创建临时路径,存放压缩文件
            String zipFilePath = rootPath + "/" + userId + ".zip";
            //压缩输出流,包装流,将临时文件输出流包装成压缩流,将所有文件输出到这里,打成zip包
            try {
                ZipOutputStream zipOut = new ZipOutputStream(new FileOutputStream(zipFilePath));
                // 循环调用压缩文件方法,将一个一个需要下载的文件打入压缩文件包
                for (String path : paths) {
                    //拼接文件全路径
                    String imagePath = parentPath + "/" + path;

                    //判断文件是否存在
                    File file = new File(imagePath);
                    if (!file.exists()) {
                        continue;
                    }
                    try {
                        //将文件添加到指定的压缩包中
                        uploader.fileToZip(imagePath, zipOut);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }
                // 压缩完成后,关闭压缩流
                zipOut.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        });
        //4、二次压缩
        //二次压缩(压缩父级目录)
        String downloadPath = rootPath + "/" + "问题反馈截图.zip";
        try {
            ZipOutputStream zipOut1 = new ZipOutputStream(new FileOutputStream(downloadPath));

            //遍历获取所有的父级目录
            questions.forEach(p -> {
                //拼接父级目录
                String path = rootPath + "/" + p.getCreatorAccountId() + ".zip";
                //获取存放文件的跟路径
                try {
                    uploader.fileToZip(path, zipOut1);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            });
            // 压缩完成后,关闭压缩流
            zipOut1.close();
            //拼接下载默认名称并转为uft-8格式
            response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode("在线问题反馈截图.zip", StandardCharsets.UTF_8.name()));

            //该流不可以手动关闭,手动关闭下载会出问题,下载完成后会自动关闭
            ServletOutputStream outputStream = response.getOutputStream();

            FileInputStream inputStream = new FileInputStream(downloadPath);

            // copy方法为文件复制,在这里直接实现了下载效果
            IOUtils.copy(inputStream, outputStream);
        } catch (IOException e) {
            e.printStackTrace();
        }

        //下载完成之后，本地删掉这个zip
        File fileTempZip = new File(downloadPath);
        fileTempZip.delete();
    }

    /**
     * 查询已上传过截图的问题反馈
     */
    List<UserQuestionsEntity> getQuestions();

    /**
     * 查询已上传过截图的问题反馈
     */
    @Override
    public List<UserQuestionsEntity> getQuestions() {

        LambdaQueryWrapper<UserQuestionsEntity> wrapper = new LambdaQueryWrapper<>();
        wrapper.isNotNull(UserQuestionsEntity::getFilePaths);//只查询截图路径不为空的数据

        return this.list(wrapper);
    }

    /**
     * 将一个文件写入压缩包(一次只压缩一次)
     *
     * @param filePath 文件路径
     * @param zipOut   压缩路径
     */
    public void fileToZip(String filePath, ZipOutputStream zipOut) throws IOException {
        // 需要压缩的文件
        File file = new File(filePath);
        // 获取文件名称,如果有特殊命名需求,可以将参数列表拓展,传fileName
        String fileName = file.getName();
        FileInputStream fileInput = new FileInputStream(filePath);
        // 缓冲
        byte[] bufferArea = new byte[1024 * 10];
        BufferedInputStream bufferStream = new BufferedInputStream(fileInput, 1024 * 10);
        // 将当前文件作为一个zip实体写入压缩流,fileName代表压缩文件中的文件名称
        zipOut.putNextEntry(new ZipEntry(fileName));
        int length = 0;
        // 最常规IO操作,不必紧张
        while ((length = bufferStream.read(bufferArea, 0, 1024 * 10)) != -1) {
            zipOut.write(bufferArea, 0, length);
        }
        //关闭流
        fileInput.close();
        // 需要注意的是缓冲流必须要关闭流,否则输出无效
        bufferStream.close();
        // 压缩流不必关闭,使用完后再关
    }

#自定义配置项
review:

  #文件存放地址
  file:
    question-zip-path: ./template/question-zip/
    question-img: ./template/question-img/ 

    @Value("${review.file.question-zip-path}")
    private String questionRootPath;

    @Value("${review.file.question-img}")
    private String questionImg;