From 6c82b0f944a0d3547659db5ec1f640a9ec7d3fd3 Mon Sep 17 00:00:00 2001 From: xiaoyong931011 <15274802129@163.com> Date: Wed, 15 Jun 2022 11:08:52 +0800 Subject: [PATCH] 20220606 --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java | 19 ++++++++++++++----- 1 files changed, 14 insertions(+), 5 deletions(-) diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java index 04baff4..54e461c 100644 --- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java +++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java @@ -251,11 +251,20 @@ List<UserRoleEntity> userRoleEntities = userRoleMapper.selectList(userRoleQueryWrapper); if(CollUtil.isNotEmpty(userRoleEntities)){ List menuIds = new ArrayList<Long>(); - for(UserRoleEntity userRoleEntity : userRoleEntities){ - List<RoleMenuEntity> roleMenuEntities = roleMenuMapper.selectListByRoleId(userRoleEntity.getRoleId()); - if(CollUtil.isNotEmpty(roleMenuEntities)){ - for(RoleMenuEntity roleMenuEntity : roleMenuEntities){ - menuIds.add(roleMenuEntity.getMenuId()); + if(UserEntity.USERNAME_DEFAULT.equals(loginUser.getUsername())){ + List<MenuEntity> menuEntities = menuMapper.selectList(new QueryWrapper<>()); + if(CollUtil.isNotEmpty(menuEntities)){ + for(MenuEntity menuEntity : menuEntities){ + menuIds.add(menuEntity.getId()); + } + } + }else{ + for(UserRoleEntity userRoleEntity : userRoleEntities){ + List<RoleMenuEntity> roleMenuEntities = roleMenuMapper.selectListByRoleId(userRoleEntity.getRoleId()); + if(CollUtil.isNotEmpty(roleMenuEntities)){ + for(RoleMenuEntity roleMenuEntity : roleMenuEntities){ + menuIds.add(roleMenuEntity.getMenuId()); + } } } } -- Gitblit v1.9.1