获取用户菜单树

请求地址

返回结果

返回结果是一个树状数组，对应左侧菜单

具体代码

@GetMapping("getRouters")
    public AjaxResult getRouters()
    {
        Long userId = SecurityUtils.getUserId();
        List<SysMenu> menus = menuService.selectMenuTreeByUserId(userId);
        return AjaxResult.success(menuService.buildMenus(menus));
    }
}

逻辑非常简单，从spring security域对象内获取用户id，然后传入selectMenuTreeByUserId方法中去获取相应的菜单数组，最后返回。

具体实现

点开selectMenuTreeByUserId方法

@Override
public List<SysMenu> selectMenuTreeByUserId(Long userId)
{
    List<SysMenu> menus = null;
    if (SecurityUtils.isAdmin(userId))
    {
        menus = menuMapper.selectMenuTreeAll();
    }
    else
    {
        menus = menuMapper.selectMenuTreeByUserId(userId);
    }
    return getChildPerms(menus, 0);
}

逻辑也非常简单，管理员拥有所有菜单，非管理员需要传入用户id查询对应的菜单，重点是selectMenuTreeByUserId和getChildPerms方法。

selectMenuTreeByUserId(Long id)

点开该方法发现他是一个mapper层的接口

我们找到它对应的xml文件，在ruoyi-system文件夹下resources之中

<select id="selectMenuTreeByUserId" parameterType="Long" resultMap="SysMenuResult">
   select distinct m.menu_id, m.parent_id, m.menu_name, m.path, m.component, m.`query`, m.visible, m.status, ifnull(m.perms,'') as perms, m.is_frame, m.is_cache, m.menu_type, m.icon, m.order_num, m.create_time
   from sys_menu m
       left join sys_role_menu rm on m.menu_id = rm.menu_id
       left join sys_user_role ur on rm.role_id = ur.role_id
       left join sys_role ro on ur.role_id = ro.role_id
       left join sys_user u on ur.user_id = u.user_id
   where u.user_id = #{userId} and m.menu_type in ('M', 'C') and m.status = 0  AND ro.status = 0
   order by m.parent_id, m.order_num
</select>

可以看到这段sql就是查询用户对应的菜单列表，而"M","C"是非管理员类别菜单，status为0是启用状态。

List getChildPerms(List list, int parentId)

通过sql语句可以知道我们查询的用户菜单列表是无序的，并不是树状的，我们需要把它变为树状的再进行返回。

这里使用了4个函数去具体实现树的转化，主要用了递归的思想。

1 找寻parentId为0的节点，也就是最上层的节点
2 将父节点传入recursionfn函数(每一个节点都有列表来存放孩子节点，此函数完善节点的孩子列表属性)
3 使用getChildList获取父节点的孩子节点列表
4 再将孩子节点列表进行遍历，用hasChild判断孩子节点是否有孩子节点
5 如果有再使用recursionfn函数。一直递归到不存在孩子节点的节点

public List<SysMenu> getChildPerms(List<SysMenu> list, int parentId)
{
   List<SysMenu> returnList = new ArrayList<SysMenu>();
   for (Iterator<SysMenu> iterator = list.iterator(); iterator.hasNext();)
   {
       SysMenu t = (SysMenu) iterator.next();
       // 一、根据传入的某个父节点ID,遍历该父节点的所有子节点
       if (t.getParentId() == parentId)
       {
           recursionFn(list, t);
           returnList.add(t);
       }
   }
   return returnList;
}

/**
* 递归列表
* 
* @param list 分类表
* @param t 子节点
*/
private void recursionFn(List<SysMenu> list, SysMenu t)
{
   // 得到子节点列表
   List<SysMenu> childList = getChildList(list, t);
   t.setChildren(childList);
   for (SysMenu tChild : childList)
   {
       if (hasChild(list, tChild))
       {
           recursionFn(list, tChild);
       }
   }
}

/**
* 得到子节点列表
*/
private List<SysMenu> getChildList(List<SysMenu> list, SysMenu t)
{
   List<SysMenu> tlist = new ArrayList<SysMenu>();
   Iterator<SysMenu> it = list.iterator();
   while (it.hasNext())
   {
       SysMenu n = (SysMenu) it.next();
       if (n.getParentId().longValue() == t.getMenuId().longValue())
       {
           tlist.add(n);
       }
   }
   return tlist;
}

/**
* 判断是否有子节点
*/
private boolean hasChild(List<SysMenu> list, SysMenu t)
{
   return getChildList(list, t).size() > 0;
}