src/main/java/com/ruoyi/project/system/controller/SysUserController.java
@@ -120,10 +120,7 @@ List<SysRole> roles = roleService.selectRoleAll(); ajax.put("roles", SysUser.isAdmin(userId) ? roles : roles.stream().filter(r -> !r.isAdmin()).collect(Collectors.toList())); ajax.put("posts", postService.selectPostAll()); SysUserDeptVo sysUserDeptVo = new SysUserDeptVo(); sysUserDeptVo.setUserId(userId); List<SysUserDeptVo> sysUserDeptVos = userDeptService.userLoginFacotryList(sysUserDeptVo); ajax.put("deptIds",sysUserDeptVos.stream().map(SysUserDeptVo::getDeptId).collect(Collectors.toList())); ajax.put("deptIds", userService.selectDeptIdsByUserId(userId)); return ajax; } src/main/java/com/ruoyi/project/system/service/ISysUserService.java
@@ -217,4 +217,11 @@ * @return */ int bindUserDept(SysUser user); /** * 根据用户ID查询所有关联的部门ID * @param userId * @return */ List<Long> selectDeptIdsByUserId(Long userId); } src/main/java/com/ruoyi/project/system/service/impl/SysUserServiceImpl.java
@@ -560,4 +560,11 @@ } return user.getDeptIds().length; } @Override public List<Long> selectDeptIdsByUserId(Long userId) { LambdaQueryWrapper<SysUserDept> queryWrapper = new LambdaQueryWrapper<>(); queryWrapper.eq(SysUserDept::getUserId, userId); return sysUserDeptMapper.selectList(queryWrapper).stream().map(SysUserDept::getDeptId).collect(Collectors.toList()); } }