Java 流收集以使用多个键进行映射

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实例的Foo.bars() 。 我从哪里开始？

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));