工作积累——stream().toMap 空指针异常解决问题时发现的小坑

这里会发现toMap最终使用的是Map的merge方法进行值的设置，而不是我们以为的put。那么在HashMap的merge中，方法的第一行就告诉你，我们不接受null的值。

  @Override
    public V merge(K key, V value,
                   BiFunction remappingFunction) {
        if (value == null)
            throw new NullPointerException();
		......
        return value;
    }

首先看另外一个方法是什么样的,首先会发现三个参数会被包裹到ReduceOps中

    @Override
    public final  R collect(Supplier supplier,
                               BiConsumer accumulator,
                               BiConsumer combiner) {
        return evaluate(ReduceOps.makeRef(supplier, accumulator, combiner));
    }

这个ReduceOps干嘛的？说实话好久没看源码了我也不知道，不管了继续往下看,到这一步就很明显了，原来第一个参数seedFactory会创建一个Map，然后第二个参数将state和循环对象作为参数传递到accumulator，而第三个方法在参数注释以及入参的ReducingSink 已经提醒你了这个是用来并发流中合并数据的内容。

    public static  TerminalOp
    makeRef(Supplier seedFactory,
            BiConsumer accumulator,
            BiConsumer reducer) {
        Objects.requireNonNull(seedFactory);
        Objects.requireNonNull(accumulator);
        Objects.requireNonNull(reducer);
        class ReducingSink extends Box
                implements AccumulatingSink {
            @Override
            public void begin(long size) {
                state = seedFactory.get();
            }

            @Override
            public void accept(T t) {
                accumulator.accept(state, t);
            }

            @Override
            public void combine(ReducingSink other) {
                reducer.accept(state, other.state);
            }
        }
        return new ReduceOp(StreamShape.REFERENCE) {
            @Override
            public ReducingSink makeSink() {
                return new ReducingSink();
            }
        };
    }

现在明白为什么使用第二个方式可以了，重点在于第二个参数。使用toMap的时候java默认使用Map的merge方法这使得我们没办法去设置null的值，而第二种方式我们可以在第二个参数中自己实现设置值的方式，比如我们可以下面这样，让我们用另外一种方式再错一次，笑:)

HashMap hashMap = 
                loop.parallelStream().collect(HashMap::new, 
                        (m, v) -> m.merge(v.getUserId(), Objects.requireNonNull(covertName(v)), (o,n) -> n), 
                        HashMap::putAll);

回到这个方法，我们没办法设置值为null，那么如果在最后一个参数(o,n) -> n中返回null会如何呢？

 Map rest = loop.stream()
                .collect(Collectors.toMap(User::getUserId, JavaMain::covertName, (o,n) -> Objects.equals(n,o)?null:n));

在HashMap的merge中有这么一段逻辑

    @Override
    public V merge(K key, V value,
                   BiFunction remappingFunction) {
      ......
        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;
        }
     .....
        return value;
    }

这个时候会发现如果合并后的值为null的时候并不会将null设置进去而是直接移除掉这个节点。就像下面的例子：

    public static void main(String[] args) {
        List loop = new ArrayList<>();
        for (int i = 0; i < 10; i++) {
            if (i == 5 || i == 6) {
                User user = new User(String.valueOf(5),"name" + String.valueOf(5));
                loop.add(user);
            } else {
                User user = new User(String.valueOf(i),"name" + String.valueOf(i));
                loop.add(user);
            }
        }
        Map rest = loop.stream()
                .collect(Collectors.toMap(User::getUserId, JavaMain::covertName,(n, o) -> Objects.equals(n,o)?null:n));
        System.out.println(JSON.toJSON(rest.keySet()));
    }

这个时候返回的key中不仅仅6不见了连5也不见了。

["0","1","2","3","4","7","8","9"]

所以这个时候使用的时候一定要注意否则在后续进行key值进行判断的时候，有可能导致错误的数量。

ps.磨磨唧唧写了一个多小时，emmmm，感觉没啥卵用的知识+1了