Java如何使用Optional与Stream取代if判空逻辑(JDK8以上)

59 2019-9-22 13:44

通过本文你可以用非常简短的代码替代业务逻辑中的判null校验,并且很容易的在出现空指针的时候进行打日志或其他操作。

注:如果对Java8新特性中的lambda表达式与Stream不熟悉的可以去补一下基础,了解概念。

首先下面代码中的List放入了很多Person对象,其中有的对象是null的,如果不加校验调用Person的getXXX()方法肯定会报空指针错误,一般我们采取的方案就是加上if判断:

public class DemoUtils {
 
  public static void main(String[] args) {
    List<Person> personList = new ArrayList<>();
    personList.add(new Person());
    personList.add(null);
    personList.add(new Person("小明",10));
    personList.add(new Person("小红",12));
     
    for (Person person : personList) {
    //if判空逻辑
      if (person != null) {
        System.out.println(person.getName());
        System.out.println(person.getAge());
      }
    }
  }
 
  static class Person {
    private String name;
    private int age;
 
    public Person() {
    }
 
    public Person(String name, int age) {
      this.name = name;
      this.age = age;
    }
 
    public String getName() {
      return name;
    }
    public void setName(String name) {
      this.name = name;
    }
    public int getAge() {
      return age;
    }
    public void setAge(int age) {
      this.age = age;
    }
  }
}

其实,Java新特性Stream API 与 Optional 提供了更加优雅的方法:

利用Stream API 中的 filter将队列中的空对象过滤掉,filter(Objects::nonNull)的意思是,list中的每个元素执行Objects的nonNull()方法,返回false的元素被过滤掉,保留返回true的元素。

public static void main(String[] args) {
   List<Person> personList = new ArrayList<>();
   personList.add(new Person());
   personList.add(null);
   personList.add(new Person("小明",10));
   personList.add(new Person("小红",12));
 
   personList.stream().filter(Objects::nonNull).forEach(person->{
     System.out.println(person.getName());
     System.out.println(person.getAge());
   });
 }

示例中的personList本身也可能会null,如果业务逻辑中要求personList为null时打日志报警,可以用Optional优雅的实现:

public static void main(String[] args) {
  List<Person> personList = new ArrayList<>();
  personList.add(new Person());
  personList.add(null);
  personList.add(new Person("小明", 10));
  personList.add(new Person("小红", 12));
 
  Optional.ofNullable(personList).orElseGet(() -> {
    System.out.println("personList为null!");
    return new ArrayList<>();
  }).stream().filter(Objects::nonNull).forEach(person -> {
    System.out.println(person.getName());
    System.out.println(person.getAge());
  });
}

代码中的

orElseGet(() -> {
  //代替log
  System.out.println("personList为null!");
  return new ArrayList<>();
})

表示如果personList为null,则执行这2句代码,返回一个不含元素的List,这样当personList为null的时候不会报空指针错误,并且还打了日志。