需求：主文件夹内有多个子级文件夹，子级文件夹内有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;
}
预下载文件夹地址
都是常规功能，任何目录层级都可以用
