从父子 ID 的原始数据创建 Json Hierarchy答案

【问题标题】：Create Json Hierarchy from from raw data of parent child id从父子 ID 的原始数据创建 Json Hierarchy
【发布时间】：2021-08-28 06:37:51
【问题描述】：

数据列如下所示：

CompanyId,Parent Id
1,1
2,1
3,1
4,2
5,2

每个公司都属于某个母公司。我需要将这些原始数据转换为以下 json 格式。

任何 java 代码实现都会有所帮助。

【问题讨论】：

数据表的第一行是混乱的。公司可以成为自己的母公司吗？
在实际场景中是的！

标签： java json tree hierarchy raw-data

【解决方案1】：

假设数据表如下所示：

Map<Integer, Integer> childToParent = new HashMap<>();
childToParent.put(1, 1);
childToParent.put(2, 1);
childToParent.put(3, 1);
childToParent.put(4, 2);
childToParent.put(5, 2);

我们可以创建一个父子映射：

Map<Integer, List<Integer>> parentToChildren = childToParent.entrySet().stream()
        .collect(Collectors.groupingBy(Map.Entry::getValue, Collectors.mapping(Map.Entry::getKey, Collectors.toList())));

为简单起见，我们同样创建一个Company 类：

@Data
@AllArgsConstructor
@JsonInclude(JsonInclude.Include.NON_NULL)
public class Company {
    private final int id;
    @JsonProperty(value = "child")
    private List<Company> children;
}

然后创建Company对象：

Map<Integer, Company> idToCompany = childToParent.keySet().stream()
        .map(integer -> new Company(integer,  null))
        .collect(Collectors.toMap(Company::getId, company -> company));

现在我们可以为每个父母设置孩子：

idToCompany.values().forEach(company -> {
    List<Integer> childrenIds = parentToChildren.get(company.getId());

    if(childrenIds != null){
        List<Company> children = childrenIds.stream()
                .filter(childId -> childId != company.getId()) // a company can't be a parent of itself
                .map(idToCompany::get)
                .collect(Collectors.toList());
        company.setChildren(children);
    }
});

现在我们可以像这样提取“头部”公司：

List<Company> companiesHeads = childToParent.entrySet().stream()
        .filter(childIdToParentId -> childIdToParentId.getKey().equals(childIdToParentId.getValue()))
        .map(Map.Entry::getKey)
        .map(idToCompany::get)
        .collect(Collectors.toList());

并以 JSON 格式打印到屏幕上：

ObjectMapper mapper = new ObjectMapper();
mapper.enable(SerializationFeature.INDENT_OUTPUT);
System.out.println(mapper.writeValueAsString(companiesHeads));

输出：

[ {
  "id" : 1,
  "child" : [ {
    "id" : 2,
    "child" : [ {
      "id" : 4
    }, {
      "id" : 5
    } ]
  }, {
    "id" : 3
  } ]
} ]

【讨论】：