java stream list转map

java 8 使用stream流来对集合进行处理:

Stream将List转换为Map,使用Collectors.toMap方法进行转换

背景:User类,类中分别有id,name,age三个属性。List集合,userList,存储User对象

1、指定key-value,value是对象中的某个属性值

 Map userMap1 = userList.stream().collect(Collectors.toMap(User::getId,User::getName));

2、指定key-value,value是对象本身,User->User 是一个返回本身的lambda表达式

Map userMap2 = userList.stream().collect(Collectors.toMap(User::getId,User->User));

3、指定key-value,value是对象本身,Function.identity()是简洁写法,也是返回对象本身

 Map userMap3 = userList.stream().collect(Collectors.toMap(User::getId, Function.identity()));

4、指定key-value,value是对象本身,Function.identity()是简洁写法,也是返回对象本身,key 冲突的解决办法,这里选择第二个key覆盖第一个key

 Map userMap4 = userList.stream().collect(Collectors.toMap(User::getId, Function.identity(),(key1,key2)->key2));

注意:用Collectors的toMap方法转换List,一般会遇到两个问题。一个是转换map,key重复问题;另一个是空指针异常,即转为map的value是null。
问题解决!!!
一、第一种问题报的错误如下:Duplicate key
在这里插入图片描述

使用(key1,key2)->key2表达式可以解决此类问题,如果出现重复的key就使用key2覆盖前面的key1,也可以定义成(key1,key2)->key1,保留key1,根据自己的业务场景来调整。

 

 

 

当使用二个list集合,根据某个值来给你list赋值时,我们之前都是采用双层for循环来实现,现在我们可以使用stream()+lamda表达式来提升效率

实例如下:

首先创建两个对象男生和女生:

// 男生 get set和toString这里省略
public class SchoolBoy {
    private String girlId;
    
    private String id;
    
    private String name;
 
    private String grilName;
}
// 女生 get set和toString这里省略
public class Girl {
    private String id;
 
    private String name;
}
根据男生对象中的girlId去匹配女生对象并把男生对象的grilName字段赋值

下面展示一个demo

public static void main(String[] args) {
    List schoolBoys = new ArrayList<>(3);
    SchoolBoy boy1 = new SchoolBoy();
    boy1.setGirlId("1");
    boy1.setId("10");
    boy1.setName("小明");
 
    SchoolBoy boy2 = new SchoolBoy();
    boy2.setGirlId("2");
    boy2.setId("11");
    boy2.setName("小豪");
 
    SchoolBoy boy3 = new SchoolBoy();
    boy3.setGirlId("3");
    boy3.setId("12");
    boy3.setName("小白");
    schoolBoys.add(boy1);
    schoolBoys.add(boy2);
    schoolBoys.add(boy3);
 
    List girls = new ArrayList<>(3);
    Girl girl1 = new Girl();
    girl1.setId("1");
    girl1.setName("小英");
 
    Girl girl2 = new Girl();
    girl2.setId("2");
    girl2.setName("小美");
 
    Girl girl3 = new Girl();
    girl3.setId("3");
    girl3.setName("小花");
    girls.add(girl1);
    girls.add(girl2);
    girls.add(girl3);
    
    List list = schoolBoys.stream()
        .map(schoolBoy -> girls.stream()
                .filter(girl -> schoolBoy.getGirlId().equals(girl.getId()))
                .findFirst()
                .map(girl -> {
                    schoolBoy.setGrilName(girl.getName());
                    return schoolBoy;
                }).
orElse(null))
        .collect(Collectors.toList());
    System.out.println(list.toString());
}

最后执行的结果:

[SchoolBoy [girlId=1, id=10, name=小明, grilName=小英],

SchoolBoy [girlId=2, id=11, name=小豪, grilName=小美],

SchoolBoy [girlId=3, id=12, name=小白, grilName=小花]]

这个表达式有一个orElse(null)这里要注意一下:通过修改id来出现id没有匹配的话,就返回null,

[SchoolBoy [girlId=1, id=10, name=小明, grilName=小英],

SchoolBoy [girlId=2, id=11, name=小豪, grilName=小美],

null]

少行数据,这样显然不是我们想要的,这里我们做了改成

我们使用orElse(schoolBoy),这样只是不在schoolBoy中设置name值而已。但不至于少一行数据

[SchoolBoy [girlId=1, id=10, name=小明, grilName=小英],

SchoolBoy [girlId=2, id=11, name=小豪, grilName=小美],

SchoolBoy [girlId=3, id=12, name=小白, grilName=null]]

你可能感兴趣的:(java)