Java 流：对链接使用可选的 filter() 操作

Question

注意：这个问题不与java.util.Optional相关。

在处理流时，我经常使用这样的逻辑：

 Stream<FooBar> stream = myInitialStream();
 if (needsFilter1) stream = stream.filter(c -> whatever1());
 if (needsFilter2) stream = stream.filter(c -> whatever2());
 ...
 return stream.collect(toList());

我想要实现的是使用链接将上面的代码转换为单个表达式。我发现这更具可读性和直截了当。 到目前为止，我发现实现这一目标的唯一方法是：

return myInitialStream()
       .filter(needsFilter1? c->whatever1() : c->true)
       .filter(needsFilter2? c->whatever2() : c->true)
       .collect(toList());

不过，这会对那些琐碎的c->true lamdas 进行不必要的调用，这可能会在扩展时产生一些性能成本。

所以我的问题是：有没有更好的方法来生成包含可选过滤的链式流表达式？

更新：也许我说得不够清楚，但我的问题的重点是找到一个单表达式解决方案。如果我必须使用多个语句（例如，初始化谓词），我还可以使用问题的第一个代码块，它本质上是相同的。

Answer 1

使用Predicate::and返回一个新的谓词根据条件链接谓词。

Predicate<FooBar> predicate = c -> whatever();

if (condition1) { predicate = predicate.and(c -> whatever1()); }
if (condition2) { predicate = predicate.and(c -> whatever2()); }

List<FooBar> dest = list.stream()
    .filter(predicate)
    .collect(Collectors.toList());

---

更新请求单个表达式时。无论如何，您需要为谓词提供映射条件的来源。使用数据结构Map<Supplier<Boolean>, Predicate<Integer>>，其中键是决定是否应使用值（Predicate<FooBar>）的条件的Supplier。

使用将这些Predicates 与Predicate::and 链接，将映射的条目减少到新的Predicate<FooBar>，它们的Supplier<Boolean> 返回true（条件有效）。

拥有Map的条件：

Map<Supplier<Boolean>, Predicate<FooBar>> map = new HashMap<>();
map.put(() -> needsFilter1, c -> whatever1());
map.put(() -> needsFilter2, c -> whatever2());
...

这是一个 Stream 声明：

List<Integer> dest = list
        .stream()
        .filter(map.entrySet()                            // filter with a predicate ...
                .stream()
                .filter(e -> e.getKey().get())            // .. where a condition is 'true'
                .map(Entry::getValue)                     // .. get Predicates
                .reduce(i -> true, (l, r) -> l.and(r)))   // .. reduce them using AND
        .collect(Collectors.toList());               

Answer 2

我的解决方案有点晚了，无论如何我会把它留在这里。

我有一个想法是编写一个构建器来构造一个复杂的Predicate，但最终得到了一个类FilterCondition 和一个方法FilterCondition.combine。

Stream.of("123", "1", "12345", "", "12", "", "2")
    .filter(FilterCondition.<String>combine(
                FilterCondition.of(() -> true, s -> s.contains("3")),
                FilterCondition.of(() -> true, s -> s.contains("2")),
                FilterCondition.of(() -> false, s -> s.isEmpty())
            ).toPredicate())
    .collect(Collectors.toList());

使用FilterCondition.of 和FilterCondition.combine, 的静态导入，它看起来会更好。

Stream.of("123", "1", "12345", "", "12", "", "2")
    .filter(combine(
                of(() -> true, s -> s.contains("3")),
                of(() -> true, s -> s.contains("2")),
                of(() -> false, String::isEmpty)
            ).toPredicate())
    .collect(Collectors.toList());

FilterCondition<T> 基本上是一个Predicate<T>，带有一个额外的条件来检查是否应该应用predicate。

FilterCondition.combine 需要一些 FilterConditions 并组成一个组合。

class FilterCondition<T> {
    private final Supplier<Boolean> filterEnabled;
    private final Predicate<T> predicate;

    private FilterCondition(Supplier<Boolean> filterEnabled, Predicate<T> predicate) {
        this.filterEnabled = filterEnabled;
        this.predicate = predicate;
    }

    public static <T> FilterCondition<T> of(Supplier<Boolean> filterEnabled, Predicate<T> predicate) {
        return new FilterCondition<>(filterEnabled, predicate);
    }

    @SafeVarargs
    public static <T> FilterCondition<T> combine(FilterCondition<T>... conditions) {
        return new FilterCondition<>(
                () -> true,
                Arrays.stream(conditions).filter(i -> i.filterEnabled.get()).map(i -> i.predicate).reduce(Predicate::and).orElse(t -> true)
        );
    }

    public Predicate<T> toPredicate() {
        return predicate;
    }

}