From 3a9aa4611897bd00a4717a3ab599d02759eb3d62 Mon Sep 17 00:00:00 2001
From: TinyAnts <tinyants@163.com>
Date: Wed, 17 Aug 2022 18:52:31 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=94=B9=E7=94=9F=E4=BA=A7=E7=8E=AF?=
 =?UTF-8?q?=E5=A2=83=E6=8B=A6=E6=88=AA=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../com/hxkj/generator/controller/GenController.java   | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java b/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java
index ef4bdae7..c1f355dd 100644
--- a/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java
+++ b/server/like-generator/src/main/java/com/hxkj/generator/controller/GenController.java
@@ -145,6 +145,11 @@ public class GenController {
      */
     @GetMapping("/genCode")
     public Object genCode(String tables) {
+        String production = YmlUtil.get("like.production");
+        if (StringUtil.isNotEmpty(production) && production.equals("true")) {
+            throw new OperateException("抱歉,演示环境不允许操作！");
+        }
+
         Assert.notNull(tables, "请选择要生成的表");
         String[] tableNames = tables.split(",");
         for (String tableName : tableNames) {
@@ -162,11 +167,6 @@ public class GenController {
      */
     @GetMapping("/downloadCode")
     public void downloadCode(HttpServletResponse response, String tables) throws IOException {
-        String production = YmlUtil.get("like.production");
-        if (StringUtil.isNotEmpty(production) && production.equals("true")) {
-            throw new OperateException("抱歉,演示环境不允许操作！");
-        }
-
         Assert.notNull(tables, "请选择要生成的表");
         String[] tableNames = tables.split(",");
         byte[] data = iGenerateService.downloadCode(tableNames);