Java stream collect to map with multiple keys
Java stream collect to map with multiple keys
我以为我在 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
中提取 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
类型)。您可以自己编写更具体的内容。
您可以为此定义一个新的收集器。一个简单的实现(总是创建 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 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
中提取 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
类型)。您可以自己编写更具体的内容。
您可以为此定义一个新的收集器。一个简单的实现(总是创建 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));