今天遇到一个新的需求,要求后台可以根据用户的角色来配置前端的菜单权限列表。其实就是很常见的需求,后台根据用户的权限来展示对应的导航菜单。
设计了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
码字不易,觉得有帮助的小伙伴点个赞吧~~