From 4404af2fb715ad6854156c7eeb8e4ead7f5bf791 Mon Sep 17 00:00:00 2001
From: mashuai <ma_shuai0327@163.com>
Date: Tue, 29 Apr 2025 10:36:15 +0800
Subject: [PATCH] =?UTF-8?q?=E6=BC=8F=E6=B4=9E=E4=BF=AE=E5=A4=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../bonus/sgzb/gen/controller/GenController.java   | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/sgzb-modules/sgzb-gen/src/main/java/com/bonus/sgzb/gen/controller/GenController.java b/sgzb-modules/sgzb-gen/src/main/java/com/bonus/sgzb/gen/controller/GenController.java
index dde0ac7..1e98985 100644
--- a/sgzb-modules/sgzb-gen/src/main/java/com/bonus/sgzb/gen/controller/GenController.java
+++ b/sgzb-modules/sgzb-gen/src/main/java/com/bonus/sgzb/gen/controller/GenController.java
@@ -6,6 +6,7 @@ import java.util.List;
 import java.util.Map;
 import javax.servlet.http.HttpServletResponse;
 
+import com.bonus.sgzb.common.security.utils.SecurityUtils;
 import com.bonus.sgzb.gen.domain.GenTable;
 import com.bonus.sgzb.gen.domain.GenTableColumn;
 import org.apache.commons.io.IOUtils;
@@ -88,9 +89,14 @@ public class GenController extends BaseController
     /**
      * 查询数据表字段列表
      */
+    @RequiresPermissions("tool:gen:query")
     @GetMapping(value = "/column/{tableId}")
     public TableDataInfo columnList(Long tableId)
     {
+        // 只有管理员才能操作
+        if (!isAdmin()) {
+            throw new RuntimeException("没有权限操作");
+        }
         TableDataInfo dataInfo = new TableDataInfo();
         List<GenTableColumn> list = genTableColumnService.selectGenTableColumnListByTableId(tableId);
         dataInfo.setRows(list);
@@ -98,6 +104,14 @@ public class GenController extends BaseController
         return dataInfo;
     }
 
+    /**
+     * 判断是否管理员
+     * @return
+     */
+    private boolean isAdmin() {
+        return SecurityUtils.isAdmin(SecurityUtils.getUserId());
+    }
+
     /**
      * 导入表结构（保存）
      */