Java 16 新方法 Stream.mapMult

2021-09-07 09:48:33 浏览数 (3008)

Java 16 引入了一种新Stream.mapMulti方法,允许你用多个元素替换流中的元素。本篇文章将和大家分享一下 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 实例的开销,正如 flatMap 所要求的。
  • 当使用命令式方法生成结果元素比以流的形式返回它们更容易时。

检查​multiMap​的代码,我们可以看到它委托给​flatMap​,但是,它​SpinedBuffer​在创建流之前使用​ a​来保存元素,从而避免了为每组结果元素创建新流的开销。

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