spring - 使用 Spring WebClient 重试时添加新 header

Question

webclientbuilder.baseUrl(url)
                .defaultHeaders(headers -> headers.addAll(requestHeader))
                .build()
                .post()
                .uri("/uri")
                .bodyValue(data)
                .exchange()
                .flatMap(response  -> {
                                if(response.statusCode() == HttpStatus.UNAUTHORIZED){
                                                //retry with updated token in header
                                }
                })
                //return bodyToMono of specific object when retry is done or if
                //response status is 2xx

如有任何关于如何处理此问题的建议，我们将不胜感激!正如评论所说，如果 statusCode 为 UNAUTHORIZED，并且如果 statusCode 为 2xx，则在重试发布请求之前，我需要将新 token 添加到 header 中，然后返回 bodyToMono。

Answer 1

您可以通过向网络客户端添加过滤器来解决此问题:

public ExchangeFilterFunction retryOn401() {
    return (request, next) -> next.exchange(request)
            .flatMap((Function<ClientResponse, Mono<ClientResponse>>) clientResponse -> {
                if (clientResponse.statusCode() == HttpStatus.UNAUTHORIZED) {
                    return authClient.getUpdatedToken() //here you get a Mono with a new & valid token
                            .map(token -> ClientRequest
                                    .from(request)
                                    .headers(headers -> headers.replace("Authorization", Collections.singletonList("Bearer " + token)))
                                    .build())
                            .flatMap(next::exchange);
                } else {
                    return Mono.just(clientResponse);
                }
            });
}

因此，当 webclient 重试未经授权的请求时，它可以获得一个新的 token 并在执行重试之前将其设置在 header 中。 确保将其添加到网络客户端:

webclientbuilder.baseUrl(url)
            .defaultHeaders(headers -> headers.addAll(requestHeader))
            .filter(retryOn401())
            .build();