Java中两种获取Stream流的方式

Posted 2021-03-10 李亦华的博客

获取流

java.util.stream.Stream 是Java 8新加入的最常用的流接口。(这并不是一个函数式接口。)

获取一个流非常简单，有以下几种常用的方式：

• 所有的 Collection 集合都可以通过 stream 默认方法获取流;
• Stream 接口的静态方法 of 可以获取数组对应的流。

根据Collection获取流

首先， java.util.Collection 接口中加入了default方法 stream 用来获取流，所以其所有实现类均可获取流。

import java.util.*;
import java.util.stream.Stream;

public class Demo03Stream {
    public static void main(String[] args) {
        
        List<String> list = new ArrayList<>();
        // ...
        Stream<String> stream1 = list.stream();
        
        Set<String> set = new HashSet<>();
        // ...
        Stream<String> stream2 = set.stream();
        
        Vector<String> vector = new Vector<>();
        // ...
        Stream<String> stream3 = vector.stream();
    }
}

Collection接口中加入的默认方法 —— stream方法，其源码如下：

default Stream<E> stream() {
    return StreamSupport.stream(spliterator(), false);
}

根据Map获取流

java.util.Map 接口不是 Collection 的子接口，且其K-V数据结构不符合流元素的单一特征，所以获取对应的流需要分key、value或entry等情况：

import java.util.HashMap;
import java.util.Map;
import java.util.Set;
import java.util.stream.Stream;

public class Demo04Stream {
    public static void main(String[] args) {
        Map<String, String> map = new HashMap<>();
        // ...
        Set<String> key = map.keySet();
        Set<String> value = (Set<String>) map.values();
        Set<Map.Entry<String, String>> entries = map.entrySet();
        
        Stream<String> keyStream = key.stream();
        Stream<String> valueStream = value.stream();
        Stream<Map.Entry<String, String>> entryStream = entries.stream();
    }
}

Map接口比较特殊，它保存的数据是 Key - Value的，所以要分为Key、Value以及Key-Value情况，Key-Value情况即entry情况。将每种分别保存在Set集合中，而Set接口继承于Collection接口，所以这里也就间接的使用了Collection接口的默认方法stream，来获取流。

根据数组获取流

如果使用的不是集合或映射而是数组，由于数组对象不可能添加默认方法，所以 Stream 接口中提供了静态方法 of ，使用很简单：

import java.util.stream.Stream;

public class Demo05GetStream {
    public static void main(String[] args) {
        String[] array = { "Java", "C", "Python", "Hadoop", "Spark" };
        Stream<String> stream = Stream.of(array);
    }
}
备注： of 方法的参数其实是一个可变参数，所以支持数组。