如何重试 Spring WebClient 以根据响应重试操作?

Question

我一直在学习Spring webflux，并陷入了这个。

我已经使用WebClient从Spring应用程序向REST API发出了请求。我想根据响应重试请求。假设响应具有属性status: 'not-ready'，那么我需要在一秒钟后重试相同的操作。

我尝试了以下方法，但不确定如何实现它

public Flux<Data> makeHttpRequest(int page) {
Flux<Data> data = webClient.get()
.uri("/api/users?page=" + page)
.retrieve()
.bodyToFlux(Data.class);
return data;
}

GET : /api/users returns the folowing response
ex: 1 {
status: 'ready',
data: [......]
}
ex: 2 {
status: 'not-ready',
data: null
}

任何帮助将不胜感激。

Answer 1

我认为实现所需的重试逻辑相当容易。 大致如下：

public class SampleRequester {
private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
private WebClient client;
public SampleRequester() {
this.client = WebClient.create();
}
public void scheduleRequest(int secondsDelay) {
scheduler.schedule(this::initiateRequest, secondsDelay, SECONDS);
}
private void initiateRequest() {
Mono<ResponseData> response = client.get()
.uri("example.com")
.retrieve()
.bodyToMono(ResponseData.class);
response.subscribe(this::handleResponse);
}
private void handleResponse(ResponseData body) {
if("ready".equals(body.getStatus())) {
System.out.println("No Retry");
// Do something with the data
} else {
System.out.println("Retry after 1 second");
scheduleRequest(1);
}
}
}

我使用以下简单模型进行响应：

public class ResponseData implements Serializable {
private String status;
public String getStatus() {
return status;
}
public void setStatus(String status) {
this.status = status;
}
}   

然后，您将调用 SampleRequester.scheduleRequest(0( 以立即发起第一次调用。 当然，您还需要进行调整，以避免对请求 url 进行硬编码、扩展响应数据、使延迟可配置和/或指数退避等。