gitee/snowy
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

【更新】演示环境开启则不允许代码生成

Browse Source
This commit is contained in:
xuyuxiang 2021-04-13 11:48:42 +08:00
parent 90c4022db7
commit b9ed92a034
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/modular/controller/CodeGenerateController.java
View File

@ -26,7 +26,9 @@ package com.cn.xiaonuo.generate.modular.controller;
import com.cn.xiaonuo.core.annotion.BusinessLog; import com.cn.xiaonuo.core.annotion.BusinessLog;
import com.cn.xiaonuo.core.annotion.Permission; import com.cn.xiaonuo.core.annotion.Permission;
import com.cn.xiaonuo.core.context.constant.ConstantContextHolder;
import com.cn.xiaonuo.core.enums.LogAnnotionOpTypeEnum; import com.cn.xiaonuo.core.enums.LogAnnotionOpTypeEnum;
import com.cn.xiaonuo.core.exception.DemoException;
import com.cn.xiaonuo.core.pojo.response.ResponseData; import com.cn.xiaonuo.core.pojo.response.ResponseData;
import com.cn.xiaonuo.core.pojo.response.SuccessResponseData; import com.cn.xiaonuo.core.pojo.response.SuccessResponseData;
import com.cn.xiaonuo.generate.modular.param.CodeGenerateParam; import com.cn.xiaonuo.generate.modular.param.CodeGenerateParam;
@ -127,6 +129,10 @@ public class CodeGenerateController {
@PostMapping("/codeGenerate/runLocal") @PostMapping("/codeGenerate/runLocal")
@BusinessLog(title = "代码生成_本地项目", opType = LogAnnotionOpTypeEnum.OTHER) @BusinessLog(title = "代码生成_本地项目", opType = LogAnnotionOpTypeEnum.OTHER)
public ResponseData runLocal(@RequestBody @Validated(CodeGenerateParam.detail.class) CodeGenerateParam codeGenerateParam) { public ResponseData runLocal(@RequestBody @Validated(CodeGenerateParam.detail.class) CodeGenerateParam codeGenerateParam) {
// 演示环境开启，则不允许操作
if (ConstantContextHolder.getDemoEnvFlag()) {
throw new DemoException();
}
this.codeGenerateService.runLocal(codeGenerateParam); this.codeGenerateService.runLocal(codeGenerateParam);
return new SuccessResponseData(); return new SuccessResponseData();
} }
@ -141,6 +147,10 @@ public class CodeGenerateController {
@GetMapping("/codeGenerate/runDown") @GetMapping("/codeGenerate/runDown")
@BusinessLog(title = "代码生成_下载方式", opType = LogAnnotionOpTypeEnum.OTHER) @BusinessLog(title = "代码生成_下载方式", opType = LogAnnotionOpTypeEnum.OTHER)
public void runDown(@Validated(CodeGenerateParam.detail.class) CodeGenerateParam codeGenerateParam, HttpServletResponse response) { public void runDown(@Validated(CodeGenerateParam.detail.class) CodeGenerateParam codeGenerateParam, HttpServletResponse response) {
// 演示环境开启，则不允许操作
if (ConstantContextHolder.getDemoEnvFlag()) {
throw new DemoException();
}
this.codeGenerateService.runDown(codeGenerateParam, response); this.codeGenerateService.runDown(codeGenerateParam, response);
} }
} }