gitee
/
snowy
mirror of https://gitee.com/xiaonuobase/snowy
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

【更新】优化返回文件流的接口的异常处理

pull/120/head
xuyuxiang 2023-05-10 09:31:38 +08:00
parent 7013fd1568
commit c4dae1ea1a
2 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/file/service/impl/DevFileServiceImpl.java
View File

@ -99,7 +99,13 @@ public class DevFileServiceImpl extends ServiceImpl<DevFileMapper, DevFile> impl
@Override
public void download(DevFileIdParam devFileIdParam, HttpServletResponse response) throws IOException {
DevFile devFile = this.queryEntity(devFileIdParam.getId());
DevFile devFile;
try {
devFile = this.queryEntity(devFileIdParam.getId());
} catch (Exception e) {
CommonResponseUtil.renderError(response, e.getMessage());
return;
}
if(!devFile.getEngine().equals(DevFileEngineTypeEnum.LOCAL.getValue())) {
CommonResponseUtil.renderError(response, "非本地文件不支持此方式下载，id值为：" + devFile.getId());
return;

24
snowy-plugin/snowy-plugin-gen/src/main/java/vip/xiaonuo/gen/modular/basic/service/impl/GenBasicServiceImpl.java
View File

@ -344,17 +344,21 @@ public class GenBasicServiceImpl extends ServiceImpl<GenBasicMapper, GenBasic> i
@Transactional(rollbackFor = Exception.class)
@Override
public void execGenZip(GenBasicIdParam genBasicIdParam, HttpServletResponse response) throws IOException {
File tempFolder = this.genTempFolder(genBasicIdParam, response, true);
if(tempFolder == null) {
CommonResponseUtil.renderError(response, "代码生成基础不存在，id值为：" + genBasicIdParam.getId());
return;
try {
File tempFolder = this.genTempFolder(genBasicIdParam, response, true);
if(tempFolder == null) {
CommonResponseUtil.renderError(response, "代码生成基础不存在，id值为：" + genBasicIdParam.getId());
return;
}
// 压缩
File zip = ZipUtil.zip(tempFolder);
// 压缩完毕删除临时目录
FileUtil.del(tempFolder);
// 下载
CommonDownloadUtil.download(zip, response);
} catch (Exception e) {
CommonResponseUtil.renderError(response, e.getMessage());
}
// 压缩
File zip = ZipUtil.zip(tempFolder);
// 压缩完毕删除临时目录
FileUtil.del(tempFolder);
// 下载
CommonDownloadUtil.download(zip, response);
}
@Transactional(rollbackFor = Exception.class)