Java 16的Stream#mapMulti()

21-06-02 banq

如今,新的 Java 版本发布得如此之快,以至于小事情常常被忽视,事实证明,Java 16 为 Stream API 带来了一个新的有趣的补充——你可能还没有听说过的mapMulti()方法。

<R> Stream<R> mapMulti(BiConsumer<? super T, ? super Consumer<R>> mapper)

其核心mapMulti()是flatMap() 的替代方案。两者都用于处理具有一对多语义的 Stream 元素。

但是,mapMulti()使您有机会使用提供的 Consumer 实例手动使用元素:

var lists = of(of(1), of(2, 3), List.<Integer>of());

var result = lists.stream()
  .mapMulti((list, consumer) -> list.forEach(consumer))
  .collect(Collectors.toList());

同样用flatMap() 实现:

// import static java.util.List.of;
var lists = of(of(1), of(2, 3), List.<Integer>of());
var result = lists.stream()
  .flatMap(i -> i.stream())
  .collect(Collectors.toList());

对于某些边缘情况(元素数量较少),mapMulti() 可以更方便、更高效, 因此您应该将新方法视为补充补充而不是改变游戏规则。归根结底,它在内部将工作委托给 flatMap()。

性能提升是通过简化最终传递给 flatMap() 的 Stream 实例的构造来实现的。这是通过将结果元素收集到 SpinedBuffer 中,然后将其转换为 Stream 来实现的:

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

源码: GitHub.

猜你喜欢