Spring Webflux - 即使第一个 Mono 响应错误信号，链接的 Mono 也得到解决

Question

这是我的代码：

public Mono<Foo> doSomething(String fooId, String barId) {

    Mono<Foo> firstResult = firstServiceCall(fooId, barId);
    Mono<List<Baz>> secondResult = secondServiceCall(fooId);

    return firstResult.flatMap(secondResult::thenReturn);
}

private Mono<Foo> firstServiceCall(String fooId, String barId) {
    return fooRepo
            .findByFooIdAndBarId(fooId, barId)
            .switchIfEmpty(Mono.error(new ResponseStatusException(NOT_FOUND)))
            .filter(Foo::isSomething)
            .switchIfEmpty(Mono.error(new ResponseStatusException(UNPROCESSABLE_ENTITY)))
            .doOnNext(foo -> foo.setSomething(false))
            .flatMap(fooRepo::save);
}

private Mono<List<Baz>> secondServiceCall(String fooId) {
    var filter = new BazSearchFilter();
    filter.setFooId(fooId);
    return bazRepo
            .findAllByFilter(filter)
            .doOnNext(baz -> baz.setStatus(BazStatus.SOMETHING))
            .collectList()
            .flatMapMany(bazRepo::saveAll)
            .collectList();
}

由于某种原因，doSomethingMethod 总是调用 secondServiceCall 方法，尽管从 firstServiceCall 方法传播了错误信号（NOT_FOUND 或 UNPROCESSABLE_ENTITY 场景）。

由于我使用的是 flatMap，我希望 secondServiceCall 不会关闭，但也许我遗漏了一些东西。

有谁知道如何解决这个问题？

Answer 1

你应该重写 doSomething(...) 方法如下：

public Mono<Foo> doSomething(String fooId, String barId) {

    Mono<Foo> firstResult = firstServiceCall(fooId, barId);

    return firstResult.doOnNext(foo -> secondServiceCall(fooId));
}

或如下：

public Mono<Foo> doSomething(String fooId, String barId) {

    Mono<Foo> firstResult = firstServiceCall(fooId, barId);
    Mono<List<Baz>> secondResult = Mono.defer(() -> secondServiceCall(fooId));

    return firstResult.flatMap(secondResult::thenReturn);
}

很有意思的是，这个方法也能按预期工作：

public Mono<Foo> doSomething(String fooId, String barId) {

    Mono<Foo> firstResult = firstServiceCall(fooId, barId);

    return firstResult.flatMap(foo -> secondServiceCall(fooId).thenReturn(foo));
}

绝对是 lambda 可以解决问题