Java stream collect to map with multiple keys

Question

我以为我在 Java 8 个流中已经相当不错了，但是后来...

我有一个Foo接口：

public interface Foo {
  String getKey();
  Stream<Bar> bars();
}

我知道我可以使用每个键将 Stream<Foo> 收集到 Map<String, Foo>：

Map<String, Foo> foosByKey = fooStream.collect(
    Collectors.toMap(Foo::getKey, Function.identity()));

但是如果我想将它们收集到一个 Map<Bar, Foo> 中怎么办？换句话说，对于 Steam 中的每个 Foo，我想将 Foo 放入映射中，并键控到 Foo.bars() 返回的每个 Bar 实例。我从哪里开始？

Answer 1

按照建议，您需要从每个 Foo 中提取 Bar 值并创建它们的对。一旦你有了这些对，你就可以将它们收集到 Map 中。例如，

Map<Bar, Foo> map = fooStream.flatMap(foo -> foo.bars().map(bar -> new SimpleEntry<>(bar, foo)))
            .collect(Collectors.toMap(Entry::getKey, Entry::getValue));

我们在这里使用 SimpleEntry 是因为它可用（Java 没有更简单的 Pair 类型）。您可以自己编写更具体的内容。

Answer 2

您可以为此定义一个新的收集器。一个简单的实现（总是创建 ArrayList 的 HashMap；没有下游支持）可以是：

public static <T, K>
Collector<T, ?, Map<K, List<T>>> multiGroupingBy(
        Function<? super T, Collection<? extends K>> multiClassifier) {
    return Collector.of(
            HashMap::new,
            (map, entry) -> {
                multiClassifier.apply(entry)
                        .forEach(
                                key -> map
                                        .computeIfAbsent(key,
                                                __ -> new ArrayList<>())
                                        .add(entry));
            },
            (map1, map2) -> {
                map2.forEach(
                        (key, list) -> map1
                                .computeIfAbsent(key,
                                        __ -> new ArrayList<>())
                                .addAll(list));
                return map1;
            });
}

那么您可以拨打：

fooStream.collect(multiGroupingBy(Foo::bars));

Java stream collect to map with multiple keys

Java stream collect to map with multiple keys

java

key

java-stream

collect