Java 16 引入了一种新Stream.mapMulti方法，允许您用多个元素替换流中的元素。
下面的示例显示了如何使用mapMulti大写和小写版本替换流中的每个字符串：
Stream.of("Twix", "Snickers", "Mars")
.mapMulti((s, c) -> {
c.accept(s.toUpperCase());
c.accept(s.toLowerCase());
})
.forEach(System.out::println);
Output:
TWIX
twix
SNICKERS
snickers
MARS
mars
|
同样的事情也可以使用flaMap这样的方式来实现：
Stream.of("Twix", "Snickers", "Mars")
.flatMap(s -> Stream.of(s.toUpperCase(), s.toLowerCase()))
.forEach(System.out::println);
|
那么mapMulti和之间有什么区别flatMap？根据javadocs：
在以下情况下，此方法比 flatMap 更可取：
- 用少量（可能为零）元素替换每个流元素时。使用这种方法可以避免为每组结果元素创建一个新的 Stream 实例的开销，而创建一个Stream实例恰恰是flatMap 所要求的前提条件。
- 当使用命令式方法生成结果元素比以流的形式返回更容易时。
打开查看multiMap内部源码，我们可以看到它委托给flatMap，但是，它在创建流之前使用一个SpinedBuffer来保存元素，从而避免了为每组结果元素创建新流的开销。
default <R> Stream<R> mapMulti(BiConsumer<? super T, ? super Consumer<R>> mapper) {
Objects.requireNonNull(mapper);
return flatMap(e -> {
SpinedBuffer<R> buffer = new SpinedBuffer<>();
mapper.accept(e, buffer);
return StreamSupport.stream(buffer.spliterator(), false);
});
}
|