From b9ed92a03417bbfce82969b9cd91913617e430bb Mon Sep 17 00:00:00 2001
From: xuyuxiang <YJLP2014>
Date: Tue, 13 Apr 2021 11:48:42 +0800
Subject: [PATCH] =?UTF-8?q?=E3=80=90=E6=9B=B4=E6=96=B0=E3=80=91=E6=BC=94?=
 =?UTF-8?q?=E7=A4=BA=E7=8E=AF=E5=A2=83=E5=BC=80=E5=90=AF=E5=88=99=E4=B8=8D?=
 =?UTF-8?q?=E5=85=81=E8=AE=B8=E4=BB=A3=E7=A0=81=E7=94=9F=E6=88=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../modular/controller/CodeGenerateController.java     | 10 ++++++++++
 1 file changed, 10 insertions(+)

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 2a28ba85..59a98c22 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
@@ -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);
     }
 }