• 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。

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

  • 相关阅读:
    TClientDataSet[7]: 辨析 Field、FieldDef、Fields、FieldDefs、FieldList、FieldDefList
    TClientDataSet[11]: 分组统计
    TClientDataSet[14]: 测试 FindFirst、FindNext、FindLast、FindPrior、Found
    TClientDataSet[9]: 计算字段和 State
    这两天的收获
    又去北京
    关于博客园融资的想法
    《别为小事抓狂》读书笔记
    下周将去北京寻找投资
    服务器搬迁预告
  • 原文地址:https://www.cnblogs.com/z941030/p/9648363.html
Copyright © 2020-2023  润新知