为什么mapMulti与flatMap相比需要类型信息

Question

我想使用mapMulti而不是flatMap，并重构了以下代码：

// using flatMap (version 1) => returns Set<Item>
var items = users.stream()
                 .flatMap(u -> u.getItems().stream())
                .collect(Collectors.toSet());

这个（版本2）：

// using mapMulti (version 2) => returns Set<Item>
var items = users.stream()
                 .<Item>mapMulti((u, consumer) -> u.getItems().forEach(consumer))
                 .collect(Collectors.toSet());

两者都返回相同的元素。然而，我怀疑我是否真的应该用mapMulti的更详细的代码替换我所有的flatMap。为什么我需要在mapMuli（）之前添加类型信息。




Answer 1

请注意，当使用flatMap时，推断结果流类型所需的类型推断与使用mapMulti时非常不同。

使用flatMap时，生成流的类型与lambda主体的返回类型相同。这是一个特殊的东西，编译器被设计用来推断类型变量（即编译器“知道”它）。

然而，在mapMulti的情况下，您可能想要的结果流的类型只能从您对消费者lambda参数所做的事情中推断出来。假设，编译器的设计可以使，例如，如果你说消费者。accept（1），然后它会查看您传递给accept的内容，并查看您想要一个流

基本上，您要求编译器根据lambda中任意表达式的类型推断lambda的参数类型。编译器并不是专门为实现这一点而设计的。

除了添加

使lambda显式键入：

var items = users.stream()
             .mapMulti((User u, Consumer<Item> consumer) -> u.getItems().forEach(consumer))
             .collect(Collectors.toSet());

添加一个临时流变量：

// By looking at the type of itemStream, the compiler can figure out that mapMulti should return a Stream<Item>
Stream<Item> itemStream = users.stream()
             .mapMulti((u, consumer) -> u.getItems().forEach(consumer));
var items = itemStream.collect(Collectors.toSet());

我不知道这是否更“简化”，但我认为如果使用方法引用，会更整洁：

var items = users.stream()
             .map(User::getItems)
             .<Item>mapMulti(Iterable::forEach)
             .collect(Collectors.toSet());