JAVA 实现树形结构数据组装

Coder&2012

已于 2022-05-13 17:42:59 修改

阅读量773

点赞数

分类专栏： JavaSE 文章标签： java javaee java-ee

于 2022-04-26 11:15:31 首次发布

本文链接：https://blog.csdn.net/SYQ20122012/article/details/124423451

版权

JavaSE 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

JAVA 实现树形结构数据组装

查出所有符合条件的数据

    public List<MenusDto> queryTree(Boolean hasResource){
        QueryWrapper<SysBasicMenu> wrapper = new QueryWrapper<>();
        wrapper.eq("status", CommonConstant.Field_Status_Normal);
        List<SysBasicMenu> sysBasicMenus = super.list(wrapper);
        return this.convertMenusTree(sysBasicMenus);
    }

树形结构数据组装

    private List<MenusDto> convertMenusTree(List<SysBasicMenu> sysBasicMenus) {
        List<MenusDto> result = new ArrayList<>();
        Map<Integer, MenusDto> map = new HashMap<>();
        sysBasicMenus.forEach(sysBasicMenu -> {
            MenusDto menusDto = new MenusDto();
            BeanUtils.copyProperties(sysBasicMenu, menusDto);
            map.put(menusDto.getId(), menusDto);
        });
        for (Map.Entry<Integer, MenusDto> entry : map.entrySet()) {
            if ("0".equals(entry.getValue().getFlg())) {
                result.add(entry.getValue());
            }
            if (null != map.get(entry.getValue().getParentId())) {
                map.get(entry.getValue().getParentId()).getChildren().add(entry.getValue());
            }
        }
        sortMenus(result);
        return result;
    }

树形结构数据排序

    private void sortMenus(List<MenusDto> menusListDtos) {
        if (!CollectionUtils.isEmpty(menusListDtos)) {
            menusListDtos.sort((o1, o2) -> o1.getOrders().compareTo(o2.getOrders()));
            menusListDtos.forEach(menusListDto -> {
                sortMenus(menusListDto.getChildren());
            });
        }
    }