告别丑陋判空,一个 Optional 类搞定

2022-10-21 15:31:24 浏览数 (1)

概述

最近项目组内做 code review,充斥着大量的、原始的、丑陋的判空语句,大致类似下面的代码:

代码语言:javascript复制
if (user != null) {
    Address address = user.getAddress();
    if (address != null) {
        Country country = address.getCountry();
        if (country != null) {
            String isocode = country.getIsocode();
            if (isocode != null) {
                isocode = isocode.toUpperCase();
            }
        }
    }
}

让整体的代码显得十分的臃肿庞大丑陋,那么怎么办呢?其实一行代码就可以搞定:

代码语言:javascript复制
String result = Optional.ofNullable(user)
    .map(u -> u.getAddress())
    .map(a -> a.getCountry())
    .map(c -> c.getIsocode())
    .orElse("default");

利用 Optional 这个 jdk8 中引入的类就可以优雅的处理,现在我们来详细讲解下这个类的使用和源码。

创建 Optional 实例

Optional 类, 是对 value 值进行了包装,它的值可能是 null, 也可能不是 null,一共有两个方法创建 Optional 实例

  1. static Optional of(T value)

说明: 传入的 value 为 null,会抛错,也就是说 value 不许不为 null

演示:

代码语言:javascript复制
@Test
public void testOf() {
    Integer value = 2;
    // 正常
    Optional<Integer> op = Optional.of(value);

    value = null;
    // 报空指针
    op = Optional.of(value);
}

源码:

  1. static Optional ofNullable(T value)

说明: 传入的 value 可能是 null, 也可能不是 null

演示:

代码语言:javascript复制
@Test
    public void testOfNullable() {
        Integer value = 2;
        // 正常
        Optional<Integer> op = Optional.ofNullable(value);
        value = null;
        // 不报错
        op = Optional.ofNullable(value);
    }

源码:

获取 Optional 中的值

以前创建了 Optional 实例,现在你可能要获取里面的 value, 有下面几个方法。

  1. T get()

说明: 最朴素的获取原生 value 的方法,如果 value 是空,则直接抛出异常,否则返回。

演示:

代码语言:javascript复制
@Test
public void testGet() {
    Integer value = 2;
    // 正常
    Optional<Integer> op = Optional.ofNullable(value);
    Integer opVal = op.get();
    Assert.assertEquals(opVal, value);
    
    op = Optional.ofNullable(null);
    // 会抛出异常
    op.get();
}

源码:

  1. T orElse(T other)

说明: 如果值存在返回,否则返回 orElse 中传入的 other

演示:

代码语言:javascript复制
@Test
public void testOrElse() {
    // 正常
    Optional<Integer> op = Optional.ofNullable(2);
    Integer opVal = op.orElse(3);
    Assert.assertEquals(opVal, new Integer(2));

    op = Optional.ofNullable(null);
    // 为空,则返回3
    opVal = op.orElse(3);
    Assert.assertEquals(opVal, new Integer(3));
}

源码:

  1. T orElseGet(Supplier<? extends T> other)

说明: 如果存在则返回该值,否则调用 other 这个函数编程并返回该调用的结果。

演示:

代码语言:javascript复制
@Test
public void testOrElseGet() {
    // 正常
    Optional<Integer> op = Optional.ofNullable(2);
    Integer opVal = op.orElseGet(()-> {
        return new Integer(3);
    });
    Assert.assertEquals(opVal, new Integer(2));

    op = Optional.ofNullable(null);
    // 为空,则返回3
    opVal = op.orElseGet(()-> {
        return new Integer(3);
    });
    Assert.assertEquals(opVal, new Integer(3));
}

源码:

问题: orElseGet 和 orElse 有什么区别吗?

orElse() 和 orElseGet() 的不同之处在于当 ofNullable() 传入参数不为空时,orElse() 方法仍然创建了 other 这个 对象。与之相反,orElseGet() 方法不创建对象。在执行较密集的调用时,比如调用 Web 服务或数据查询,这个差异会对性能产生重大影响。

而且我们还可以在 orElseGet 方法中加些日志,可以把这种为空的异常情况暴露出来。

  1. T orElseThrow(Supplier<? extends X> exceptionSupplier)

说明: 如果存在则返回该值,否则为空的话可以抛出自定义的异常。

演示:

代码语言:javascript复制
@Test
public void testOrElseThrow() {
    Optional<Integer> op = Optional.ofNullable(null);
    // 为空,则抛出指定的异常类型
    Integer opVal = op.orElseThrow(()-> {
        return new RuntimeException();
    });
    // 或抛出runtime异常
    Assert.assertEquals(opVal, new Integer(3));
}

源码:

判断 Optional 是否为空

  1. boolean isPresent()

说明: 判断 value 是否为空

演示:

代码语言:javascript复制
@Test
public void testIsPresent() {
    Optional<Integer> op = Optional.ofNullable(null);
    // 为空
    Assert.assertFalse(op.isPresent());
}

源码:

  1. void ifPresent(Consumer<? super T> consumer)

说明: 如果存在值,则调用对应的 consumer 方法,否则不执行任何操作。

演示:

代码语言:javascript复制
@Test
public void testIfPresent() {
    Optional<Integer> op = Optional.ofNullable(5);
    op.ifPresent(value -> {
        // 如果存在,打印出来
        System.out.println(value);
    });
}

源码:

Optional 中的过滤、转换方法

此外 Optional 额外还提供了一些过滤、转换的方法。

  1. Optional filter(Predicate<? super T> predicate)

说明: 对 Optional 中的 value 进行过滤,如果不匹配,返回空

演示:

代码语言:javascript复制
@Test
public void testFilter() {
    // 不满足过滤条件,返回空 Optional
    Optional<String> op = Optional.ofNullable("10").filter(item -> "15".equals(item));
    Assert.assertFalse(op.isPresent());
}

源码:

  1. Optional map(Function<? super T, ? extends U> mapper)

说明: 对 Optional 中的 value 进行转换映射为另外一个对象,如果 value 为空,返回 empty Optional

演示:

源码:

小结: 回到概述的案例,就是通过不断的 map, 链式调用返回内层对象的值。

  1. Optional flatMap(Function<? super T, Optional> mapper)

说明: 接受一个返回值为 Optional 的映射函数参数,该返回值亦是 flatMap 方法的返回值若结果为空,则返回 空 Optional。它也 map 的区别,我们用一个例子演示出来。

演示:

源码:

小结: 如果对于返回值非 Optional 类型,可以用 map 方法, 否则使用 flatMap 更加方便

总结

本文讲解了 Optional 这个类的使用和源码,非常的简单,实用,大家可以在自己的项目中用起来了。

0 人点赞