Happyjava's blog site

Happyjava's blog site,分享编程知识,顺便发发牢骚

0%

Java8使用Stream优雅地处理集合

说明

集合和数组是我们经常会用到的数据结构,在jdk1.8之前,集合和数组的处理并不是很便捷。但是到了JDK1.8之后,使用Stream处理集合会使代码变得更加的简洁明了。作为一名开发者,其实很有必要学习新特性的用法,即使是说项目中没有采用。虽然JDK1.8已经发布数年,但是还是发现很多人都不会使用JDK1.8给我们带来的新的API。比如之前写过的一篇文章还在用SimpleDateFormat?Java8都发布N年了,转LocalDateTime吧,讲解了JDK1.8时间上的新API,大家也可以看看。


Stream类的常用api

这里列举了比较常用的api,想要查看更加具体API,大家可以参考下jdk1.8官方文档,那里有更加详细的描述。

初始化

初始化一个Stream常用到两种方法:

方法一:List的stream()方法

1
2
3
4
5
6
7
@Test
public void listInit(){
List<String> list = new ArrayList<>();
list.add("apple");
list.add("banana");
Stream<String> stream = list.stream();
}

方法二:Stream.of()方法

1
2
3
4
5
6
7
8
9
@Test
public void streamOfInit(){
// 第一种
Stream<String> stream = Stream.of("a", "b", "c", "d");

// 第二种
String [] strings = {"a","b","c"};
Stream<String> stream1 = Stream.of(strings);
}

filter过滤

挑选需要的元素,比如:下方例子把dog字符串过滤掉

1
2
3
4
5
6
7
8
9
/**
* 过滤
*/
@Test
public void filter() {
String[] strings = {"apple", "banana", "cat", "dog"};
List<String> list = Stream.of(strings).filter(e -> !StringUtils.equals(e, "dog")).collect(Collectors.toList());
System.out.println(JSON.toJSONString(list));
}

输出结果:

1
["apple","banana","cat"]

map 生成新的集合

这里有两个类

1
2
3
4
5
6
7
8
9
10
11
12
 @Data
@AllArgsConstructor
public static class Person {
private Integer id;
private String name;
}

@Data
public static class Man {
private Integer id;
private String name;
}

把一个List转化成List

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
@Test
public void map2() {
List<Person> list = new ArrayList<>();
list.add(new Person(1, "happyjava1"));
list.add(new Person(2, "happyjava2"));

List<Man> manList = list.stream().map(e -> {
Man man = new Man();
man.setId(e.getId());
man.setName(e.getName());
return man;
}).collect(Collectors.toList());

System.out.println(JSON.toJSONString(manList));
}

输出结果

1
[{"id":1,"name":"happyjava1"},{"id":2,"name":"happyjava2"}]

Collectors.toMap 把list拼接成需要的map

这个在向数据库查询list数据的时候经常用到

1
2
3
4
5
6
7
8
9
10
11
12
@Test
public void testCollectToMap() {
List<Person> list = new ArrayList<>();
list.add(new Person(1, "happyjava1"));
list.add(new Person(2, "happyjava2"));
Map<Integer, Person> personMap = list.stream().collect(Collectors.toMap(Person::getId, e -> e));

System.out.println(JSON.toJSONString(personMap));
Map<Integer, String> nameMap = list.stream().collect(Collectors.toMap(Person::getId, Person::getName));

System.out.println(JSON.toJSONString(nameMap));
}

输出结果

1
2
{1:{"id":1,"name":"happyjava1"},2:{"id":2,"name":"happyjava2"}}
{1:"happyjava1",2:"happyjava2"}

sort排序

无参的sort方法支持基本数据类型,包装类型,String等类型的自然顺序排序

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
@Test
public void sort() {
// 自然顺序排序 基本数据类型 字符串
List<String> collect = Stream.of("1", "5", "2", "9", "3", "4").sorted().collect(Collectors.toList());
System.out.println(JSON.toJSONString(collect));
List<Person> list = new ArrayList<>();
list.add(new Person(1, "happyjava1"));
list.add(new Person(2, "happyjava2"));

// 自然顺序排序
List<Person> personList = list.stream().sorted((o1, o2) -> {
if(o1.getId() > o2.getId()){
return 1;
} else if(o1.getId().equals(o2.getId())){
return 0;
} else {
return -1;
}
}).collect(Collectors.toList());
System.out.println(JSON.toJSONString(personList));
}

输出结果

1
2
["1","2","3","4","5","9"]
[{"id":1,"name":"happyjava1"},{"id":2,"name":"happyjava2"}]

skip和limit

这个相当于MySQL的limit n,m,与MongoDB的skip、limit用法是一致的。意思就是跨过几行记录,取几行记录的意思。

1
2
3
4
5
@Test
public void skipAndLimit(){
List<String> list = Stream.of("1", "5", "2", "9", "3", "4").skip(2).limit(3).collect(Collectors.toList());
System.out.println(JSON.toJSONString(list));
}

输出结果

1
["2","9","3"]

总结

常用的Stream操作都在这里列举了,当然Stream还有很多其他的用法,这里也没法一一列举,需要读者自己去学习。