首先是表结构需要有id,与parentId这两个字段,不然无法构成父子节点关系
查出当前菜单中所有的菜单,使用递归算法生成树桩菜单
private List<MenuVo> respMenu(List<MenuVo> menus){
// 最后的结果
// HashMap<String, Object> menuMap = new HashMap<>();
List<MenuVo> menuList = new ArrayList<MenuVo>();
// 先找到所有的一级菜单
for (int i = 0; i < menus.size(); i++) {
// 一级菜单parentId = 0
if (menus.get(i).getParentId().equals("0")) {
menuList.add(menus.get(i));
}
}
// 为一级菜单设置子菜单,getChild是递归调用的
for (MenuVo menu : menuList) {
menu.setChild(getChild(menu.getId(), menus));
}
return menuList;
}
/**
* 递归查找子菜单
*
* @param id
* 当前菜单id
* @param rootMenu
* 要查找的列表
* @return
*/
private List<MenuVo> getChild(String id, List<MenuVo> rootMenu) {
// 子菜单
List<MenuVo> childList = new ArrayList<>();
for (MenuVo menu : rootMenu) {
// 遍历所有节点,将父菜单id与传过来的id比较
if (menu.getParentId().equals(id)) {
childList.add(menu);
}
}
// 把子菜单的子菜单再循环一遍
for (MenuVo menu : childList) {// 没有url子菜单还有子菜单
menu.setChild(getChild(menu.getId(), rootMenu));
} // 递归退出条件
if (childList.size() == 0) {
return null;
}
return childList;
}