java - Project Reactor - 使用 defer() 使方法可重试

Question

最近，我尝试通过设置模拟依赖项来测试单元测试中的重试行为，以在最后返回成功的 mono.just() 之前返回多个 Mono.error() :

@Mock
Dependency dependency;

@InjectMocks
ClassUnderTest classUnderTest;

@Test
void someTest() {
    final Object object = new Object();
    when(dependency.method(anyString()))
        .thenReturn(Mono.error(new Exception()))
        .thenReturn(Mono.error(new Exception()))
        .thenReturn(Mono.error(new Exception()))
        .thenReturn(Mono.just(object));

    StepVerifier.create(classUnderTest.method("abc"))
        .expectNext(object)
        .verifyComplete();

    verify(dependency, times(4)).method("abc");
}

上面的设置不起作用，正如我后来发现的，Reactor中的重试不是通过在特定时间内调用该方法来完成的，而是通过调用该方法一次，获取发布者，然后重新订阅它一次又一次。

class ClassUnderTest {
    private Dependency dependency;

    public Mono<Object> method(final String str) {
        return this.dependency.method(str).retryWhen(Retry.max(3));
    }
}

如果 Dependency#method 实现为:

，则重新订阅不会起作用:

class Dependency {
    private OtherDependency otherDependency;

    public Mono<Object> method(final String str) {
        return this.otherDependency.get(str).map(/* some mapping logic */);
    }
}

Dependency#method 无法对 OtherDependency#get 是否延迟做出太多假设。因此，依赖需要:

class Dependency {
    private OtherDependency otherDependency;

    public Mono<Object> method(final String str) {
        return Mono.defer(() -> this.otherDependency.get(str)).map(/* some mapping logic */);
    }
}

既然我们想说每个方法都应该是“可重试的”，这是否意味着我们需要始终使用defer(...)？

或者我误解了什么？

Answer 1

我应该考虑一下。

更简单的方法不是让所有方法本身“可重试”，而是在附加 retryWhen 运算符之前用 defer 包装发布者。

之前:

class ClassUnderTest {
    private Dependency dependency;

    public Mono<Object> method(final String str) {
        return dependency.method(str).retryWhen(Retry.max(3));
    }
}

之后:

class ClassUnderTest {
    private Dependency dependency;

    public Mono<Object> method(final String str) {
        return Mono.defer(() -> dependency.method(str)).retryWhen(Retry.max(3));
    }
}

现在，我们不必说所有方法都应该是 native “可重试的”，而是始终用 defer 包装要重试的发布者。