1 为什么引入流
先从一个例子开始,看看为什么在Java8中要引入流(Stream)?
比如实现这么一个需求:在学生集合中查找男生的数量。
传统的写法为:
public
long
getCountsOf
MaleStudent(List<Student> students) {
long
count =
0
;
for
(Student student : students) {
if
(student.isMale()) {
count++;
}
}
return
count;
}
|
看似没什么问题,因为我们写过太多类似的“样板”代码,尽管智能的IDE通过code template功能让这一枯燥过程变得简化,但终究不能改变冗余代码的本质。
再看看使用流的写法:
public
long
getCountsOfMaleStudent(List<Student> students) {
return
students.stream().filter(Student::isMale).count();
}
|
一行代码就把问题解决了!
虽然读者可能还不太熟悉流的语法特性,但这正是函数式编程思想的体现:
- 回归问题本质,按照心智模型思考问题。
- Stream提供的高阶函数以及链式调用,可以对数据进行更高层的抽象和控制。
- 很容易转成并行流,提供并行计算的能力。
- 延迟加载。
- 简化代码。
2 创建流
创建流的方式可以有很多种,其中最常见的方式是通过Collection的Stream()方法或者Arrays的Stream()方法来生成流。
-
比如:
List<Integer> numbers = Arrays.asList(
1
,
2
,
3
);
Stream<Integer> numberStream = numbers.stream();
String[] words =
new
String[]{
"one"
,
"two"
};
Stream<String> wordsStream = Arrays.stream(words);
|
当然Stream接口本身也提供了许多和流相关的操作。
Stream<Integer> numbers = Stream.of(
1
,
2
,
3
);
Stream<String> emptyStream = Stream.empty();
Stream<String> infiniteString = Stream.generate(() ->
"hi"
);
Stream<BigInteger> integers = Stream.iterate(BigInteger.ZERO, n -> n.add(BigInteger.ONE));
|
其中Stream.generate()和Stream.iterate()产生的都是无限流,如果要把他们截取为有限流,可以使用limit()方法,
比如:
Stream<Double> top10 = Stream.generate(Math::random).limit(
10
);
|
另外,可以通过skip()方法跳过元素,concat()方法连接两个流。
Stream<Integer> skipedStream = Stream.of(
1
,
2
,
3
,
4
).skip(
2
);
Stream<String> concatedStream = Stream.concat(Stream.of(
"hello"
), Stream.of(
",world"
));
|
3 常用的流操作
filter
filter()方法的作用就是根据输入的条件表达式过滤元素。
接口定义如下:
Stream<T> filter(Predicate<?
super
T> predicate);
|
从中可以看出,输入参数是一个Predicate,也即是一个条件表达式。
一个例子:
Stream.of(
"a"
,
"1b"
,
"c"
,
"0x"
).filter(value -> isDigit(value.charAt(
0
)));
|
过滤出第一个字符是数字的元素。
输出结果为:
1b, 0x
map
map()的主要作用是通过映射函数转换成新的数据。
接口定义如下:
<R> Stream<R> map(Function<?
super
T, ?
extends
R> mapper);
|
从中可以看出,输入参数是一个Function。
一个例子:
Stream.of(
"a"
,
"b"
,
"c"
).map(String::toUpperCase);
|
把字符串转换成大写。
输出结果:
A, B, C
有状态的转换
在前面介绍的函数中,无论是map还是filter,都不会改变流的状态,也即结果并不依赖之前的元素。
除此之外,Java8也提供了有状态的转换,常用的操作是distinct和sorted。
distinct
distinct()的主要作用是去除流中的重复元素。和Oracle的distinct一个作用。
举例如下:
Stream<String> distinctStream = Stream.of(
"one"
,
"one"
,
"two"
,
"three"
).distinct();
|
去除字符串中的重复元素,返回结果为:
one, two, three
sorted
sorted()的主要作用是对流按照指定的条件进行排序。
接口定义如下:
Stream<T> sorted(Comparator<?
super
T> comparator);
|
从中可以看出,入参是一个Comparator,也即是一个函数式接口。
一个例子:
Stream<String> sortedStream = Stream.of(
"one"
,
"two"
,
"three"
).sorted(Comparator.comparing(String::length).reversed());
|
对字符串按照长度进行降序排列。
注意,这里使用了Comparator.comparing方法来简化调用。
输出结果为:
[three, one, two]
4 分组操作(groupingBy)
groupingBy操作也是基于collect操作完成的,功能是根据条件进行分组操作,他和partitioningBy不同的一点是,它的输入是一个Function,这样返回结果的Map中的Key就不再是boolean型,而是符合条件的分组值,使用场景会更广泛。
接口定义如下:
public
static
<T, K> Collector<T, ?, Map<K, List<T>>> groupingBy(Function<?
super
T, ?
extends
K> classifier)
|
一个例子
public
Map<String, List<Student>> studentByName(Stream<Student> students) {
return
students.collect(Collectors.groupingBy(student -> student.getName()));
}
|
按照学生的姓名进行分组。