From 882959fb6bb60c72b39cb1d66a49634c4579159e Mon Sep 17 00:00:00 2001
From: 无关风月 <443237572@qq.com>
Date: 星期三, 21 八月 2024 20:29:37 +0800
Subject: [PATCH] Merge branch 'master' of http://120.76.84.145:10101/gitblit/r/java/mx_charging_pile

---
 ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java |   28 ++++++++++++++++++++++------
 1 files changed, 22 insertions(+), 6 deletions(-)

diff --git a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java
index 3f42081..247bbe3 100644
--- a/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java
+++ b/ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/SysUserServiceImpl.java
@@ -21,6 +21,7 @@
 import com.ruoyi.system.domain.SysUserRole;
 import com.ruoyi.system.query.GetSysUserList;
 import com.ruoyi.system.service.ISysConfigService;
+import com.ruoyi.system.service.ISysRoleService;
 import com.ruoyi.system.service.ISysUserRoleService;
 import com.ruoyi.system.service.ISysUserService;
 import org.slf4j.Logger;
@@ -86,6 +87,9 @@
     
     @Resource
     private SiteClient siteClient;
+
+    @Resource
+    private ISysRoleService sysRoleService;
     
     
     
@@ -578,18 +582,30 @@
         for (SysUser sysUser : list) {
             List<SysUserRole> list1 = sysUserRoleService.list(new LambdaQueryWrapper<SysUserRole>().eq(SysUserRole::getUserId, sysUser.getUserId()));
             List<Integer> data1 = userSiteClient.getSiteIds(sysUser.getUserId()).getData();
+            if(null == data1){
+                data1 = new ArrayList<>();
+            }
             for (SysUserRole sysUserRole : list1) {
                 List<Integer> data = roleSiteClient.getSiteIds(sysUserRole.getRoleId()).getData();
+                if (null == data){
+                    continue;
+                }
                 data1.addAll(data);
             }
             Set<Integer> siteIds = new HashSet<>(data1);
+            sysUser.setSiteIds(siteIds.stream().collect(Collectors.toList()));
             List<Site> data = siteClient.getSiteByIds(siteIds.stream().collect(Collectors.toList())).getData();
-            List<String> siteNames = data.stream().map(Site::getName).collect(Collectors.toList());
-            sysUser.setSiteNames(siteNames);
-    
-            Set<Long> collect = list1.stream().map(SysUserRole::getRoleId).collect(Collectors.toSet());
-            List<SysRole> sysRoles = roleMapper.selectBatchIds(collect);
-            List<String> roleNames = sysRoles.stream().map(SysRole::getRoleName).collect(Collectors.toList());
+            if(null != data){
+                List<String> siteNames = data.stream().map(Site::getName).collect(Collectors.toList());
+                sysUser.setSiteNames(siteNames);
+            }
+            List<String> roleNames = new ArrayList<>();
+            for (SysUserRole sysUserRole : list1) {
+                SysRole sysRole = sysRoleService.selectRoleById(sysUserRole.getRoleId());
+                roleNames.add(sysRole.getRoleName());
+            }
+            Long[] roleIds = new Long[]{};
+            sysUser.setRoleIds(list1.stream().map(SysUserRole::getRoleId).collect(Collectors.toList()).toArray(roleIds));
             sysUser.setRoleNames(roleNames);
         }
         return pageInfo.setRecords(list);

--
Gitblit v1.7.1