我们有一个服务器来检索OAUTH令牌,并且通过WebClient.filter方法将oauth令牌添加到每个请求中,例如
webClient
.mutate()
.filter((request, next) -> tokenProvider.getBearerToken()
.map(token -> ClientRequest.from(request)
.headers(httpHeaders -> httpHeaders.set("Bearer", token))
.build()).flatMap(next::exchange))
.build();
TokenProvider.getBearerToken returns Mono<String> since it is a webclient request (this is cached)
Run Code Online (Sandbox Code Playgroud)
我想要重试功能,发生401错误,将使令牌无效并再次尝试请求,我这样工作
webClient.post()
.uri(properties.getServiceRequestUrl())
.contentType(MediaType.APPLICATION_JSON)
.body(fromObject(createRequest))
.retrieve()
.bodyToMono(MyResponseObject.class)
.retryWhen(retryOnceOn401(provider))
private Retry<Object> retryOnceOn401(TokenProvider tokenProvider) {
return Retry.onlyIf(context -> context.exception() instanceof WebClientResponseException && ((WebClientResponseException) context.exception()).getStatusCode() == HttpStatus.UNAUTHORIZED)
.doOnRetry(objectRetryContext -> tokenProvider.invalidate());
}
Run Code Online (Sandbox Code Playgroud)
有没有办法将其移至webClient.mutate()..... build()函数?这样所有请求都将具有此重试功能?
我尝试添加作为过滤器,但似乎没有用,例如
.filter(((request, next) -> next.exchange(request).retryWhen(retryOnceOn401(tokenProvider))))
Run Code Online (Sandbox Code Playgroud)
对解决此问题的最佳方法有何建议?问候