Java8中通过Stream使用Collectors.toMap及踩坑记录

1. 说明

        Java8中添加Stream流操作，可以简化部分操作，使得代码更为简洁。本文主要记录部分使用方法及踩坑记录

2. 简单示例

        如果需要将一个List转为Map，key为对象的userId, value为userName。

2.1 非Stream实现

在Java7中通常可以使用以下代码实现：

@Test
public void test01() {
        List list = new ArrayList<>();
        list.add(new User("1", null, null));
        list.add(new User("2", "zhang", "wu"));
        list.add(new User("1", "qiqi", "wu"));
        HashMap map = new HashMap<>();
        for (User user : list) {
            map.put(user.getUserId(), user.getUserName());
        }
        System.out.println(map);
}

2.2 Stream实现

@Test
public void test02() {
        List list = new ArrayList<>();
        list.add(new User("2", "zhang", "wu"));
        list.add(new User("1", "qiqi", "wu"));
        Map collect = list.stream().collect(Collectors.toMap(User::getUserId, User::getUserName));
        System.out.println(collect);
}

2.3 踩坑点

        但是如果按照2.2中在项目中去实现是远远不够的，实际使用中会出现很多bug。例如更换下输入数据便会报错。

List list = new ArrayList<>();
list.add(new User("1", null, null));
list.add(new User("2", "zhang", "wu"));
list.add(new User("1", "qiqi", "wu"));

报错场景及原因

• 需要自定义key重复处理策略，不然key重复会报错java.lang.IllegalStateException: Duplicate key xxx
• value为null，会报空指针异常java.lang.NullPointerException

解决方案

• 自定义重复策略， 例如：(s1, s2) -> s1
• 处理value为空问题，例如： u.getUserName() == null ? "" : u.getUserName()

@Test
public void test03() {
        List list = new ArrayList<>();
        list.add(new User("1", null, null));
        list.add(new User("2", "zhang", "wu"));
        list.add(new User("1", "qiqi", "wu"));
        Map collect = list.stream().collect(Collectors.toMap(User::getUserId, u ->
                        u.getUserName() == null ? "" : u.getUserName()
                , (s1, s2) -> s1));
        System.out.println(collect);
}