diff --git a/src/perf/java/rx/operators/OperatorFlatMapPerf.java b/src/perf/java/rx/operators/OperatorFlatMapPerf.java index 6739fb0a1d..c20cff67f1 100644 --- a/src/perf/java/rx/operators/OperatorFlatMapPerf.java +++ b/src/perf/java/rx/operators/OperatorFlatMapPerf.java @@ -28,6 +28,7 @@ import rx.Observable; import rx.functions.Func1; import rx.jmh.InputWithIncrementingInteger; +import rx.jmh.LatchedObserver; import rx.schedulers.Schedulers; @BenchmarkMode(Mode.Throughput) @@ -62,6 +63,7 @@ public Observable call(Integer i) { @Benchmark public void flatMapIntPassthruAsync(Input input) throws InterruptedException { + LatchedObserver latchedObserver = input.newLatchedObserver(); input.observable.flatMap(new Func1>() { @Override @@ -69,7 +71,8 @@ public Observable call(Integer i) { return Observable.just(i).subscribeOn(Schedulers.computation()); } - }).subscribe(input.observer); + }).subscribe(latchedObserver); + latchedObserver.latch.await(); } @Benchmark