Java之多文件夾和文件打包壓縮包下載

需求:主文件夾內(nèi)有多個子級文件夾,子級文件夾內(nèi)有txt、圖片和pdf等類型文件,需要將主文件夾打包壓縮包下載。目錄不一樣也通用。
實現(xiàn)方式改過一版,這里只展示改過之后的方式。
改之前思路:查詢出所需下載的文件,將文件封裝進文件夾,將文件夾壓縮,返回文件供前端下載。但是這種方法每次下載的時候就要生成新的文件夾和壓縮包,接口的響應時間較長。
改之后思路:提前生成文件夾。在用戶上傳完文件之后,提前將文件夾層級生成好,導出時只需要將指定文件夾提取出來添加壓縮包輸出流即可。
代碼:只展示service層代碼

    /**
     * 導出建議列表
     * ManuscriptListVO:這個實體包含了我需要的文件夾名稱和文件地址 文件是存在oss里的
     * RecordCategory:這個實體是存放文件層級的 下面會展示
     *
     * @param httpServletResponse http servlet響應
     * @paramstatus         審核狀態(tài)
     * @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;
            // 處理數(shù)據(jù)
            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()));
                    // 向壓縮文件中輸出數(shù)據(jù)
                    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;
}

預下載文件夾地址

都是常規(guī)功能,任何目錄層級都可以用

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容