递归构建树菜单节点

一、获取全部分类上下级信息

1. /**
2. * 获取所有分类上下级信息
3. */
4. public R<List<ResearchTypeTreeVO>> getTypeTreeList(){
5.     //获取所有分类数据
6.     List<ResearchTypeVO> list = ibResearchTypeService.getSuperList(null);
7.     List<ResearchTypeTreeVO> researchTypeTreeVOList = null;
8.     if(CollectionUtil.isNotEmpty(list)){
9.         researchTypeTreeVOList = buildMenuTrees(list);
10.     }
11.     return R.data(researchTypeTreeVOList);
12. }

复制代码

二、构建数据

1. public static List<ResearchTypeTreeVO> buildMenuTrees(List<ResearchTypeVO> list) {
2.     if(CollectionUtils.isEmpty(list)){
3.         return new ArrayList<>();
4.     }
5.     Map<Long, ResearchTypeVO> researchTypeVOMap = list.stream().collect((Collectors.toMap(ResearchTypeVO::getId, item -> item, (o, n) -> o)));
7.     //上级数据
8.     List<ResearchTypeTreeVO> parentList = new ArrayList<>();
9.     List<ResearchTypeTreeVO> childList = new ArrayList<>();
10.     for (ResearchTypeVO vo : list) {
11.         long parentId = vo.getSuperId();
12.         ResearchTypeTreeVO option = new ResearchTypeTreeVO();
13.         option.setId(vo.getId());
14.         option.setSuperId(parentId);
15.         option.setSuperName(vo.getSuperName());
16.         option.setLabel(vo.getTypeName());
17.         option.setValue(vo.getTypeCode());
18.         // 尝试查找父菜单项，如果不存在则dictBiz是顶级菜单项
19.         ResearchTypeVO researchTypeVO = researchTypeVOMap.get(parentId);
20.         if(Objects.isNull(researchTypeVO)){
21.             parentList.add(option);
22.         }else {
23.             childList.add(option);
24.         }
25.     }
26.     //对查询出来的菜单进行整理 整理为树状
27.     sortMenu(parentList, childList);
28.     return parentList;
29. }

复制代码

三、递归

1. private static void sortMenu(List<ResearchTypeTreeVO> parentList, List<ResearchTypeTreeVO> childMenus) {
2.     for (ResearchTypeTreeVO parent : parentList) {
3.         List<ResearchTypeTreeVO> children = childMenus.stream()
4.                 .filter(child -> child.getSuperId().equals(parent.getId()) )
5.                 .collect(Collectors.toList());
6.         // 设置父节点的子节点列表
7.         parent.setChildren(children);
9.         // 如果子节点列表不为空，则递归地对子节点进行排序
10.         if (!children.isEmpty()) {
11.             sortMenu(children, childMenus);
12.         }
13.     }
14. }

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。