数据库存储的菜单结果:
parentid为0的都是根节点,也就是一级菜单,后面的子菜单的parentid为父菜单的ID。
MenuDTO类(菜单类)的结构:
public class MenuDTO {
private Integer id;
private String content;
private Integer parentid;
private Date createtime;
private Integer num;
private List<MenuDTO> childs;
//get和set方法省略
}
采用递归方法,遍历成树级结构菜单。
//获得树级结构菜单
public List<MenuDTO> getMenuList() throws IOException {
//拿到菜单的所有数据
List<MenudTO> list=menuMapper.getMenuList();
//存储根节点的菜单,即一级菜单
List<MenuDTO> rootlist=new ArrayList<>();
//遍历所有数据,找到根节点菜单
for (MenuDTO menuDTO:
list) {
if(menuDTO.getParentid().equals(0)){
//找到根节点菜单的时候,寻找这个根节点菜单下的子节点菜单。
findChilds(menuDTO,list);
//添加到根节点的列表中
rootlist.add(menuDTO);
}
}
return rootlist;
}
private void findChilds(MenuDTO root,List<MenuDTO> list){
List<MenuDTO> childlist=new ArrayList<>();
//遍历所有数据,找到是入参父节点的子节点的数据,然后加到childlist集合中。
for (MenuDTO menu :
list) {
if (root.getId().equals(menu.getParentid()))
childlist.add(menu);
}
//若子节点不存在,那么就不必再遍历子节点中的子节点了 直接返回。
if(childlist.size()==0)
return;
//设置父节点的子节点列表
root.setChilds(childlist);
//若子节点存在,接着递归调用该方法,寻找子节点的子节点。
for (MenuDTO childs :
childlist) {
findChilds(childs, list);
}
}
结果:
总结: 先拿到所有的菜单数据,然后遍历菜单数据,找到根节点,找到根节点。然后调用找到子节点的方法,在子节点方法中递归调用自己,也就是说如果这个节点有子节点,那么递归调用方法找到子节点的子节点,直到某个节点下面没有子节点。