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(); @Data
public 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(); @Data public 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); } }