Stream流实现Tree树状结构无限递归

Stream流实现Tree树状结构无限递归

场景：在使用mybatis plus操作多层数据格式，多层遍利操作影响数据查询效率

代码

例子：需要返回给前端多级菜单 省(直辖市)/市/区(县)

/**
     * 查询国家列表
     *
     * @author MYH
     * @date 2022/1/14 16:27
     */
    public List<Region> regionList(String countryName) {
        List<Region> rootList = null;
        if(countryName.equals("china")){
            // 1 :查询所有省市区
            List<Region> regionList = this.list();
            // 2: 找到所有的根节点parent_id = 0 (省级)
             rootList = regionList.stream()
                    .filter(region -> region.getParentId().equals(0)).collect(Collectors.toList());
            //查询所有非父节点(市/区县级)
            List<Region> subList = regionList.stream()
                    .filter(e -> !e.getParentId().equals(0)).collect(Collectors.toList());
            //循环根节点查找对应的子节点
            rootList.forEach(e -> findChildrenNode(e, subList));
        }
        return rootList;
    }

    //递归查找子节点方法, 参数 一个根节点，所有非根节点集合
    private void findChildrenNode(Region region, List<Region> regionList) {
        //通过根节点id和子节点id是否相等，如果相等代表当前根的子集
        List<Region> regions = regionList.stream()
                .filter(region1 -> region1.getParentId().equals(region.getId())).collect(Collectors.toList());
        if (!CollectionUtils.isEmpty(regions)) {
            region.setRegionList(regions);//赋值
            //再次递归查询根节点是否讯在子节点
            regions.forEach(e -> findChildrenNode(e, regionList));
        } else {
            //如果当前没有一个子节点，初始化一个空数组
            region.setRegionList(new ArrayList<>());
        }
    }

总结

第一次写用mp多级遍历后查数据结果用了6~7秒左右，使用上面Stream递归方式结果为1.5秒左右