Java Stream flatMap解析
2022-01-14 23:36:39 653
flatMap
对流扁平化处理
运行如下代码
String[] words = new String[]{"Hello", "World"};
List<String> a = Arrays.stream(words)
.map(word -> word.split(""))
.flatMap((item) -> {
System.out.println(item.getClass());
return Arrays.stream(item);
})
.collect(Collectors.toList());
输出
class [Ljava.lang.String;
class [Ljava.lang.String;
以下代码, 等效于上面的
String[] words = new String[]{"Hello", "World"};
List<String> a = Arrays.stream(words)
.map(word -> word.split(""))
.flatMap(new Function<String[], Stream<String>>() {
@Override
public Stream<String> apply(String[] strings) {
System.out.println(strings.getClass());
return Arrays.stream(strings);
}
})
.collect(Collectors.toList());
输出
class [Ljava.lang.String;
class [Ljava.lang.String;
Function
接口的文档
/**
* 表示接受一个参数并产生结果的函数
*
* 这是一个函数接口, 其函数方法是apply
*
* @param <T> 函数的输入类型
* @param <R> 函数结果的类型
*
* @since 1.8
*/
@FunctionalInterface
public interface Function<T, R> {
可以看到demo中flatMap
方法接受Function<String[], Stream<String>>
的实例对象
以此对应lambda
的简写, String[]
为参数, Stream<String>
为返回值
对应demo中的代码
- 使用String数组构建stream, 此时流类型为 Stream<String>
- 用map处理流内元素, String#split 返回为String[], 此时流类型为 Stream<String[]>
- 再用flatMap处理, String[]就是上一层流的元素类型, 扁平化后流内元素变为String
flatMap与map最直观的区别就是 flatMap返回的是流
, 而map是流内元素
最终flatMap返回的多个流的内容被添加到一个流中