Lambda表达式递归构造树

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": []
    }
  ]
}