Java Optional 多次 ifPresent 调用链式处理的优雅之道

// 期望的链式调用方式，但不可行
animalService.getAllAnimals().findFirst()
    .ifPresent(Animal::drink) // 返回void
    .ifPresent(Animal::eat);   // 编译错误

default Consumer<T> andThen(Consumer<? super T> after) { ... }

import java.util.Arrays;
import java.util.function.Consumer;

public final class OptionalUtils {

    private OptionalUtils() {
        // 私有构造函数，防止实例化
    }

    /**
     * 组合多个 Consumer，使其按顺序执行。
     * 如果 Optional 存在值，所有组合的 Consumer 将被依次应用。
     *
     * @param first 第一个要执行的 Consumer。
     * @param others 其他要按顺序执行的 Consumer。
     * @param <T> Consumer 接受的类型。
     * @return 一个复合的 Consumer，它将按传入顺序执行所有 Consumer。
     */
    @SafeVarargs
    public static <T> Consumer<T> combine(Consumer<T> first, Consumer<T>... others) {
        // 使用 Stream 的 reduce 操作将所有 Consumer 串联起来
        // first 作为初始值，后续的 Consumer 通过 andThen 累积
        return Arrays.stream(others).reduce(first, Consumer::andThen);
    }
}

import java.util.Optional;
import java.util.function.Consumer;

// 假设 Animal 类和 animalService 已经定义
class Animal {
    private String name;

    public Animal(String name) {
        this.name = name;
    }

    public void eat() {
        System.out.println(name + " is eating.");
    }

    public void drink() {
        System.out.println(name + " is drinking.");
    }

    public void sleep() {
        System.out.println(name + " is sleeping.");
    }
}

class AnimalService {
    public Optional<Animal> findFirst() {
        // 模拟从数据库或其他源获取第一个动物
        return Optional.of(new Animal("Lion"));
        // return Optional.empty(); // 测试没有值的情况
    }
}

public class OptionalChainingDemo {
    public static void main(String[] args) {
        AnimalService animalService = new AnimalService();

        // 使用 combine 方法将多个 Consumer 组合起来
        animalService.findFirst()
            .ifPresent(OptionalUtils.combine(
                Animal::drink,
                Animal::eat,
                Animal::sleep
            ));

        System.out.println("---");

        // 也可以直接组合两个 Consumer
        animalService.findFirst()
            .ifPresent(Animal::drink.andThen(Animal::eat));

        System.out.println("---");

        // 如果 Optional 为空，则什么也不会发生
        new AnimalService() {
            @Override
            public Optional<Animal> findFirst() {
                return Optional.empty();
            }
        }.findFirst()
        .ifPresent(OptionalUtils.combine(Animal::drink, Animal::eat));
    }
}

Lion is drinking.
Lion is eating.
Lion is sleeping.
---
Lion is drinking.
Lion is eating.
---