前言
页面中有一个上传文件的功能,可以新建目录,目录下也可以上传文件,如图:
提示:选中目录点击打包下载将该目录下的文件夹和文件按照层级导出成zip,如果没有选择任务目录或者文件则将所有目录和文件按照层级压缩成zip
Controller
@ApiOperation(value = "打包下载")
@GetMapping(value = "/downloadZip")
public void downloadZip(@ApiParam(name = "id", value= "工程id/目录id/文件id", required = true) @RequestParam(value = "id") Long id, HttpServletResponse response) throws Exception {
projectMaterialsCatalogService.downloadZip(id, response);
}
IService
/**
* 打包下载
*
* @param id 目录id或者文件id
* @param response
* @throws Exception
*/
void downloadZip(Long id, HttpServletResponse response) throws Exception;
ServiceImpl
/**
* 打包下载
*
* @param id 目录id或者文件id
* @param response
* @throws Exception
*/
@Override
public void downloadZip(Long id, HttpServletResponse response) throws Exception {
// 先判断是否是工程, 直接根据工程id查目录
LambdaQueryWrapper<ProjectMaterialsCatalog> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(ProjectMaterialsCatalog::getBusinessId, id);
queryWrapper.eq(ProjectMaterialsCatalog::getIsDeleted, "0");
List<ProjectMaterialsCatalog> projectMaterialsCatalogs = baseMapper.selectList(queryWrapper);
// 生成的ZIP文件路径
String zipFilePath = "";
// 生成的临时文件夹层级
String temFolderPath = "";
// 如果是工程id则需要将所有目录数据进行打包下载
if (CollectionUtils.isNotEmpty(projectMaterialsCatalogs)) {
// 多个1级文件夹放在临时文件夹下,以当前工程id为文件夹
temFolderPath = filesPath + File.separator + id;
for (ProjectMaterialsCatalog catalog : projectMaterialsCatalogs) {
// 先创建目录再创建文件
String catalogName = catalog.getCatalogName();
// 先根据目录生成文件夹包含层级结构
genCatalog(catalog, temFolderPath + File.separator + catalogName);
}
zipFilePath = temFolderPath + ".zip";
// 对生成的文件夹进行压缩
FileUtil.folderToZip(zipFilePath, temFolderPath, true);
} else {
// 再判断是目录还是文件
ProjectMaterialsCatalog catalog = baseMapper.selectById(id);
if (catalog != null) {
// 先创建目录再创建文件
String catalogName = catalog.getCatalogName();
temFolderPath = filesPath + File.separator + catalogName;
zipFilePath = temFolderPath + ".zip";
// 先根据目录生成文件夹包含层级结构
genCatalog(catalog, temFolderPath);
// 对生成的文件夹进行压缩
FileUtil.folderToZip(zipFilePath, temFolderPath, true);
} else {
// 说明是文件,将此文件打包为zip
ProjectMaterialsFiles fileInfo = projectMaterialsFilesMapper.selectById(id);
// 获取文件路径
String filePath = filesPath + File.separator + fileInfo.getFilePath();
// 获取文件名前缀
String prefix = cn.hutool.core.io.FileUtil.getPrefix(fileInfo.getFileName());
// 生成的ZIP文件路径
zipFilePath = filesPath + File.separator + prefix + ".zip";
// 对文件进行压缩
List<File> fileList = new ArrayList<>();
fileList.add(new File(filePath));
FileUtil.fileToZip(zipFilePath, fileList);
}
}
try {
// 下载生成的文件zip
File zipFile = new File(zipFilePath);
response.setCharacterEncoding("UTF-8");
response.setContentType("application/zip");
response.setHeader("Content-Disposition", "attachment; filename=" + URLEncoder.encode(zipFile.getName(), "UTF-8"));
InputStream in = new FileInputStream(zipFile);
// 获得输出流--通过response获得的输出流 用于向客户端写内容
ServletOutputStream out = response.getOutputStream();
// 文件拷贝代码
int len = 0;
byte[] buffer = new byte[1024];
while ((len = in.read(buffer)) != -1) {
out.write(buffer, 0, len);
}
out.flush();
out.close();
in.close();
response.flushBuffer();
} catch (IOException e) {
throw new RuntimeException(e);
} finally {
// 删除压缩包
cn.hutool.core.io.FileUtil.del(zipFilePath);
// 删除创建的文件夹
if (StringUtils.isNotBlank(temFolderPath)) {
cn.hutool.core.io.FileUtil.del(temFolderPath);
}
}
}
/**
* 递归生成文件夹和文件
*
* @param rootCatalog
* @param catalogPath
*/
public void genCatalog(ProjectMaterialsCatalog rootCatalog, String catalogPath) throws Exception {
File file = new File(catalogPath);
if (!file.exists()) {
file.mkdirs();
}
// 判断该目录下是否有文件
List<ProjectMaterialsFiles> filesList = projectMaterialsFilesMapper.getFileByCatalogId(rootCatalog.getId());
if (CollectionUtils.isNotEmpty(filesList)) {
// 循环创建文件到此目录下
for (ProjectMaterialsFiles f : filesList) {
File sourceFile = new File(filesPath + File.separator + f.getFilePath());
// 源文件不存在
if (!sourceFile.exists()) {
continue;
}
File destFile = new File(catalogPath + File.separator + f.getFileName());
Files.copy(sourceFile, destFile);
}
}
LambdaQueryWrapper<ProjectMaterialsCatalog> queryWrapper = new LambdaQueryWrapper<>();
queryWrapper.eq(ProjectMaterialsCatalog::getIsDeleted, "0");
queryWrapper.eq(ProjectMaterialsCatalog::getParentId, rootCatalog.getId());
List<ProjectMaterialsCatalog> catalogList = baseMapper.selectList(queryWrapper);
if (CollectionUtils.isNotEmpty(catalogList)) {
for (ProjectMaterialsCatalog catalog : catalogList) {
catalogPath += File.separator + catalog.getCatalogName();
genCatalog(catalog, catalogPath);
}
}
}
FileUtil
/**
* 文件夹压缩成ZIP
*
* @param zipFileName 生成压缩文件夹路径
* @param sourceFileName 要压缩的文件路径
* @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
* false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败)
* @throws RuntimeException 压缩失败会抛出运行时异常
*/
public static Boolean folderToZip(String zipFileName, String sourceFileName, boolean KeepDirStructure) {
Boolean result = true;
long start = System.currentTimeMillis();//开始
ZipOutputStream zos = null;
try {
FileOutputStream fileOutputStream = new FileOutputStream(zipFileName);
zos = new ZipOutputStream(fileOutputStream);
File sourceFile = new File(sourceFileName);
compress(sourceFile, zos, sourceFile.getName(), KeepDirStructure);
long end = System.currentTimeMillis();//结束
System.out.println("压缩完成,耗时:" + (end - start) + " 毫秒");
} catch (Exception e) {
result = false;
e.printStackTrace();
} finally {
if (zos != null) {
try {
zos.close();
} catch (IOException e) {
e.getStackTrace();
}
}
}
return result;
}
/**
* 文件压缩成ZIP 一次性压缩多个文件
*
* @param srcFiles 需要压缩的文件列表
* @param zipFileName 压缩文件输出
* @throws RuntimeException 压缩失败会抛出运行时异常
*/
public static void fileToZip(String zipFileName, List<File> srcFiles) throws Exception {
long start = System.currentTimeMillis();
ZipOutputStream zos = null;
try {
FileOutputStream fileOutputStream = new FileOutputStream(zipFileName);
zos = new ZipOutputStream(fileOutputStream);
for (File srcFile : srcFiles) {
compress(srcFile, zos, srcFile.getName(), true);
}
long end = System.currentTimeMillis();
System.out.println("压缩完成,耗时:" + (end - start) + " 毫秒");
} catch (Exception e) {
throw new RuntimeException("zip error from ZipUtils", e);
} finally {
if (zos != null) {
try {
zos.close();
} catch (IOException e) {
e.printStackTrace();
}
}
}
}
/**
* 递归压缩方法
*
* @param sourceFile 源文件
* @param zos zip输出流
* @param name 压缩后的名称
* @param KeepDirStructure 是否保留原来的目录结构,true:保留目录结构;
* false:所有文件跑到压缩包根目录下(注意:不保留目录结构可能会出现同名文件,会压缩失败)
* @throws Exception
*/
public static void compress(File sourceFile, ZipOutputStream zos, String name,
boolean KeepDirStructure) throws Exception {
if (sourceFile.isFile()) {
// 向zip输出流中添加一个zip实体,构造器中name为zip实体的文件的名字
zos.putNextEntry(new ZipEntry(name));
// copy文件到zip输出流中
int len;
byte[] buf = new byte[1024];
FileInputStream in = new FileInputStream(sourceFile);
while ((len = in.read(buf)) != -1) {
zos.write(buf, 0, len);
}
// Complete the entry
zos.closeEntry();
in.close();
} else {
File[] listFiles = sourceFile.listFiles();
if (listFiles == null || listFiles.length == 0) {
// 需要保留原来的文件结构时,需要对空文件夹进行处理
if (KeepDirStructure) {
// 空文件夹的处理
zos.putNextEntry(new ZipEntry(name + "/"));
// 没有文件,不需要文件的copy
zos.closeEntry();
}
} else {
for (File file : listFiles) {
// 判断是否需要保留原来的文件结构
if (KeepDirStructure) {
// 注意:file.getName()前面需要带上父文件夹的名字加一斜杠,
// 不然最后压缩包中就不能保留原来的文件结构,即:所有文件都跑到压缩包根目录下了
compress(file, zos, name + "/" + file.getName(), KeepDirStructure);
} else {
compress(file, zos, file.getName(), KeepDirStructure);
}
}
}
}
}