java递归查询子节点

//module为根节点id
public List<ModuleChildOutVo> getModuleByProjectNo(String moduleId) {
        //查询该根模块id下所有模块
        LambdaQueryWrapper<Module> moduleLbq = new LambdaQueryWrapper<>();
        //设置查询条件
        moduleLbq.eq(Module::getId,moduleId);
        //查询
        List<Module> moduleList = this.list(moduleLbq);
        //存放根节点
        List<ModuleChildOutVo> rootModule = new ArrayList<>();
        //循环模块集合把第一级过滤出来
        moduleList.stream().forEach(module->{
            if(module.getLevel().equals("0")){
                ModuleChildOutVo moduleChildOutVo = new ModuleChildOutVo();
                BeanUtils.copyProperties(module,moduleChildOutVo);
                rootModule.add(moduleChildOutVo);
            }
        });
        //为根模块设置子模块,getClild是递归调用的
        for (ModuleChildOutVo module : rootModule) {
            /* 获取根节点下的所有子节点 使用getChild方法*/
            List<ModuleChildOutVo> childList = getChild(module.getId(), moduleList);
            module.setModuleChilds(childList);//给根节点设置子节点
        }
        return rootModule;
    }


    /**
     * 获取子模块
     * @param id 父节点id
     * @param allModule 所有模块列表
     * @return 每个根节点下,所有子模块列表
     */
    public List<ModuleChildOutVo> getChild(Integer id, List<Module> allModule){
        //子模块
        List<ModuleChildOutVo> childList = new ArrayList<ModuleChildOutVo>();
        for (Module module : allModule) {
            // 遍历所有节点,将所有模块的父id与传过来的根节点的id比较
            //相等说明:为该根节点的子节点。
            if(module.getParentId().equals(id+"")){
                ModuleChildOutVo moduleChildOutVo = new ModuleChildOutVo();
                BeanUtils.copyProperties(module,moduleChildOutVo);
                childList.add(moduleChildOutVo);
            }
        }
        //递归
        for (ModuleChildOutVo nav : childList) {
            nav.setModuleChilds(getChild(nav.getId(), allModule));
        }
        //Collections.sort(childList,order());//排序
        //如果节点下没有子节点,返回一个空List(递归退出)
        if(childList.size() == 0){
            return new ArrayList<ModuleChildOutVo>();
        }
        return childList;
    }

/** * 根据等级查询类目树 * * @param level * @return */ @Override public List queryCategoryTree(Integer level) { //查询当前级别下类目 List list = categoryDAO.list(level); //组装好的类目树,返回前端 List categoryTree = new ArrayList(); //所有类目 List allDTOList = new ArrayList(); if (CollectionUtils.isEmpty(list)) { return categoryTree; } for (CategoryDO categoryDO : list) { allDTOList.add(new CategoryTreeDTO().convertDOToDTO(categoryDO)); } //当前等级类目 categoryTree = allDTOList.stream().filter(dto -> level.equals(dto.getLevel())).collect(Collectors.toList()); for (CategoryTreeDTO categoryTreeDTO : categoryTree) { //组装类目为树结构 assembleTree(categoryTreeDTO, allDTOList,Constants.CATEGORY_MAX_LEVEL - level); } return categoryTree; } /** * 组装树 * * @param categoryTreeDTO * @param allList * @param remainRecursionCount 剩余递归次数 * @return */ public CategoryTreeDTO assembleTree(CategoryTreeDTO categoryTreeDTO, List allList, int remainRecursionCount) { remainRecursionCount--; //最大递归次数不超过Constants.CATEGORY_MAX_LEVEL-level次,防止坏数据死循环 if(remainRecursionCount < 0){ return categoryTreeDTO; } String categoryCode = categoryTreeDTO.getCategoryCode(); Integer level = categoryTreeDTO.getLevel(); //到达最后等级树返回 if (Constants.CATEGORY_MAX_LEVEL == level) { return categoryTreeDTO; } //子类目 List child = allList.stream().filter(a -> categoryCode.equals(a.getParentCode())).collect(Collectors.toList()); if (null == child) { return categoryTreeDTO; } categoryTreeDTO.setChildren(child); //组装子类目 for (CategoryTreeDTO dto : child) { assembleTree(dto, allList,remainRecursionCount); } return categoryTreeDTO; }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值