From 36a753a60428a27130b1dae019b01b8f8edb86c1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BE=90=E7=8E=89=E7=A5=A5?= Date: Fri, 25 Dec 2020 22:42:49 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90=E6=BC=94?= =?UTF-8?q?=E7=A4=BA=E7=8E=AF=E5=A2=83=E7=8A=B6=E6=80=81=E5=88=A4=E6=96=AD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../controller/CodeGenerateController.java | 16 +++++++++++++++- 1 file changed, 15 insertions(+), 1 deletion(-) diff --git a/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/modular/controller/CodeGenerateController.java b/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/modular/controller/CodeGenerateController.java index d9d3b351..a31e4f9d 100644 --- a/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/modular/controller/CodeGenerateController.java +++ b/xiaonuo-base/xiaonuo-gen/src/main/java/com/cn/xiaonuo/generate/modular/controller/CodeGenerateController.java @@ -27,7 +27,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.page.PageResult; import com.cn.xiaonuo.core.pojo.response.ResponseData; import com.cn.xiaonuo.core.pojo.response.SuccessResponseData; @@ -35,7 +37,11 @@ import com.cn.xiaonuo.generate.modular.entity.CodeGenerate; import com.cn.xiaonuo.generate.modular.param.CodeGenerateParam; import com.cn.xiaonuo.generate.modular.service.CodeGenerateService; import org.springframework.validation.annotation.Validated; -import org.springframework.web.bind.annotation.*; +import org.springframework.web.bind.annotation.GetMapping; +import org.springframework.web.bind.annotation.PostMapping; +import org.springframework.web.bind.annotation.RequestBody; +import org.springframework.web.bind.annotation.RestController; + import javax.annotation.Resource; import javax.servlet.http.HttpServletResponse; import java.util.List; @@ -130,6 +136,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(); } @@ -144,6 +154,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); } }