Java8 Collectors.toMap的坑

按照常规思维,往一个map里put一个已经存在的key,会把原有的key对应的value值覆盖,然而通过一次线上问题,发现Java8中的Collectors.toMap反其道而行之,它默认给抛异常,抛异常...

线上业务代码出现Duplicate Key的异常,影响了业务逻辑,查看抛出异常部分的代码,类似以下写法:

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

然后list里面有id相同的对象,结果转map的时候居然直接抛异常了。。查源码发现toMap方法默认使用了个throwingMerger

 1 public static <T, K, U>
 2 Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
 3                                 Function<? super T, ? extends U> valueMapper) {
 4     return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new);
 5 }
 6  
 7  
 8 private static <T> BinaryOperator<T> throwingMerger() {
 9     return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); };
10 }

那么这个throwingMerger是哪里用的呢?

 1 public static <T, K, U, M extends Map<K, U>>
 2 Collector<T, ?, M> toMap(Function<? super T, ? extends K> keyMapper,
 3                             Function<? super T, ? extends U> valueMapper,
 4                             BinaryOperator<U> mergeFunction,
 5                             Supplier<M> mapSupplier) {
 6     BiConsumer<M, T> accumulator
 7             = (map, element) -> map.merge(keyMapper.apply(element),
 8                                           valueMapper.apply(element), mergeFunction);
 9     return new CollectorImpl<>(mapSupplier, accumulator, mapMerger(mergeFunction), CH_ID);
10 }

这里传进去的是HashMap,所以最终走的是HashMap的merge方法。merge方法里面有这么一段代码:

 1 if (old != null) {
 2     V v;
 3     if (old.value != null)
 4         v = remappingFunction.apply(old.value, value);
 5     else
 6         v = value;
 7     if (v != null) {
 8         old.value = v;
 9         afterNodeAccess(old);
10     }
11     else
12         removeNode(hash, key, null, false, true);
13     return v;
14 }

相信只看变量名就能知道这段代码啥意思了。。如果要put的key已存在,那么就调用传进来的方法。而throwingMerger的做法就是抛了个异常。所以到这里就可以知道写的代码为什么呲了。。

如果不想抛异常的话,自己传进去一个方法即可,上述代码可以改成:

1 Map<Integer, String> map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName,(oldValue, newValue) -> newValue));

这样就做到了使用新的value替换原有value。

写代码调方法时,多看源码实现,注意踩坑!

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏Ryan Miao

java枚举类型学习

用的不多,但用的时候仅仅简单的使用,不太明白原理,今天就系统的学一下枚举。参考:java编程思想。 Update: 枚举可以当做数据字典来存储,通常只要一个字段...

4099
来自专栏行者常至

如何遍历一个实例的所有属性,得到属性的名称和值

1022
来自专栏函数式编程语言及工具

Scalaz(28)- ST Monad :FP方式适用变量

    函数式编程模式强调纯代码(pure code),主要实现方式是使用不可变数据结构,目的是函数组合(composability)最终实现函数组件的重复使用...

2008
来自专栏小L的魔法馆

第十四届浙江财经大学程序设计竞赛重现赛--A-A Sad Story

2997
来自专栏java初学

java异常处理及自定义异常的使用

1297
来自专栏企鹅号快讯

【新手笔记】关于Split方法

Split,把一个字符串分割成字符串数组。 以前是做C#的,前段时间做安卓开发,用了一下Split方法,发现分割后的长度比预期的少,检查了一下,发现java中的...

2207
来自专栏菩提树下的杨过

Flash/Flex学习笔记(8):ActionScript3.0中的面对对象

首先要习惯AS3.0的几个BT约定: 1.一个.as文件中,只能定义一个类 2.类名称必须与.as的文件名相同 3.类定义中必须要有package包声明 4.一...

1789
来自专栏行者常至

003.python科学计算库pandas(上)

版权声明:本文为博主原创文章,允许转载,请标明出处。 https://blog.csdn.net/qwdafedv/article/deta...

932
来自专栏算法修养

ZOJ 3715 Kindergarten Election

At the beginning of the semester in kindergarten, the n little kids (indexed fro...

2894
来自专栏编程心路

Java基础-序列化与反序列化

序列化和反序列化在面试中也经常考查,下面就总结一下 Java 中的序列化和反序列化。

893

扫码关注云+社区

领取腾讯云代金券