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

pull/22/head
xuyuxiang 2021-04-13 11:48:42 +08:00
parent 90c4022db7
commit b9ed92a034
1 changed files with 10 additions and 0 deletions

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