【Java】Map steam merge NPE

今天踩了一坑，记录下。

java的map有一个merge方法，里面传入的key和mergeFunction不能是空，否则报错：

    default V merge(K key, V value,
            BiFunction<? super V, ? super V, ? extends V> remappingFunction) {
        Objects.requireNonNull(remappingFunction);
        Objects.requireNonNull(value);
        V oldValue = get(key);
        V newValue = (oldValue == null) ? value :
                   remappingFunction.apply(oldValue, value);
        if(newValue == null) {
            remove(key);
        } else {
            put(key, newValue);
        }
        return newValue;
    }

这个其实是由于调用stream的toMap方法发现的：

@Test
    public void test2(){
        List<String> list = new ArrayList<>();
        list.add("ly");
        list.add("ly1");
        Map m = list.stream().map(i -> {
            if ("ly".equals(i)) {
                return null;
            }
            return i;
        }).collect(Collectors.toMap(i -> i, i -> i));
        System.out.println(m);
    }

比如这里有一个list，然后如果在转map时，value有一个null，整个stream的调用就会失败，抛NPE。

这个case还是很常见的，比如：

我先有一个id的list然后想通过id查出一个pojo，组成一个<id, pojo>的map，但是通过一个service查pojo时，可能返回null，此时如果直接调用collect方法生成一个map，就会抛NPE。代码可能是这样的：

class Service{
    public String queryName(int id){
        if (id == 2) {
            return null;
        }
        return "name-" + id;
    }
}

这是一个mock的查询接口。

生成map的代码如下：

@Test
    public void test1(){
        Service service = new Service();
        List<Integer> userIds = new ArrayList<>();
        userIds.add(1);
        userIds.add(2);
        userIds.add(3);
        Map<Integer, String> data = userIds.stream()
                .filter(id -> id > 0)
                .collect(Collectors.toMap(Function.identity(), service::queryName));
    }

由于collect可能返回空的value，所以报错：

java.lang.NullPointerException
	at java.util.HashMap.merge(HashMap.java:1224)
	at java.util.stream.Collectors.lambda$toMap$58(Collectors.java:1320)
	at java.util.stream.ReduceOps$3ReducingSink.accept(ReduceOps.java:169)
	at java.util.stream.ReferencePipeline$2$1.accept(ReferencePipeline.java:175)

怎么解决呢？

原则肯定是保证在collect调用之前不存在null的value，但是最后又要以id为key，所以前面不能直接map（map了id信息就拿不到了），所以看似不能优雅的直接在stream的api里做一次filter。

有一个办法，借助guava的Tuple类，先map成一个tuple，这样就能记录下id和value，并且即使value是null也没有关系，然后再对value做filter，最后再collect，代码：

@Test
    public void test1(){
        Service service = new Service();
        List<Integer> userIds = new ArrayList<>();
        userIds.add(1);
        userIds.add(2);
        userIds.add(3);
        Map<Integer, String> data = userIds.stream()
                .filter(id -> id > 0)
                .map(id -> Pair.of(id, service.queryName(id)))
                .filter(pair -> Objects.nonNull(pair.getValue()))
                .collect(Collectors.toMap(pair -> pair.getKey(), pair -> pair.getValue()));
        System.out.println(data);
    }

{1=name-1, 3=name-3}

Process finished with exit code 0
 

这样就在stream调用中优雅地解决了过滤空value的问题。