From 79ad74b1ce07b96a2d93ac20e86ac46e00c9f2a9 Mon Sep 17 00:00:00 2001
From: wl <173@qq.com>
Date: 星期四, 20 十月 2022 17:55:21 +0800
Subject: [PATCH] 修改用户获取权限

---
 ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java |   25 +++++++++++--------------
 1 files changed, 11 insertions(+), 14 deletions(-)

diff --git a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java
index 65afe5c..bb6d7fe 100644
--- a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java
+++ b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminServiceImpl.java
@@ -133,27 +133,24 @@
         }
         //缁戝畾鑿滃崟浜庣敤鎴�
         if (!ids.isEmpty()) {
-            Integer size = 0;
-            Long maxId = null;
+            ArrayList<UmsRoleMenuRelation> umsRoleMenuRelations = new ArrayList<>();
             for (Long id : ids) {
-                if (size < umsRoleMenuRelationMapper
+                umsRoleMenuRelations.addAll(umsRoleMenuRelationMapper
                         .selectList(new LambdaQueryWrapper<UmsRoleMenuRelation>()
-                                .eq(UmsRoleMenuRelation::getRoleId, id)).size()) {
-                    size = umsRoleMenuRelationMapper
-                            .selectList(new LambdaQueryWrapper<UmsRoleMenuRelation>()
-                                    .eq(UmsRoleMenuRelation::getRoleId, id)).size();
-                    maxId = id;
-                }
+                                .eq(UmsRoleMenuRelation::getRoleId, id)));
             }
-            umsRoleMenuRelationMapper
-                    .selectList(new LambdaQueryWrapper<UmsRoleMenuRelation>()
-                            .eq(UmsRoleMenuRelation::getRoleId, maxId))
+            umsRoleMenuRelations
+                    .stream()
+                    .map(item->item.getMenuId())
+                    .collect(Collectors.toList())
+                    .stream()
+                    .distinct()
+                    .collect(Collectors.toList())
                     .stream()
                     .forEach(item -> {
                         AdminMenuRelation adminMenuRelation = new AdminMenuRelation();
                         adminMenuRelation.setAdminId(umsAdmin.getId());
-                        adminMenuRelation.setMenuId(item.getMenuId());
-                        adminMenuRelation.setSort(umsMenuMapper.selectOne(new LambdaQueryWrapper<UmsMenu>().eq(UmsMenu::getId, item.getMenuId())).getSort());
+                        adminMenuRelation.setMenuId(item);
                         adminMenuRelationMapper.insert(adminMenuRelation);
                     });
         }

--
Gitblit v1.8.0