ToB企服应用市场:ToB评测及商务社交产业平台

标题: Collectors.toMap的暗坑与避免方式 [打印本页]
作者: 我爱普洱茶    时间: 2024-4-23 19:59
标题: 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()），则不会报错。
小结


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



欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/) Powered by Discuz! X3.4