三级分-类查询-递归树形结构数据获取

admin码云

已于 2022-03-30 08:44:29 修改

阅读量590

点赞数 1

文章标签： java

于 2022-03-24 14:29:56 首次发布

本文链接：https://blog.csdn.net/weixin_55181056/article/details/123710079

版权

数据库结构

在这里插入图片描述

 /**
     * 查询类目
     *
     * @param dto
     * @return
     */
    @Override
    public ResultMessage queryCategory(PublishCategoryTypeDto dto) {

        if (dto == null || StringUtils.isEmpty(dto.getType())) {
            return ResultUtil.error(ResultCode.PARAMS_ERROR);
        }

        //查询所有分类
        List<PublishClassify> publishClassifies = publishClassifyMapper.selectList(new LambdaQueryWrapper<PublishClassify>().eq(PublishClassify::getOnlinePlatformId, dto.getType()));

        //赋值到集合中
        List<PublishClassifyChildrenVo> list=copyClassify(publishClassifies);

        //获取一级分类
        List<PublishClassifyChildrenVo> collect = list.stream().filter((publishClassifyChildrenVo) -> {
            return publishClassifyChildrenVo.getHierarchy() == 1;
        }).map((menu) -> {
            menu.setChildren(getChildren(menu, list));
            return menu;
        }).collect(Collectors.toList());

        return ResultUtil.data(collect);
    }
    /**
     * 递归查询
     * @param root
     * @param all
     */
    private List<PublishClassifyChildrenVo> getChildren(PublishClassifyChildrenVo root,List<PublishClassifyChildrenVo> all){

        List<PublishClassifyChildrenVo> child=all.stream().filter(classify ->{
            return StringUtils.isNotEmpty(classify.getParentClassifyId())&&StringUtils.isNotEmpty(root.getClassifyId())
                    &&classify.getParentClassifyId().equals(root.getClassifyId());
        }).map(childer->{
            childer.setChildren(getChildren(childer,all));
            return childer;
        }).collect(Collectors.toList());

        return child;
    }