Collectors.toMap的暗坑与避免方式

使用Java的stream中的Collectors可以很方便地做容器间的转换，可以少写很多代码。但是其中有暗含的坑需要注意和避免，本文探讨Collectors.toMap（JDK8版本）。
Collectors.toMap可以将一个流转化成Map，常见于需要将List转换成Map以便于进一步操作的场景，比如在通过RPC接口获取一个返回结果、从DB中查询到匹配的多条数据后，对其按某个字段（经常是主键id）做分组。这里先定义一个简单的类：

1. public class User {
2.     private Long id;
4.     private String name;
5. }

复制代码

后续所有代码的目的均为将这个User类组成的List转换为id作为key、name作为value的Map。
1. 不使用stream

不使用stream时，需要先new一个map，然后手动把list的每一项放入map

1.     public void test0() {
2.         User user1 = new User();
3.         user1.setId(1L);
4.         user1.setName("1");
6.         User user2 = new User();
7.         user2.setId(2L);
8.         user2.setName("2");
10.         List<User> list = Lists.newArrayList();
11.         list.add(user1);
12.         list.add(user2);
13.         Map<Long, String> map = new HashMap<>();
14.         for(User user : list) {
15.             map.put(user.getId(), user.getName());
16.         }
17.     }

复制代码

2. 使用stream的Collectors.toMap

构造list的代码不变，转化map的代码可以简化为

1. Map<Long, String> map = list.stream().collect(Collectors.toMap(User::getId, User::getName));

复制代码

相较第一种，简洁多了。
但是，如果User1和User2的id是一样的，会发生什么情况？将代码中user2.setId(2L)修改为user2.setId(1L)，再次执行，发现抛异常java.lang.IllegalStateException: Duplicate key 1，说明在merge时报错了：key不允许重复。

但是有些情况下key确实是可以重复的，比如我调用上游的数据，上游没做校验和控制；再或者这个字段本身不是惟一的，多个数据可能重复。那么如何改进呢？
3. Collectors.toMap指定merge函数

可以自定义一个merge函数来确定key重复时，如何取value。比如下面这种写法，是保留第一个value。你也可以保留第二个，或者是做一些更复杂的处理。

1. Map<Long, String> map = list.stream().collect(Collectors.toMap(User::getId, User::getName, (x1,x2)->x1));

复制代码

4. value为null的场景

传入merge方法以后，看似万事大吉了，没想到还有坑。将user2.setName(null)，会发现抛了NullPointerException异常，异常栈信息为：

可以看到对应的源码里，value是不允许为null的。

虽然说Map的value是支持null值的，但是map自己的merge方法天生不支持，此时仅靠自定义merge方法也已经无能为力了。如果仍然想使用Collectors.toMap，需要手动处理null的值，比如：

1. Map<Long, String> map = list.stream().collect(Collectors.toMap(User::getId, value -> Optional.ofNullable(value.getName()).orElse("")));

复制代码

当然，这样处理后的map的value并不是实际的值，并不适用于所有场景。
这样看来，Collectors.toMap的局限性无法避免了，使用的时候要确认不会发生预期的问题。
当然你也可以做一些预处理，比如使用filter过滤掉value=null的数据，来规避这个问题。
5. key=null时会怎么样？

不会怎么样，一切正常。

1. public void test5() {    User user1 = new User();    user1.setId(null);    user1.setName("1");    User user2 = new User();    user2.setId(2L);    user2.setName("2");    List list = Lists.newArrayList();    list.add(user1);    list.add(user2);    Map<Long, String> map = list.stream().collect(Collectors.toMap(User::getId, User::getName));    System.out.print(map);}

复制代码

6. 还有什么要注意的？

上述的例子，均是建立在list不为空的前提下进行的。如果list本身为null，在调用stream()时自然也会抛NullPointerException；如果list是空的（内容为空但是容器本身初始化过，如list = new ArrayList()），则不会报错。
小结

• 调用stream()前先确定容器本身是否为null
  
• 如果不确定要通过Collectors.toMap转换为map的源容器的数据
  
  
  
  • 对应key是否会重复，可以在toMap()传入merge方法
    
  • 对应value是否为null，可以做过滤或指定默认值
    
  
  
• 如果不想思考和求证，还是继续用for循环吧
  

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！