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);
     }
 }