Lambda表达式递归构造树

园园的铁粉QAQ

已于 2023-04-25 14:16:42 修改

阅读量364

点赞数

文章标签： java

于 2023-04-24 14:26:32 首次发布

本文链接：https://blog.csdn.net/addice1019/article/details/130341796

版权

文章展示了如何在Java中利用Lambda表达式和流API来递归地构造树形数据结构，具体是一个名为SpcMemberRole的实体类，包含了节点信息如ID、角色、父ID和排序序号。通过查询数据库获取数据，然后过滤、排序并设置子节点，最终形成树状结构。返回的数据结构显示了层级关系，包括父节点和子节点的信息。

摘要由CSDN通过智能技术生成

Lambda表达式递归构造树

声明一个实体类,用于记录每个节点的信息

@Entity
@Table(name = "spc_member_role")
@Data
public class SpcMemberRole {
    /**
     * 主键
     */
    @Id
    @GeneratedValue(strategy = GenerationType.IDENTITY)
    @Column(name = "id", nullable = false)
    private Long id;

    /**
     * 名称
     */
    @Column(name = "role")
    private String role;

    /**
     * 父类id
     */
    @Column(name = "parete_id")
    private Long pareteId;

    /**
     * 站点ID
     */
    @Column(name = "site_id")
    private Long siteId;
    
    /**
     * 节点排序序号
     */
    @Column(name = "sort")
    private Integer sort;

    @Transient
    private List<SpcMemberRole> children;
}

递归构造树结构

public List<SpcMemberRole> listWithTree(Long siteId) {
        // 查询出当前节点和关联当前节点parentId下所有分类
        List<SpcMemberRole> memberRoleList = Optional.ofNullable(this.findBySiteId(siteId)).orElse(Lists.newArrayList());

        return memberRoleList.stream()
                // 过滤出父级
                .filter(spcMemberRole -> spcMemberRole.getPareteId() == 0L)
                // 给父级设置子级
                .peek(spcMemberRole -> spcMemberRole.setChildren(this.listChildren(spcMemberRole, memberRoleList)))
                .sorted(Comparator.comparing(SpcMemberRole::getSort))
                .collect(Collectors.toList());
    }

private List<SpcMemberRole> listChildren(SpcMemberRole spcMemberRole, List<SpcMemberRole> list) {
        // 递归查找子分类
        return list.stream()
                .filter(item -> item.getPareteId().equals(spcMemberRole.getId()))
                .peek(item -> item.setChildren(this.listChildren(item, list)))
                .sorted(Comparator.comparing(SpcMemberRole::getSort))
                .collect(Collectors.toList());
    }

查询结果

页面效果

页面效果

返回数据结构

{
  "status": 0,
  "message": "操作成功",
  "data": [
    {
      "id": 52,
      "role": "全部",
      "pareteId": 0,
      "siteId": 55,
      "sort": 1,
      "children": [
        {
          "id": 62,
          "role": "分组测试",
          "pareteId": 52,
          "siteId": 55,
          "sort": -4,
          "children": [
            {
              "id": 186,
              "role": "孙分组",
              "pareteId": 62,
              "siteId": 55,
              "sort": 1,
              "children": []
            }
          ]
        },
        {
          "id": 63,
          "role": "test",
          "pareteId": 52,
          "siteId": 55,
          "sort": 0,
          "children": []
        },
        {
          "id": 183,
          "role": "test1",
          "pareteId": 52,
          "siteId": 55,
          "sort": 1,
          "children": []
        }
      ]
    },
    {
      "id": 53,
      "role": "黑名单",
      "status": 1,
      "pareteId": 0,
      "siteId": 55,
      "sort": 2,
      "children": []
    }
  ]
}