zl程序教程

您现在的位置是:首页 >  后端

当前栏目

java8 stream().map().collect()的Collectors.toList()、Collectors.toMap()、Collectors.groupingBy()的

Map stream java8 collectors collect
2023-09-11 14:16:28 时间

一、Collectors.toList()

现在有个集合:

List<User> users = getUserList();

现在需要将这些user的id提取出来。这个很简单,for循环嘛,谁不会啊(不会吧不会吧,不会有人还不会用for循环的吧)。

List<Long> idList = new ArrayList<Long>();

for(int i = 0; i < users.size(); i++){

  idList.add(users.get(i).getId());

}

然而Java8有个更简便的方法,一行代码搞定:

List<Long> idList = users.stream().map(User::getId).collect(Collectors.toList());

其中有个User::getId,这个其实就是调用User类的getId()方法。

再举个例子,将集合里的元素由小写变大写:

List<String> list = Arrays.asList("a", "b", "c", "d");

List<String> results = list.stream().map(String::toUpperCase).collect(Collectors.toList());

System.out.println(results); //{A, B, C, D}

二、Collectors.toMap()

Collectors.toMap(),一般用于将一个List转换为Map。常见用法:

list.stream().collect(Collectors.toMap(Function keyMapper, Function valueMapper))

在这里插入图片描述

可以接收2个、3个、4个参数,但是我一般只用2个的或者3个的就已经足够了。这里我也就只讲一个前两个用法,也就是2个参数的和3个参数的用法。

第一个参数,用于指定key的Function。
第二个参数,用于指定value的Function。
第三个参数,若在转换中,出现多个相同的key,如何进行合并的Function。

1. 两个参数的用法

现在有个User对象:

@Getter
@Setter
public class User{
	private Long id;
	private String name;
	private Integer age;

	public User(Long id, String name, Integer age) {
        this.id = id;
        this.name = name;
        this.age = age;
    }
}

然后进行以下操作:

List<User> userList = new ArrayList<>();
userList.add(new User(1, "张三", 18));
userList.add(new User(2, "李四", 19));
userList.add(new User(3, "王五", 18));
//将userList转化为key为id,value为User对象的map
Map<Long, User> map = userList.stream().collect(Collectors.toMap(User::getId, p -> p));

Map<Long, User> map = userList.stream().collect(Collectors.toMap(User::getId, p -> p));这一步就是将userList 转换为key为id,value为User对象的map。

User::getId ===》 User对象的getId方法
p -> p ===》就是进来的是什么,最终就是什么,这里就是进来的是User对象,出去的也就是User对象

而这时map里的(模拟)值是:

{
	1: User(1, "张三", 18)
	2: User(2, "李四", 19)
	3: User(3, "王五", 18)
}

还可以换一下:

Map<Long, String> map = userList.stream().collect(Collectors.toMap(User::getId, User::getName));

这个获取的就是key为id,value为name的map了。

2. 三个参数的用法

还是沿用上面那个例子,如果这个时候你想获取key是age,value是name的map呢?如果你还是沿用上面的方法,就会出问题了,因为有两个age是 18 的数据,也就是存在重复的key,会直接报错,想不报错的话,就可以利用第三个参数了。

Map<Integer, String> map = userList.stream().collect(Collectors.toMap(User::getAge, User::getName, (a, b) -> b));

(a, b) -> b的意思就是,如果存在重复的,永远取后面一个

这时,map里的值就是:

{
	18: "王五"
	19: "李四"
}

三、Collectors.groupingBy()

还是沿用上面那个例子。当你想获取key是age的map,又不想覆盖掉重复项数据,这个时候就可以用 Collectors.groupingBy 了。

Map<Integer, List<User>> map = userList.stream().collect(Collectors.groupingBy(User::getAge));

可以看到,这次的返回值变成了 Map<Integer, List> 了,也就是说,变成了key是age,value是User对象的集合了。这时,map里的值就变成了:

{
	18: [User(1, "张三", 18), User(3, "王五", 18)]
	19: [User(2, "李四", 19)]
}

以上就是这三种Collectors的用法了,如果你有什么疑点,或者是觉得博主有哪里写的不对的,欢迎在下方留言讨论哦~