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