Java 根据角色权限查询菜单列表（递归）

        今天遇到一个新的需求，要求后台可以根据用户的角色来配置前端的菜单权限列表。其实就是很常见的需求，后台根据用户的权限来展示对应的导航菜单。

        设计了2个表。DDL如下。

菜单地址表

CREATE TABLE `t_app_bar_url` (
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键',
  `url` varchar(255) DEFAULT NULL COMMENT '地址',
  `name` varchar(255) DEFAULT NULL COMMENT '地址名字',
  `sort` int(4) DEFAULT NULL COMMENT '地址排序',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=19 DEFAULT CHARSET=utf8;

角色权限菜单表

CREATE TABLE `t_app_bar_role` (
  `id` int(11) NOT NULL AUTO_INCREMENT COMMENT '主键',
  `role` varchar(200) DEFAULT NULL COMMENT '用户角色',
  `p_id` int(11) DEFAULT NULL COMMENT '父级id ，没有父节点时为0',
  `url_id` varchar(200) DEFAULT NULL COMMENT '菜单地址id',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=19 DEFAULT CHARSET=utf8;

数据就不给出来了，写完以后，后台根据权限去查询所有的菜单，返回来的是一个List<Map<String,Object>> 结构的数据。

查询的sql如下：

SELECT
	u.id as u_id,
	r.p_id,
	u.`name`,
	u.url,
	u.sort 
FROM
	t_app_bar_role r
	LEFT JOIN t_app_bar_url u ON r.url_id = u.id WHERE r.role = "role1";

结果如下：

 这里返回的数据是这种格式的，但是前端要返回的是树形的结构。如下图。

 正片来了。如何把上面的List<Map<String,Object>>结构转换成一颗或多颗树的形式呢？这里采用递归的思想来实现。

     //3.拿到最顶级的菜单列表数据
    List<Map<String, Object>> rootMenuList = menuByRole.stream()
            .filter(item -> "0".equals(item.get("p_id").toString()))
            .collect(Collectors.toList());
    log.info("最顶级菜单列表为={}", rootMenuList);
    List<Map<String, Object>> menuList = new ArrayList<>(1);
    //4.递归思想，查找所有的子菜单树
    rootMenuList.stream().forEach(rootMenu -> {
      rootMenu.put("child", getChild(menuByRole, rootMenu.get("u_id").toString()));
      menuList.add(rootMenu);
    });
    //5.返回结果
    returnMap.put("menuList", menuList);

递归方法 

 /**
   * 递归查询所有子菜单.
   *
   * @return
   */
  private List<Map<String, Object>> getChild(List<Map<String, Object>> table,
                                             String uId) {
    //1.根据uId查找该菜单的所有子菜单
    List<Map<String, Object>> children = new ArrayList<>(10);
    table.stream().forEach(item -> {
      //判断此节点是否为该菜单的子菜单
      if (uId.equals(item.get("p_id").toString())) {
        children.add(item);
      }
    });
    log.info("排序前={}", children);
    //2.给子菜单排序
    if (!children.isEmpty()) {
      //自定义集合排序
      Collections.sort(children, (o1, o2) -> {
        //从小到大
        return (int) o1.get("sort") > (int) o2.get("sort")
                ? 1 : ((int) o1.get("sort") == (int) o2.get("sort") ? 0 : -1);
      });
    }
    log.info("排序后={}", children);
    //3.遍历所有子菜单，递归查找所有子菜单
    children.stream().forEach(item -> item.put("child", getChild(table, item.get("u_id").toString())));
    return children;
  }
}

 这里采用自定义集合排序的原因是，前端返回的菜单有排序，根据排序字段的值来组装菜单列表。这里的排序很有意思。借鉴了一个博主的想法，地址：https://www.cnblogs.com/xiaolonger/p/4428537.html

    码字不易，觉得有帮助的小伙伴点个赞吧~~