这篇文章主要介绍了Java8 Collectors.toMap的坑,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
按照常规思维,往一个map里put一个已经存在的key,会把原有的key对应的value值覆盖,然而通过一次线上问题,发现Java8中的Collectors.toMap反其道而行之,它默认给抛异常,抛异常…
线上业务代码出现Duplicate Key的异常,影响了业务逻辑,查看抛出异常部分的代码,类似以下写法:
Map map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName));
然后list里面有id相同的对象,结果转map的时候居然直接抛异常了。。查源码发现toMap方法默认使用了个throwingMerger
public static Collector<T, ?, Map> toMap(Function keyMapper, Function valueMapper) { return toMap(keyMapper, valueMapper, throwingMerger(), HashMap::new); } private static BinaryOperator throwingMerger() { return (u,v) -> { throw new IllegalStateException(String.format("Duplicate key %s", u)); }; }
那么这个throwingMerger是哪里用的呢?
public static <T, K, U, M extends Map> C<i style="color:transparent">来源gaodai$ma#com搞$代*码网</i>ollector toMap(Function keyMapper, Function valueMapper, BinaryOperator<u> mergeFunction, Supplier mapSupplier) { BiConsumer accumulator = (map, element) -> map.merge(keyMapper.apply(element), valueMapper.apply(element), mergeFunction); return new CollectorImpl(mapSupplier, accumulator, mapMerger(mergeFunction), CH_ID); }
这里传进去的是HashMap,所以最终走的是HashMap的merge方法。merge方法里面有这么一段代码:
if (old != null) { V v; if (old.value != null) v = remappingFunction.apply(old.value, value); else v = value; if (v != null) { old.value = v; afterNodeAccess(old); } else removeNode(hash, key, null, false, true); return v; }
相信只看变量名就能知道这段代码啥意思了。。如果要put的key已存在,那么就调用传进来的方法。而throwingMerger的做法就是抛了个异常。所以到这里就可以知道写的代码为什么呲了。。
如果不想抛异常的话,自己传进去一个方法即可,上述代码可以改成:
Map map = list.stream().collect(Collectors.toMap(Person::getId, Person::getName,(oldValue, newValue) -> newValue));
这样就做到了使用新的value替换原有value。
写代码调方法时,多看源码实现,注意踩坑!
以上就是Java8 Collectors.toMap的坑的详细内容,更多请关注gaodaima搞代码网其它相关文章!