需求：主文件夾內(nèi)有多個子級文件夾骤宣，子級文件夾內(nèi)有txt、圖片和pdf等類型文件序愚，需要將主文件夾打包壓縮包下載憔披。目錄不一樣也通用。
實現(xiàn)方式改過一版爸吮，這里只展示改過之后的方式芬膝。
改之前思路：查詢出所需下載的文件，將文件封裝進文件夾形娇，將文件夾壓縮锰霜，返回文件供前端下載。但是這種方法每次下載的時候就要生成新的文件夾和壓縮包桐早，接口的響應(yīng)時間較長癣缅。
改之后思路：提前生成文件夾厨剪。在用戶上傳完文件之后，提前將文件夾層級生成好友存，導(dǎo)出時只需要將指定文件夾提取出來添加壓縮包輸出流即可祷膳。
代碼：只展示service層代碼
/**
* 導(dǎo)出建議列表
* ManuscriptListVO：這個實體包含了我需要的文件夾名稱和文件地址 文件是存在oss里的
* RecordCategory：這個實體是存放文件層級的 下面會展示
*
* @param httpServletResponse http servlet響應(yīng)
* @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 = "壓縮包導(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 -> {
// 獲取該文件層級
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();
}
}
/**
* 封裝文件層級
* HOME_FOLDER_RUL 是配置文件中的預(yù)下載文件夾地址 見下圖
*
* @param manuscript 建言
* @return RecordCategory
*/
private RecordCategory getRecordCategory(DataListVO data) {
// 文件夾名稱：姓名 手機號 防重復(fù)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;
}
預(yù)下載文件夾地址
都是常規(guī)功能，任何目錄層級都可以用
