如何在Stream上短路减少?

时间:2014-08-04 07:34:22

标签: java java-8 java-stream

假设我有一个布尔值流,我正在编写的reduce操作是||(OR)。我能否以一种方式编写它,以便在遇到true值时放弃对至少某些元素的评估?

我正在寻找一些优化(可能是一个并行流),不一定是完全优化,尽管后者会很棒。

1 个答案:

答案 0 :(得分:18)

我怀疑你想要这种类型的构造。

// stop when any element evaluates to true
boolean any = stream.anyMatch(t -> t);

你可以用peek

来检查
Stream.of(1, 2, 3, 4).peek(System.out::println).anyMatch(i -> i == 2);

打印

1
2

对于并行示例

AtomicInteger count = new AtomicInteger();
IntStream.range(0, 1000).parallel().peek(t -> count.incrementAndGet()).anyMatch(i -> i == 2);
System.out.println("count: " + count);

打印一个类似

的数字
count: 223

具体数字各不相同。

对于referencePipeline,anyMatch调用

@Override
public final boolean anyMatch(Predicate<? super P_OUT> predicate) {
    return evaluate(MatchOps.makeRef(predicate, MatchOps.MatchKind.ANY));
}

调用此

public static <T> TerminalOp<T, Boolean> makeRef(Predicate<? super T> predicate,
        MatchKind matchKind) {
    Objects.requireNonNull(predicate);
    Objects.requireNonNull(matchKind);
    class MatchSink extends BooleanTerminalSink<T> {
        MatchSink() {
            super(matchKind);
        }

        @Override
        public void accept(T t) {
            if (!stop && predicate.test(t) == matchKind.stopOnPredicateMatches) {
                stop = true;
                value = matchKind.shortCircuitResult;
            }
        }
    }

    return new MatchOp<>(StreamShape.REFERENCE, matchKind, MatchSink::new);
}

您可以在那里看到短路代码。