Java8使用Stream优雅地处理集合

Happyjava

发布于 2019-07-17 16:32:01

4500

发布于 2019-07-17 16:32:01

文章被收录于专栏：Happy的分享

说明

集合和数组是我们经常会用到的数据结构，在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()方法

 @Test
 public void listInit(){
 List<String> list = new ArrayList<>();
 list.add("apple");
 list.add("banana");
 Stream<String> stream = list.stream();
 }
复制代码

方法二：Stream.of()方法

 @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字符串过滤掉

/**
 * 过滤
 */
 @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));
 }
复制代码

输出结果：

["apple","banana","cat"]
复制代码

map 生成新的集合

这里有两个类

 @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

 @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));
 }
复制代码

输出结果

[{"id":1,"name":"happyjava1"},{"id":2,"name":"happyjava2"}]
复制代码

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

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

 @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:{"id":1,"name":"happyjava1"},2:{"id":2,"name":"happyjava2"}}
{1:"happyjava1",2:"happyjava2"}
复制代码

sort排序

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

@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","3","4","5","9"]
[{"id":1,"name":"happyjava1"},{"id":2,"name":"happyjava2"}]
复制代码

skip和limit

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

 @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));
 }
复制代码

输出结果