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

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

    /**
     * 導(dǎo)出建議列表
     * ManuscriptListVO:這個(gè)實(shí)體包含了我需要的文件夾名稱和文件地址 文件是存在oss里的
     * RecordCategory:這個(gè)實(shí)體是存放文件層級(jí)的 下面會(huì)展示
     *
     * @param httpServletResponse http servlet響應(yīng)
     * @paramstatus         審核狀態(tài)
     * @param httpServletRequest  http servlet請(qǐng)求
     */
    @Override
    public void compressedPackageExport(HttpServletResponse httpServletResponse, String status, HttpServletRequest httpServletRequest) {
        try {
            // 獲取臨時(shí)目錄
            String tempDir = System.getProperty("java.io.tmpdir") + "temp" + File.separator;
            File tempDirFile = new File(tempDir);
            if (!tempDirFile.exists()) tempDirFile.mkdirs();
            // 壓縮包名稱
            String zipName = "壓縮包導(dǎo)出" + 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 -> {
                // 獲取該文件層級(jí)
                RecordCategory recordCategory = getRecordCategory(manuscript);
                addFileAndFolder(recordCategory, zos);
            });
            zos.flush();
            zos.close();
            /*
            導(dǎo)出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();
        }
    }

    /**
     * 封裝文件層級(jí)
     * HOME_FOLDER_RUL 是配置文件中的預(yù)下載文件夾地址 見下圖
     *
     * @param manuscript 建言
     * @return RecordCategory
     */
    private RecordCategory getRecordCategory(DataListVO data) {
        // 文件夾名稱:姓名 手機(jī)號(hào) 防重復(fù)id
        String mainFolderName = data.getName() + " " + data.getPhone() + " " + data.getId();
        String mainFolder = HOME_FOLDER_RUL + "/" + mainFolderName;
        RecordCategory result = new RecordCategory().setFileName(mainFolderName).setFilePath(mainFolder);
        // 封裝子級(jí)文件
        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);
    }

    /**
     * 添加文件夾和文件
     * 注意:如果想要保持完整的目錄 需要在文件夾名稱后邊加上`/` 否則下載的文件夾和文件都會(huì)跑到一個(gè)目錄里
     *
     * @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;

    /**
     * 下級(jí)目錄
     */
    List<RecordCategory> subordinateDirectory;
}

預(yù)下載文件夾地址

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

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

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

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