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放在地图中，键入Bar返回的每个Foo.bars()个实例。我从哪里开始？

Answer 1

根据建议here，您需要从每个Bar中提取Foo值并创建它们对。完成配对后，您可以将它们收集到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流收集以使用多个键进行映射

2 个答案: