P47 Stream流
系统:Win10
Java:1.8.0_333
IDEA:2020.3.4
说到 Stream 便容易想到 I/O Stream,而实际上,谁规定“流”就一定是“IO流”呢?在 Java 8中,得益于 Lambda 所带来的函数式编程,引入了一个全新的 Stream 概念,用于解决已有集合类库既有的弊端
1.引言
传统集合的多步遍历代码几乎所有的集合(如 Collection 接口或 Map 接口等)都支持直接或间接的遍历操作。而当我们需要对集合中的元素进行操作的时候,除了必需的添加、删除、获取外,最典型的就是集合遍历。例如
public class Demo01 {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("张三");
list.add("李四");
list.add("王二");
list.add("麻子");
for (String s : list) {
System.out.println(s);
}
}
}
这是一段非常简单的集合遍历操作:对集合中的每一个字符串都进行打印输出操作
循环遍历的弊端
Java 8的Lambda让我们可以更加专注于做什么(What),而不是怎么做(How),这点此前已经结合内部类进行了对比说明。现在,我们仔细体会一下上例代码,可以发现:
for 循环的语法就是“怎么做”
for 循环的循环体才是“做什么”
为什么使用循环?因为要进行遍历。但循环是遍历的唯一方式吗?遍历是指每一个元素逐一进行处理,而并不是从第一个到最后一个顺次处理的循环。前者是目的,后者是方式
试想一下,如果希望对集合中的元素进行筛选过滤
- 将集合 A 根据条件一过滤为子集 B
- 然后再根据条件二过滤为子集 C
那怎么办?在 Java 8 之前的做法可能为
public class Demo02 {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("张无忌");
list.add("周芷若");
list.add("赵敏");
list.add("张三");
list.add("张三丰");
List<String> list1 = new ArrayList<>();
for (String s : list) {
if (s.startsWith("张")){
list1.add(s);
}
}
List<String> list2 = new ArrayList<>();
for (String s : list1) {
if (s.length() == 3){
list2.add(s);
}
}
System.out.println(list2);
}
}
这段代码中含有三个循环,每一个作用不同
- 首先筛选所有姓张的人
- 然后筛选名字有三个字的人
- 最后进行对结果进行打印输出
每当我们需要对集合中的元素进行操作的时候,总是需要进行循环、循环、再循环。这是理所当然的么?不是。循环是做事情的方式,而不是目的。另一方面,使用线性循环就意味着只能遍历一次。如果希望再次遍历,只能再使用另一个循环从头开始
而 Lambda 的衍生物 Stream 能给我们带来怎样更加优雅的写法呢?
Stream 的更优写法
下面来看一下借助 Java 8 的 Stream API,什么才叫优雅
public class Demo03 {
public static void main(String[] args) {
List<String> list = new ArrayList<>();
list.add("张无忌");
list.add("周芷若");
list.add("赵敏");
list.add("张三");
list.add("张三丰");
List<String> list1 = new ArrayList<>();
list.stream().filter(s -> s.startsWith("张"))
.filter(s -> s.length() == 3)
.forEach(s -> System.out.println(s));
}
}
直接阅读代码的字面意思即可完美展示无关逻辑方式的语义:获取流、过滤姓张、过滤长度为3、逐一打印。代码中并没有体现使用线性循环或是其他任何算法进行遍历,我们真正要做的事情内容被更好地体现在代码中
2.流式思想概述
注意:请暂时忘记对传统 IO 流的固有印象!
整体来看,流式思想类似于工厂车间的“生产流水线”
当需要对多个元素进行操作(特别是多步操作)的时候,考虑到性能及便利性,我们应该首先拼好一个“模型”步骤方案,然后再按照方案去执行它
这张图中展示了过滤、映射、跳过、计数等多步操作,这是一种集合元素的处理方案,而方案就是一种“函数模 型”。图中的每一个方框都是一个“流”,调用指定的方法,可以从一个流模型转换为另一个流模型。而最右侧的数字3是最终结果
这里的 filter 、 map 、 skip 都是在对函数模型进行操作,集合元素并没有真正被处理。只有当终结方法 count 执行的时候,整个模型才会按照指定策略执行操作。而这得益于Lambda的延迟执行特性
备注:“Stream流”其实是一个集合元素的函数模型,它并不是集合,也不是数据结构,其本身并不存储任何 元素(或其地址值)
Stream(流)是一个来自数据源的元素队列
元素是特定类型的对象,形成一个队列。 Java 中的 Stream 并不会存储元素,而是按需计算
数据源:流的来源。可以是集合,数组等
和以前的 Collection 操作不同, Stream 操作还有两个基础的特征:
Pipelining: 中间操作都会返回流对象本身。这样多个操作可以串联成一个管道, 如同流式风格(fluent style)。这样做可以对操作进行优化,比如延迟执行(laziness)和短路( short-circuiting)
内部迭代:以前对集合遍历都是通过Iterator或者增强for的方式, 显式的在集合外部进行迭代,这叫做外部迭代。Stream 提供了内部迭代的方式,流可以直接调用遍历方法
当使用一个流的时候,通常包括三个基本步骤:获取一个数据源(source)→ 数据转换→执行操作获取想要的结果,每次转换原有 Stream 对象不改变,返回一个新的 Stream 对象(可以有多次转换),这就允许对其操作可以像链条一样排列,变成一个管道
3.获取流
java.util.stream.Stream 是 Java 8 新加入的最常用的流接口。(这并不是一个函数式接口)
获取一个流非常简单,有以下几种常用的方式
- 所有的 Collection 集合都可以通过 stream 默认方法获取流
- Stream 接口的静态方法 of 可以获取数组对应的流
- 根据 Collection 获取流
3.1 根据 Collection 获取流
首先, java.util.Collection 接口中加入了default方法 stream 用来获取流,所以其所有实现类均可获取流
public class Demo04 {
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();
}
}
3.2 根据 Map 获取流
java.util.Map 接口不是 Collection 的子接口,且其 K-V 数据结构不符合流元素的单一特征,所以获取对应的流需要分 key、value 或 entry 等情况
public class Demo05 {
public static void main(String[] args) {
Map<String,String> map = new HashMap<>();
Stream<String> stream1 = map.keySet().stream();
Stream<String> stream2 = map.values().stream();
Stream<Map.Entry<String, String>> stream3 = map.entrySet().stream();
}
}
3.3 根据数组获取流
如果使用的不是集合或映射而是数组,由于数组对象不可能添加默认方法,所以 Stream 接口中提供了静态方法 of,使用很简单:
public class Demo06 {
public static void main(String[] args) {
String[] arr = {"一", "二", "三"};
Stream<String> stream1 = Stream.of(arr);
Stream<Integer> stream2 = Stream.of(1, 2, 3, 4, 5);
}
}
备注: of 方法的参数其实是一个可变参数,所以支持数组
4.常用方法
流模型的操作很丰富,这里介绍一些常用的API。这些方法可以被分成两种:
延迟方法:返回值类型仍然是 Stream 接口自身类型的方法,因此支持链式调用。(除了终结方法外,其余方法均为延迟方法)
终结方法:返回值类型不再是 Stream 接口自身类型的方法,因此不再支持类似 StringBuilder 那样的链式调用。本小节中,终结方法包括 count 和 forEach 方法
备注:本小节之外的更多方法,请自行参考API文档。
4.1 逐一处理:forEach
虽然方法名字叫 forEach ,但是与 for 循环中的“for-each”昵称不同
void forEach(Consumer<? super T> action);
该方法接收一个 Consumer 接口函数,会将每一个流元素交给该函数进行处理
复习 Consumer 接口
java.util.function.Consumer接口是一个消费型接口
Consumer接口中包含抽象方法void accept(T t),意为消费一个指定泛型的数据。
基本使用
public class Demo07 {
public static void main(String[] args) {
Stream<Character> stream = Stream.of('A', 'B', 'C', 'D', 'E');
stream.forEach(s-> System.out.println(s));
}
}
4.2 过滤:filter
可以通过 filter 方法将一个流转换成另一个子集流。源码如下
Stream<T> filter(Predicate<? super T> predicate);
该接口接收一个 Predicate 函数式接口参数(可以是一个Lambda或方法引用)作为筛选条件
复习 Predicate 接口
此前我们已经学习过 java.util.function.Predicate 函数式接口,其中唯一的抽象方法为
boolean test(T t);
该方法将会产生一个 boolean 值结果,代表指定的条件是否满足。如果结果为 true,那么 Stream 流的 filter 方法将会调用元素;如果结果为 false,那么 filter 方法将会舍弃元素
基本使用
Stream 流中的 filter 方法基本使用的代码如
public class Demo08 {
public static void main(String[] args) {
Stream<String> stream = Stream.of("张三", "李四", "张伟");
Stream<String> stream1 = stream.filter(s -> s.startsWith("张"));
}
}
在这里通过 Lambda 表达式来指定了筛选的条件:必须姓张
4.3 映射:map
如果需要将流中的元素映射到另一个流中,可以使用 map 方法。源码如下
<R> Stream<R> map(Function<? super T, ? extends R> mapper);
该接口需要一个 Function 函数式接口参数,可以将当前流中的T类型数据转换为另一种R类型的流
复习Function接口
此前我们已经学习过 java.util.function.Function 函数式接口,其中唯一的抽象方法为
R apply(T t);
这可以将一种 T 类型转换成为 R 类型,而这种转换的动作,就称为“映射”
基本使用
Stream 流中的 map 方法基本使用的代码如
public class Demo09 {
public static void main(String[] args) {
Stream<String> stream = Stream.of("12", "34", "56", "78");
Stream<Integer> stream1 = stream.map(s -> Integer.parseInt(s));
}
}
这段代码中, map 方法的参数通过方法引用,将字符串类型转换成为了 int 类型(并自动装箱为 Integer 类对 象)
4.4 统计个数:count
正如集合 Collection 当中的 size 方法一样,流提供 count 方法来数一数其中的元素个数
long count();
该方法返回一个 long 值代表元素个数(不再像旧集合那样是 int 值)。基本使用
public class Demo10 {
public static void main(String[] args) {
Stream<String> stream = Stream.of("张三", "李四", "张伟");
long count = stream.filter(s -> s.startsWith("张")).count();
System.out.println(count);
}
}
4.5 取用前几个:limit
limit 方法可以对流进行截取,只取用前 n 个。源码如下
Stream<T> limit(long maxSize);
参数是一个 long 型,如果集合当前长度大于参数则进行截取;否则不进行操作。基本使用
public class Demo11 {
public static void main(String[] args) {
Stream<String> stream = Stream.of("张三", "李四", "张伟");
Stream<String> stream1 = stream.limit(2);
}
}
4.6 跳过前几个:skip
如果希望跳过前几个元素,可以使用 skip 方法获取一个截取之后的新流
Stream<T> skip(long n);
如果流的当前长度大于 n,则跳过前 n 个;否则将会得到一个长度为 0 的空流。基本使用
public class Demo12 {
public static void main(String[] args) {
Stream<String> stream = Stream.of("张三", "李四", "张伟");
Stream<String> stream1 = stream.skip(2);
}
}
4.7 组合:concat
如果有两个流,希望合并成为一个流,那么可以使用 Stream 接口的静态方法 concat
public static <T> Stream<T> concat(Stream<? extends T> a, Stream<? extends T> b) {
Objects.requireNonNull(a);
Objects.requireNonNull(b);
@SuppressWarnings("unchecked")
Spliterator<T> split = new Streams.ConcatSpliterator.OfRef<>(
(Spliterator<T>) a.spliterator(), (Spliterator<T>) b.spliterator());
Stream<T> stream = StreamSupport.stream(split, a.isParallel() || b.isParallel());
return stream.onClose(Streams.composedClose(a, b));
}
备注:这是一个静态方法,与 java.lang.String 当中的 concat 方法是不同的
该方法的基本使用代码如
public class Demo13 {
public static void main(String[] args) {
Stream<String> stream = Stream.of("张三", "李四", "张伟");
Stream<String> stream1 = Stream.of("小曾", "小吕");
Stream<String> stream2 = Stream.concat(stream, stream1);
}
}
5.练习:集合元素处理(Stream 方式)
题目
现在有两个 ArrayList 集合存储队伍当中的多个成员姓名,要求使用传统的 for 循环(或增强 for 循环)依次进行以下若干操作步骤
- 第一个队伍只要名字为3个字的成员姓名
- 第一个队伍筛选之后只要前3个人
- 第二个队伍只要姓张的成员姓名
- 第二个队伍筛选之后不要前2个人
- 将两个队伍合并为一个队伍
- 根据姓名创建 Person 对象
- 打印整个队伍的Person对象信息
两个队伍(集合)的代码如下
//第一支队伍
ArrayList<String> one = new ArrayList<>();
one.add("老子");
one.add("宋远桥");
one.add("庄子");
one.add("苏星河");
one.add("洪七公");
one.add("石破天");
one.add("黄老邪");
one.add("石中玉");
//第二支队伍
ArrayList<String> two = new ArrayList<>();
two.add("张三");
two.add("张无忌");
two.add("赵丽颖");
two.add("赵四");
two.add("张三丰");
two.add("张天爱");
two.add("张二狗");
而 Person 类的代码为
public class Person {
String name;
public Person() {
}
public Person(String name) {
this.name = name;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
@Override
public String toString() {
return "Person{" +
"name='" + name + '\'' +
'}';
}
}
解答
public class Test {
public static void main(String[] args) {
//第一支队伍
ArrayList<String> one = new ArrayList<>();
one.add("老子");
one.add("宋远桥");
one.add("庄子");
one.add("苏星河");
one.add("洪七公");
one.add("石破天");
one.add("黄老邪");
one.add("石中玉");
//第二支队伍
ArrayList<String> two = new ArrayList<>();
two.add("张三");
two.add("张无忌");
two.add("赵丽颖");
two.add("赵四");
two.add("张三丰");
two.add("张天爱");
two.add("张二狗");
/*
1. 第一个队伍只要名字为3个字的成员姓名
2. 第一个队伍筛选之后只要前3个人
*/
Stream<String> limit = one.stream().filter(s -> s.length() == 3)
.limit(3);
/*
3. 第二个队伍只要姓张的成员姓名
4. 第二个队伍筛选之后不要前2个人
*/
Stream<String> skip = two.stream().filter(s -> s.startsWith("张"))
.skip(2);
/*
5. 将两个队伍合并为一个队伍
6. 根据姓名创建 Person 对象
7. 打印整个队伍的Person对象信息
*/
Stream.concat(limit, skip).map(s -> new Person(s))
.forEach(p -> System.out.println(p));
}
}
运行结果