Java Stream flatMap解析

kyaa111 2年前 ⋅ 646 阅读

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中的代码

  1. 使用String数组构建stream, 此时流类型为 Stream<String>

  2. map处理流内元素, String#split 返回为String[], 此时流类型为 Stream<String[]>

  3. 再用flatMap处理, String[]就是上一层流的元素类型, 扁平化后流内元素变为String

flatMap与map最直观的区别就是 flatMap返回的是, 而map是流内元素

最终flatMap返回的多个流的内容被添加到一个流中