From 5b3a92c80d46306091d3e32a5d56453170deb6b5 Mon Sep 17 00:00:00 2001
From: daiyongfei <974332738@qq.com>
Date: Mon, 22 Sep 2025 16:34:32 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9F=A5=E8=AF=A2=E6=89=80=E6=9C=89=E8=AE=BE?=
 =?UTF-8?q?=E5=A4=87=E7=B1=BB=E5=9E=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/impl/DeviceTypeServiceImpl.java   | 29 +++++++++++++++----
 1 file changed, 23 insertions(+), 6 deletions(-)

diff --git a/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java b/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java
index dccaa495..d23572c9 100644
--- a/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java
+++ b/fys-modules/fys-equipment/src/main/java/com/fuyuanshen/equipment/service/impl/DeviceTypeServiceImpl.java
@@ -98,18 +98,35 @@ public class DeviceTypeServiceImpl extends ServiceImpl<DeviceTypeMapper, DeviceT
     public List<DeviceType> queryDeviceTypes() {
         DeviceTypeQueryCriteria criteria = new DeviceTypeQueryCriteria();
 
-        // 管理员
-        String username = LoginHelper.getUsername();
-        if (!username.equals("admin")) {
-            criteria.setCustomerId(LoginHelper.getUserId());
+        // // 管理员
+        // String username = LoginHelper.getUsername();
+        // if (!username.equals("admin")) {
+        //     criteria.setCustomerId(LoginHelper.getUserId());
+        //
+        //     Long userId = LoginHelper.getUserId();
+        //     criteria.setCustomerId(userId);
+        // }
 
-            Long userId = LoginHelper.getUserId();
-            criteria.setCustomerId(userId);
+        // 角色管理员
+        Long userId = LoginHelper.getUserId();
+        List<SysRoleVo> roles = roleService.selectRolesAuthByUserId(userId);
+        boolean isAdmin = false;
+        if (CollectionUtil.isNotEmpty(roles)) {
+            for (SysRoleVo role : roles) {
+                if (role.getRoleKey().contains("admin")) {
+                    isAdmin = true;
+                    break;
+                }
+            }
+        }
+        if (!isAdmin) {
+            criteria.setCustomerId(LoginHelper.getUserId());
         }
 
         return deviceTypeMapper.findAll(criteria);
     }
 
+
     /**
      * 根据设备类型名称查询设备类型
      *