一、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)]
}