为什么Stream :: flatMap的使用错误？

Question

我希望能够像这样使用Stream :: flatMap

public static List<String> duplicate(String s) {

    List<String> l = new ArrayList<String>();
    l.add(s);
    l.add(s);

    return l;
}


listOfStrings.stream().flatMap(str -> duplicate(str)).collect(Collectors.toList());

但是我得到以下编译器错误

Test.java:25:错误:不兼容的类型:无法推断类型变量R listOfStrings.stream().flatMap(str - > duplicate(str)).collect(Collectors.toList());

(参数不匹配; lambda表达式中的错误返回类型List无法转换为Stream)
其中R,T是类型变量:R extends方法flatMap中声明的Object(Function>)T extends Interface在Interface Stream中声明的Object

在scala我可以做我认为相同的事情

scala> List(1,2,3).flatMap(duplicate(_))
res0: List[Int] = List(1, 1, 2, 2, 3, 3)

为什么这不是java中flatMap的有效用法？

Answer 1

lambda表达式flatMap需要返回a Stream,正如其参数flatMap类型所示Function<? super T, ? extends Stream<? extends R>>.

以下代码将编译并运行正常:

listOfStrings.stream()
             .flatMap(str -> duplicate(str).stream()) // note the .stream() here
             .collect(Collectors.toList());

因为lambda表达式str -> duplicate(str).stream()是类型的Function<String, Stream<String>>.

Answer 2

如果要多次复制流中的每个对象,则不需要在此处浪费内存ArrayList.有几种更短更快的替代品.

• 使用生成新流Stream.generate,然后限制它:

  listOfStrings.stream()
               .flatMap(str -> Stream.generate(() -> str).limit(2))
               .collect(Collectors.toList());
  

• 通过生成数字序列IntStream.range并将它们映射到相同的字符串:

  listOfStrings.stream()
               .flatMap(str -> IntStream.range(0, 2).mapToObj(i -> str))
               .collect(Collectors.toList());
  

• 使用好旧Collections.nCopies:

  listOfStrings.stream()
               .flatMap(str -> Collections.nCopies(2, str).stream())
               .collect(Collectors.toList());
  

如果你确定你总是会重复两次,那么最短的选择是:

listOfStrings.stream()
             .flatMap(str -> Stream.of(str, str))
             .collect(Collectors.toList());