jiangqs
2023-06-26 b4882b114f082e010d4cc3359d78d18095ef3b83
ruoyi-modules/ruoyi-system/src/main/java/com/ruoyi/system/service/impl/sys/SysMenuServiceImpl.java
@@ -14,9 +14,10 @@
import com.ruoyi.system.mapper.sys.SysRoleMapper;
import com.ruoyi.system.mapper.sys.SysRoleMenuMapper;
import com.ruoyi.system.service.sys.ISysMenuService;
import org.springframework.beans.factory.annotation.Autowired;
import com.ruoyi.system.service.sys.ISysUserService;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.*;
import java.util.stream.Collectors;
@@ -30,14 +31,19 @@
{
    public static final String PREMISSION_STRING = "perms[\"{0}\"]";
    @Autowired
    @Resource
    private SysMenuMapper menuMapper;
    @Autowired
    @Resource
    private SysRoleMapper roleMapper;
    @Autowired
    @Resource
    private SysRoleMenuMapper roleMenuMapper;
    @Resource
    private ISysUserService sysUserService;
    /**
     * 根据用户查询系统菜单列表
@@ -50,6 +56,21 @@
    {
        return selectMenuList(new SysMenu(), userId);
    }
    /**
     * 根据用户查询系统菜单列表
     *
     * @param userId 用户ID
     * @return 菜单列表
     */
    @Override
    public List<SysMenu> selectDeptMenuList(Long userId){
        SysUser sysUser = sysUserService.selectUserById(userId);
        Long deptId = sysUser.getDeptId();
        List<SysMenu> menuList = menuMapper.selectDeptMenuList(deptId);
        return menuList;
    }
    /**
     * 根据用户查询系统菜单列表
     *
@@ -59,6 +80,19 @@
    @Override
    public List<SysMenu> selectStaffMenuList(Long userId){
        return selectStaffMenuList(new SysMenu(), userId);
    }
    /**
     * 根据用户查询系统菜单列表
     *
     * @param userId 用户ID
     * @return 菜单列表
     */
    @Override
    public List<SysMenu> selectStaffDeptMenuList(Long userId){
        SysUser sysUser = sysUserService.selectUserById(userId);
        Long deptId = sysUser.getDeptId();
        return menuMapper.selectStaffDeptMenuList(deptId);
    }
    /**
     * 查询系统菜单列表
@@ -78,7 +112,13 @@
        else
        {
            menu.getParams().put("userId", userId);
            SysUser sysUser = sysUserService.selectUserById(userId);
            if(sysUser.getUserName().equals("admin")||sysUser.getUserName().equals("hongruitang")){
            menuList = menuMapper.selectMenuListByUserId(menu);
            }else{
                menuList = menuMapper.selectMenuListByStaffUserId(menu);
            }
        }
        return menuList;
    }
@@ -115,7 +155,13 @@
    @Override
    public Set<String> selectMenuPermsByUserId(Long userId)
    {
        List<String> perms = menuMapper.selectMenuPermsByUserId(userId);
        SysUser sysUser = sysUserService.selectUserById(userId);
        List<String> perms = null;
        if(sysUser.getUserName().equals("admin")||sysUser.getUserName().equals("hongruitang")){
            perms = menuMapper.selectMenuPermsByUserId(userId);
        }else{
            perms = menuMapper.selectMenuPermsByStaffUserId(userId);
        }
        Set<String> permsSet = new HashSet<>();
        for (String perm : perms)
        {