diff --git a/snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/file/service/impl/DevFileServiceImpl.java b/snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/file/service/impl/DevFileServiceImpl.java index cf949ba4..e4feaa5b 100644 --- a/snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/file/service/impl/DevFileServiceImpl.java +++ b/snowy-plugin/snowy-plugin-dev/src/main/java/vip/xiaonuo/dev/modular/file/service/impl/DevFileServiceImpl.java @@ -99,7 +99,13 @@ public class DevFileServiceImpl extends ServiceImpl 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; diff --git a/snowy-plugin/snowy-plugin-gen/src/main/java/vip/xiaonuo/gen/modular/basic/service/impl/GenBasicServiceImpl.java b/snowy-plugin/snowy-plugin-gen/src/main/java/vip/xiaonuo/gen/modular/basic/service/impl/GenBasicServiceImpl.java index 6a4bb502..fb306374 100644 --- a/snowy-plugin/snowy-plugin-gen/src/main/java/vip/xiaonuo/gen/modular/basic/service/impl/GenBasicServiceImpl.java +++ b/snowy-plugin/snowy-plugin-gen/src/main/java/vip/xiaonuo/gen/modular/basic/service/impl/GenBasicServiceImpl.java @@ -344,17 +344,21 @@ public class GenBasicServiceImpl extends ServiceImpl 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)