当前位置: 代码迷 >> 综合 >> 使用stream的Collectors.toMap()方法常见的两个问题
  详细解决方案

使用stream的Collectors.toMap()方法常见的两个问题

热度:40   发布时间:2023-12-18 14:08:12.0

java8开始的流式编程很大程度上简化了我们的代码,提高了开发效率。

我们经常会使用到stream的Collectors.toMap()来将List转换Map,但是在使用过程中有两个小坑需要注意:

1、java.lang.IllegalStateException: Duplicate key

2、java.lang.NullPointerException

第一个是由于在List转Map过程中Map集合的key重复导致的;

第二个是由于在List转Map过程中Map集合的value有null导致的(当存在value值为空时,使用Collectors.toMap()会报NPE,因为底层调用了Map的merge方法,而map方法规定了此处的vlue不能为null,从而抛出空指针异常);

解决方案:

1、Collectors.toMap(dto ->key值 , dto -> dto,(v1,v2) -> v1)
在后面添加(v1,v2)->v1 指定选取第一个值 当key值重复的时候,根据情况而定选取第一个还是第二个)

2、自定义一个Map来接收,不使用Collectors.toMap()

第一种情况示例:

import com.google.common.collect.Lists;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import lombok.Data;public class Test {private static List<User> userList = Lists.newArrayList();@Datapublic static class User {private String userCode;private String userName;}/*** 初始化数据* (这里的userCode=10002重复)*/public static void initData() {User user1 = new User();user1.setUserCode("10001");user1.setUserName("张三");User user2 = new User();user2.setUserCode("10002");user2.setUserName("李四");User user3 = new User();user3.setUserCode("10002");user3.setUserName("王五");userList.add(user1);userList.add(user2);userList.add(user3);}public static void main(String[] args) {initData();//反例//  Map<String, String> userMap = userList.stream().collect(Collectors.toMap(User::getUserCode, User::getUserName));//正例,在后面添加(u1,u2)->u1 指定选取第一个值 当key值重复的时候,根据情况而定选取第一个还是第二个Map<String, String> userMap = userList.stream().collect(Collectors.toMap(User::getUserCode, User::getUserName, (u1, u2) -> u1));System.out.println(userMap);}
}

第二种情况示例:

import com.google.common.collect.Lists;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
import lombok.Data;public class Test {private static List<User> userList = Lists.newArrayList();@Datapublic static class User {private String userCode;private String userName;}/*** 初始化数据* (这里的userCode=10003的userName为空)*/public static void initData() {User user1 = new User();user1.setUserCode("10001");user1.setUserName("张三");User user2 = new User();user2.setUserCode("10002");user2.setUserName("李四");User user3 = new User();user3.setUserCode("10003");user3.setUserName(null);userList.add(user1);userList.add(user2);userList.add(user3);}public static void main(String[] args) {initData();//反例// Map<String, String> userMap = userList.stream().collect(Collectors.toMap(User::getUserCode, User::getUserName));//正例 (如果对转换后的顺序有要求,这里还可以使用LinkedHashMap)Map<String, String> userMap = userList.stream().collect(HashMap::new, (map, user) -> map.put(user.getUserCode(), user.getUserName()), HashMap::putAll);System.out.println(userMap);}}

  相关解决方案