java - 按键将对象列表分组到具有唯一对象子列表的列表中(使用java流)

Question

我有一个列表，其中包含键和一些附加对象的组合，这些对象之间不以其他方式相关。

考虑这个结构:

record A0(String id, String name, B b, C c) {}

record A(String id, String name, Set<B> bs, Set<C> cs) {}

record B(String id, String name) {}

record C(String id, String name) {}

a0s.add(new A0("1", "n1", new B("1", "nb1"), new C("1", "nc1")));
a0s.add(new A0("1", "n1", new B("1", "nb1"), new C("2", "nc2")));
a0s.add(new A0("1", "n1", new B("2", "nb2"), new C("3", "nc3")));
a0s.add(new A0("2", "n2", new B("2", "nb2"), new C("4", "nc4")));
a0s.add(new A0("2", "n2", new B("1", "nb1"), new C("5", "nc5")));
a0s.add(new A0("2", "n2", new B("2", "nb2"), new C("6", "nc6")));
a0s.add(new A0("3", "n3", new B("3", "nb3"), new C("7", "nc7")));
a0s.add(new A0("3", "n3", new B("3", "nb3"), new C("8", "nc8")));
a0s.add(new A0("4", "n4", new B("4", "nb4"), new C("9", "nc9")));
a0s.add(new A0("4", "n4", new B("5", "nb5"), new C("10", "nc10")));

我想用 java-streams 来实现这一点:

[ {
  "id" : "1",
  "name" : "n1",
  "bs" : [ {
    "id" : "1",
    "name" : "nb1"
  }, {
    "id" : "2",
    "name" : "nb2"
  } ],
  "cs" : [ {
    "id" : "1",
    "name" : "nc1"
  }, {
    "id" : "2",
    "name" : "nc2"
  }, {
    "id" : "3",
    "name" : "nc3"
  } ]
}, {
  "id" : "2",
  "name" : "n2",
  "bs" : [ {
    "id" : "2",
    "name" : "nb2"
  }, {
    "id" : "1",
    "name" : "nb1"
  } ],
  "cs" : [ {
    "id" : "4",
    "name" : "nc4"
  }, {
    "id" : "5",
    "name" : "nc5"
  }, {
    "id" : "6",
    "name" : "nc6"
  } ]
}, {
  "id" : "3",
  "name" : "n3",
  "bs" : [ {
    "id" : "3",
    "name" : "nb3"
  } ],
  "cs" : [ {
    "id" : "7",
    "name" : "nc7"
  }, {
    "id" : "8",
    "name" : "nc8"
  } ]
}, {
  "id" : "4",
  "name" : "n4",
  "bs" : [ {
    "id" : "4",
    "name" : "nb4"
  }, {
    "id" : "5",
    "name" : "nb5"
  } ],
  "cs" : [ {
    "id" : "10",
    "name" : "nc10"
  }, {
    "id" : "9",
    "name" : "nc9"
  } ]
} ]

这是我的代码，没有(显然)java-streams:

import java.util.*;
import java.util.stream.Collectors;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;

class Scratch {

  record A0(String id, String name, B b, C c) {}

  record A(String id, String name, Set<B> bs, Set<C> cs) {}

  record B(String id, String name) {}

  record C(String id, String name) {}

