用户工具类添加角色搜索

ruoyi-system/src/main/java/com/ruoyi/system/utils/UserUtil.java

@@ -1,7 +1,9 @@
 package com.ruoyi.system.utils;
 
 import cn.hutool.extra.spring.SpringUtil;
+import com.ruoyi.common.core.domain.entity.SysRole;
 import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.system.mapper.SysRoleMapper;
 import com.ruoyi.system.service.ISysUserService;
 
 import java.util.ArrayList;
@@ -15,6 +17,7 @@ import java.util.stream.Collectors;
 public class UserUtil {
 
     private static final ISysUserService sysUserService = SpringUtil.getBean(ISysUserService.class);
+    private static final SysRoleMapper roleMapper = SpringUtil.getBean(SysRoleMapper.class);
 
     public static List<SysUser> getListByIds(List<Long> idList) {
         if (idList.size() == 0) {
@@ -78,4 +81,17 @@ public class UserUtil {
         return sysUserService.getUserIdsByRoleKey(roleKey,companyId);
     }
 
+    /**
+     * 获取用户角色列表
+     */
+    public static List<SysRole> getUserRoles(Long userId){
+        return roleMapper.selectRolePermissionByUserId(userId);
+    }
+    /**
+     * 获取用户角色Key列表
+     */
+    public static List<String> getUserRoleKeys(Long userId){
+        return getUserRoles(userId).stream().map(SysRole::getRoleKey).collect(Collectors.toList());
+    }
+
 }