Java后端将查询数据组装成树结构

柔情只为你懂 2024-05-02 20:22 9阅读 0赞

public List<DictVo> getDictVoByPid(Long pId) {
            List<SysDict> sysDicts = sysDictMapper.selectList(new LambdaQueryWrapper<SysDict>()
                    .eq(SysDict::getPid, pId).eq(SysDict::getIsDeleted, YesOrNoEnum.NO.getCode()));
            List<DictVo> dictVoList = new ArrayList<>();
            for(SysDict temp: sysDicts){
                DictVo dictVo  ;
                dictVo = new DictVo();
                BeanUtils.copyProperties(temp, dictVo);
                dictVoList.add(dictVo);
            }
            return dictVoList;
    
        }
    
        public List<DictVo> selectDictVoList() {
            List<SysDict> sysDicts = sysDictMapper.selectList(new LambdaQueryWrapper<SysDict>()
                    .eq(SysDict::getIsDeleted, YesOrNoEnum.NO.getCode()));
            List<DictVo> dictVoList = new ArrayList<>();
            for(SysDict temp: sysDicts){
                DictVo dictVo = new DictVo();
                BeanUtils.copyProperties(temp, dictVo);
                dictVoList.add(dictVo);
            }
            return dictVoList;
        }
    
    
        /**
         * 获取物料分类树结构
         * @return
         */
        @Override
        public List<DictVo> getDictVoTree() {
            // 查询所有节点
            List<DictVo> voList = selectDictVoList();
            // 查询最高父节点
            List<DictVo> rootNode = getDictVoByPid(0L);
            // 设置子节点
            rootNode.forEach(info -> {
                List<DictVo> childrenList = this.getChildren(voList,info.getId());
                info.setChildrenDictVoList(childrenList);
            });
            return rootNode;
        }
    
        /**
         * 获取根据父节点ID获取子节点
         * @param allNode
         * @param parentId
         * @return
         */
        public List<DictVo> getChildren(List<DictVo> allNode, Long parentId){
            // 使用Linq匹配出传入父节点的下一级子节点
            List<DictVo> infoList =   allNode.stream().filter(t -> t.getPid().equals(parentId)).collect(Collectors.toList());
                    //Linq.of(allNode).where(item->item.getMaterialtypeParent().equals(parentId)).toList();
            /**
             * 2023.02.03 更新
             * 这里可以用流处理，省去引入依赖
             */
            // List<BasMaterialType> basDischargeInfoList = allNode.stream().filter(item -> item.getMaterialtypeParent().equals(parentId)).collect(Collectors.toList());
            if (infoList.size()==0){
                return new ArrayList<>();
            }
            //递归下一级
            infoList.forEach(basDischargeInfo -> {
                basDischargeInfo.setChildrenDictVoList(getChildren(allNode,basDischargeInfo.getId()));
            });
            return infoList;
        }

[Java后端将查询数据组装成树结构\_查询数据并生成树-CSDN博客][Java_-CSDN]

[Java_-CSDN]: https://blog.csdn.net/weixin_46099269/article/details/116230115