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