  public static void main(String[] args) throws JsonProcessingException {
    List<A0> a0s = new ArrayList<>();
    a0s.add(new A0("1", "n1", new B("1", "nb1"), new C("1", "nc1")));
    a0s.add(new A0("1", "n1", new B("1", "nb1"), new C("2", "nc2")));
    a0s.add(new A0("1", "n1", new B("2", "nb2"), new C("3", "nc3")));
    a0s.add(new A0("2", "n2", new B("2", "nb2"), new C("4", "nc4")));
    a0s.add(new A0("2", "n2", new B("1", "nb1"), new C("5", "nc5")));
    a0s.add(new A0("2", "n2", new B("2", "nb2"), new C("6", "nc6")));
    a0s.add(new A0("3", "n3", new B("3", "nb3"), new C("7", "nc7")));
    a0s.add(new A0("3", "n3", new B("3", "nb3"), new C("8", "nc8")));
    a0s.add(new A0("4", "n4", new B("4", "nb4"), new C("9", "nc9")));
    a0s.add(new A0("4", "n4", new B("5", "nb5"), new C("10", "nc10")));

    Set<A> collectA = new HashSet<>();
    Map<String, Set<B>> mapAB = new HashMap<>();
    Map<String, Set<C>> mapAC = new HashMap<>();

    a0s.forEach(
        a0 -> {
          mapAB.computeIfAbsent(a0.id, k -> new HashSet<>());
          mapAC.computeIfAbsent(a0.id, k -> new HashSet<>());
          mapAB.get(a0.id).add(a0.b);
          mapAC.get(a0.id).add(a0.c);
          collectA.add(new A(a0.id, a0.name, new HashSet<>(), new HashSet<>()));
        });

    Set<A> outA = new HashSet<>();

    collectA.forEach(
        a -> {
          outA.add(new A(a.id, a.name, mapAB.get(a.id), mapAC.get(a.id)));
        });

    ObjectMapper objectMapper = new ObjectMapper();
    objectMapper.enable(SerializationFeature.INDENT_OUTPUT);
    String json =
        objectMapper.writeValueAsString(
            outA.stream()
                .sorted(Comparator.comparing(A::id))
                .collect(Collectors.toList()));

    System.out.println(json);
  }
}

我有红色帖子和文档，但无法实现。 This为我指明了某个方向，但我无法继续与其他解决方案结合并阅读 API 文档。 对我来说，什么“错误”是我有多个重复的对象来分组(收集)并保持唯一。我使用 Set 来利用其唯一性，但也可以使用 List。

Answer 1

groupingBy + teeing

实现这一目标的方法之一是围绕标准收集器构建解决方案。

为了让人信服，我们可以引入一些自定义类型。

旨在保存独特属性的记录 id和name :

record IdName(String id, String name) {}

还有另一个存储集的记录 Set<B> , Set<C>与相同的id相关联:

record BCSets(Set<B> bs, Set<C> cs) {}

流的逻辑:

• 使用 IdName 对数据进行分组使用 Collector groupingBy() 作为 key
• 利用收集器 teeing()作为分组的下游。 teeing()需要三个参数:两个收集器和一个组合它们产生的结果的函数。作为 teeing() 的下游收集者我们可以利用 mapping() 的组合和toSet() ，并通过生成辅助记录来组合它们的结果 BCSets .
• 然后在映射条目上创建一个流，并将每个条目转换为 A 类型的实例。 .
• 对流元素进行排序并将它们收集到列表中。

List<A> listA = a0s.stream()
    .collect(Collectors.groupingBy(
        a0 -> new IdName(a0.id(), a0.name()),
        Collectors.teeing(
            Collectors.mapping(A0::b, Collectors.toSet()),
            Collectors.mapping(A0::c, Collectors.toSet()),
            BCSets::new
        )
    ))
    .entrySet().stream()
    .map(e -> new A(e.getKey().id(), e.getKey().name(), e.getValue().bs(), e.getValue().cs()))
    .sorted(Comparator.comparing(A::id))
    .toList();

groupingBy + 自定义收集器

另一个选择是创建一个自定义收集器，它将用作grouping()的下游。

为此，我们需要定义一个自定义累积类型来使用流中的元素并收集 B 的实例。和C成套。为了方便起见，我实现了 Consumer接口(interface):

public static class ABCAccumulator implements Consumer<A0> {
    private Set<B> bs = new HashSet<>();
    private Set<C> cs = new HashSet<>();

    @Override
    public void accept(A0 a0) {
        bs.add(a0.b());
        cs.add(a0.c());
    }
    
    public ABCAccumulator merge(ABCAccumulator other) {
        bs.addAll(other.bs);
        cs.addAll(other.cs);
        return this;
    }
    
    // getters
}

要创建自定义收集器，我们可以使用静态工厂方法 Collector.of() .

总体逻辑保持不变，但有一点不同 - 现在我们只有两个收集器，辅助 map 的值类型也不同(将是 ABCAccumulator )。

List<A> listA = a0s.stream()
    .collect(Collectors.groupingBy(
        a0 -> new IdName(a0.id(), a0.name()),
        Collector.of(
            ABCAccumulator::new,
            ABCAccumulator::accept,
            ABCAccumulator::merge
        )
    ))
    .entrySet().stream()
    .map(e -> new A(e.getKey().id(), e.getKey().name(), e.getValue().getBs(), e.getValue().getCs()))
    .sorted(Comparator.comparing(A::id))
    .toList();