Java 16 新方法 Stream.mapMult

Java 16 引入了一種新Stream.mapMulti方法，允許你用多個(gè)元素替換流中的元素。本篇文章將和大家分享一下 Stream.mapMulti方法在代碼中的具體使用。

下面的示例顯示了如何使用mapMulti大寫和小寫版本替換流中的每個(gè)字符串：

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這樣的方式來實(shí)現(xiàn)：

Stream.of("Twix", "Snickers", "Mars")
  .flatMap(s -> Stream.of(s.toUpperCase(), s.toLowerCase()))
  .forEach(System.out::println);

那么?mapMulti?和之間有什么區(qū)別?flatMap?？根據(jù)javadocs：

在以下情況下，此方法比 ?flatMap ?更可?。?/p>

• 用少量（可能為零）元素替換每個(gè)流元素時(shí)。使用這種方法可以避免為每組結(jié)果元素創(chuàng)建一個(gè)新的 Stream 實(shí)例的開銷，正如 flatMap 所要求的。
• 當(dāng)使用命令式方法生成結(jié)果元素比以流的形式返回它們更容易時(shí)。

檢查?multiMap?的代碼，我們可以看到它委托給?flatMap?，但是，它?SpinedBuffer?在創(chuàng)建流之前使用? a?來保存元素，從而避免了為每組結(jié)果元素創(chuàng)建新流的開銷。

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