Java 8中Collectors.toMap空指針異常源碼解析

更新時間：2023年08月10日 11:55:38 作者：十方

這篇文章主要為大家介紹了Java 8中Collectors.toMap空指針異常源碼解析,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進步,早日升職加薪

引言

當需要將一個List轉(zhuǎn)換為Map時，可以使用 Java 8 中的 Collectors.toMap() 方法，Map是由key-value組成的鍵值對集合，在使用Collectors.toMap() 方法時，如果值為空，會報空指針異常，下面通過一個實例來驗證一下。

Student.java 類

首先定義一個 Student.java 類

package com.magic.npe;
public class Student {
    private String name;
    private Integer age;
    public Student(String name, Integer age) {
        this.name = name;
        this.age = age;
    }
    public String getName() {
        return name;
    }
    public void setName(String name) {
        this.name = name;
    }
    public Integer getAge() {
        return age;
    }
    public void setAge(Integer age) {
        this.age = age;
    }
}

Test.java 類創(chuàng)建

再創(chuàng)建一個 Test.java 類，用來驗證將 List<Student> 轉(zhuǎn)換為 Map<String, Integer>。

package com.magic.npe;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
public class Test {
    public static void main(String[] args) {
        List<Student> students = new ArrayList<>();
        students.add(new Student("張三", 18));
        students.add(new Student("李四", 21));
        students.add(new Student("王五", null));
        Map<String, Integer> studentMap = students.stream().collect(Collectors.toMap(Student::getName, Student::getAge));
        System.out.println(studentMap);
    }
}

運行程序，直接報出如下的錯誤信息

Exception in thread "main" 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.ArrayList$ArrayListSpliterator.forEachRemaining(ArrayList.java:1374)
    at java.util.stream.AbstractPipeline.copyInto(AbstractPipeline.java:481)
    at java.util.stream.AbstractPipeline.wrapAndCopyInto(AbstractPipeline.java:471)
    at java.util.stream.ReduceOps$ReduceOp.evaluateSequential(ReduceOps.java:708)
    at java.util.stream.AbstractPipeline.evaluate(AbstractPipeline.java:234)
    at java.util.stream.ReferencePipeline.collect(ReferencePipeline.java:499)
    at com.magic.npe.Test.main(Test.java:16)

查看Collectors.toMap()

查看一下 Collectors.toMap() 方法的源碼，如下：

public static <T, K, U>
    Collector<T, ?, Map<K,U>> toMap(Function<? super T, ? extends K> keyMapper,
                                    Function<? super T, ? extends U> valueMapper,
                                    BinaryOperator<U> mergeFunction) {
    return toMap(keyMapper, valueMapper, mergeFunction, HashMap::new);
}

默認會創(chuàng)建一個 HashMap，繼續(xù)查看源碼：

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

通過 Map.merge() 方法來合并，源碼如下：

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;
}

可以看到，在 merge() 方法中，要求 value 值不能為空

Objects.requireNonNull(value);

繼續(xù)查看一下 Objects.requireNonNull() 方法的源碼

public static <T> T requireNonNull(T obj) {
    if (obj == null)
        throw new NullPointerException();
    return obj;
}

NullPointerException 異常

如果值為空，則會直接報出 NullPointerException 異常。

那么，對于這個空指針異常問題，如何解決呢？一般有兩種方式：

（1）替換空值null為一個默認值，比如0

package com.magic.npe;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Optional;
import java.util.stream.Collectors;
public class Test {
  public static void main(String[] args) {
      List<Student> students = new ArrayList<>();
      students.add(new Student("張三", 18));
      students.add(new Student("李四", 21));
      students.add(new Student("王五", null));
      // 如果age為null，則直接賦值默認值0
      Map<String, Integer> studentMap = students.stream().collect(Collectors.toMap(Student::getName, s -> Optional.ofNullable(s.getAge()).orElse(0)));
      System.out.println(studentMap);
  }
}

運行程序，可以看到如下輸出

{李四=21, 張三=18, 王五=0}

（2）調(diào)用 collect() 的其他實現(xiàn)方法

package com.magic.npe;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
public class Test {
  public static void main(String[] args) {
      List<Student> students = new ArrayList<>();
      students.add(new Student("張三", 18));
      students.add(new Student("李四", 21));
      students.add(new Student("王五", null));
      Map<String, Integer> studentMap = students.stream().collect(HashMap::new, (map, student) -> map.put(student.getName(), student.getAge()), HashMap::putAll);
      System.out.println(studentMap);
  }
}

運行程序，可以看到如下輸出