stream实现返回树状结构数据

最新推荐文章于 2024-08-28 17:14:38 发布

邱可爱

最新推荐文章于 2024-08-28 17:14:38 发布

阅读量311

点赞数

分类专栏： java # SpringBoot 文章标签： java 后端 spring boot

本文链接：https://blog.csdn.net/weixin_43626356/article/details/123035142

版权

java 同时被 2 个专栏收录

67 篇文章 0 订阅

订阅专栏

SpringBoot

21 篇文章 0 订阅

订阅专栏

//这里只是针对二级分类噢
 @Override
    public List<SubjectNodeTreeVo> getAllSubjects() {
        QueryWrapper<EduSubject> wrapper = new QueryWrapper<>();
        //查询指定的列
        wrapper.select("id","title","parent_id");
        List<EduSubject> eduSubjects = baseMapper.selectList(wrapper);
        List<SubjectNodeTreeVo> nodes = new ArrayList<>();
		//查出得数据存到Vo对象中 
        eduSubjects.forEach(eduSubject -> {
            SubjectNodeTreeVo node = new SubjectNodeTreeVo();
            BeanUtils.copyProperties(eduSubject,node);
            nodes.add(node);
        });
        //这里通过stream实现返回树状数据
        Map<String, List<SubjectNodeTreeVo>> children = nodes.stream().filter(node -> (!node.getParentId().equals("0")))
            .collect(Collectors.groupingBy(node -> node.getParentId()));
        nodes.forEach(node -> node.setChildren(children.get(node.getId())));
        return nodes.stream().filter(node -> node.getParentId().equals("0")).collect(Collectors.toList());
    }

package com.njxzc.eduService.entity;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import java.util.Date;

import com.baomidou.mybatisplus.annotation.TableField;
import com.baomidou.mybatisplus.annotation.TableId;
import java.io.Serializable;
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.EqualsAndHashCode;
import lombok.experimental.Accessors;

/**
 * projectName:  online-edu
 * packageName: com.njxzc.eduService.entity.vo
 * date: 2022-02-20 19:26
 * copyright(c) 2020 njxzc qyl
 * @author qyl
 */
@Data
@EqualsAndHashCode(callSuper = false)
@Accessors(chain = true)
@ApiModel(value="EduSubject对象", description="课程科目")
public class EduSubject implements Serializable {

    private static final long serialVersionUID = 1L;

    @ApiModelProperty(value = "课程类别ID")
    @TableId(value = "id", type = IdType.ID_WORKER_STR)
    private String id;

    @ApiModelProperty(value = "类别名称")
    private String title;

    @ApiModelProperty(value = "父ID")
    private String parentId;

    @ApiModelProperty(value = "排序字段")
    private Integer sort;

    @ApiModelProperty(value = "创建时间")
    @TableField(fill = FieldFill.INSERT)
    private Date gmtCreate;

    @ApiModelProperty(value = "更新时间")
    @TableField(fill = FieldFill.INSERT_UPDATE)
    private Date gmtModified;


}

package com.njxzc.eduService.entity.vo;

import lombok.Data;

import java.util.List;

/**
 * projectName:  online-edu
 * packageName: com.njxzc.eduService.entity.vo
 * date: 2022-02-20 19:16
 * copyright(c) 2020 njxzc qyl
 *
 * @author qyl
 */
@Data
public class SubjectNodeTreeVo {

    /*结点父类id*/
    private String parentId;

    /*结点名称*/
    private String title;

    /*当前节点id*/
    private String id;

    /*孩子结点list*/
    private List<SubjectNodeTreeVo> children;
//返回树状结构
//    private List<SubjectNodeTreeVo> buildTree(List<SubjectNodeTreeVo> nodes) {
//        Map<String, List<SubjectNodeTreeVo>> children = nodes.stream().filter(node -> (!node.getParentId().equals("0")))
//                .collect(Collectors.groupingBy(node -> node.getParentId()));
//        nodes.forEach(node -> node.setChildren(children.get(node.getNodeId())));
//        return nodes.stream().filter(node -> node.getParentId().equals("0")).collect(Collectors.toList());
//    }

}