Java Stream如何优雅实现函数组合或回溯流数据？

场景说明 #

现有类结构如下：

class A {
    String s;

    String getS() {return s;}

    static boolean aFilter(String in) {return true or false;}

    static A aTransformation(A a) {return another A;}
}

希望通过Stream完成以下逻辑：从List<A>中提取s字段，用aFilter过滤，再回到A的流执行aTransformation，最终转成列表。最初设想的链式写法如下：

List<A> myAs = provideTheList();

myAs.stream()
    .map(A::getS)
    .filter(A::aFilter)
    .map("back to stream of A")
    .map(A::aTransformation)
    .toList();

但中途将流转换为String后，无法回溯到原A实例；现有可行写法又依赖lambda，破坏了链式表达的简洁性：

myAs.stream()
    .filter(a -> A.aFilter(a.getS()))
    .map(A::aTransformation)
    .toList();

优雅解决方案：自定义函数组合工具 #

Java本身没有提供直接的函数组合操作符，但可以通过自定义静态工具方法，将Function和Predicate组合成目标Predicate，从而避免lambda，保持链式风格。

1. 编写通用组合工具类 #

public class FunctionComposer {
    // 将Function<T,U>和Predicate<U>组合成Predicate<T>
    public static <T, U> Predicate<T> composePredicate(Predicate<U> predicate, Function<T, U> mapper) {
        return t -> predicate.test(mapper.apply(t));
    }
}

2. 组合方法引用实现过滤 #

使用工具类的组合方法，直接用方法引用完成过滤逻辑，保持链式表达的一致性：

myAs.stream()
    .filter(FunctionComposer.composePredicate(A::aFilter, A::getS))
    .map(A::aTransformation)
    .toList();

补充说明 #

• Stream是单向流式处理，一旦将元素转换为其他类型（如这里的String），就无法再获取原A实例，因此“回溯到A的流”的思路不可行。
• 上述工具方法是通用的，可复用在所有需要将字段提取和过滤逻辑组合的场景中，无需为每个场景编写单独的辅助函数。

内容的提问来源于stack exchange，提问作者Jean-Luc Delarbre