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

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);
    }

}