Java8使用Stream优雅地处理集合

2024-02-02 15:16:23 浏览数 (2)

说明

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

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

方法二:Stream.of()方法

代码语言:javascript复制
 @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字符串过滤掉

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

输出结果:

代码语言:javascript复制
["apple","banana","cat"]

map 生成新的集合

这里有两个类

代码语言:javascript复制
 @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

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

输出结果

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

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

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

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

输出结果

代码语言:javascript复制
{1:{"id":1,"name":"happyjava1"},2:{"id":2,"name":"happyjava2"}}
{1:"happyjava1",2:"happyjava2"}

sort排序

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

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

输出结果

代码语言:javascript复制
["1","2","3","4","5","9"]
[{"id":1,"name":"happyjava1"},{"id":2,"name":"happyjava2"}]

skip和limit

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

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

输出结果

代码语言:javascript复制
["2","9","3"]

总结

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

0 人点赞