web-dev-qa-db-ja.com

WebClientのすべてのリクエストの再試行を追加する

OAUTHトークンを取得するサーバーがあり、oauthトークンがWebClient.filterメソッドを介して各リクエストに追加されます。

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)

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());
    }

これをwebClient.mutate（）..... build（）関数に移動する方法はありますか？すべてのリクエストがこの再試行機能を持つように？

フィルターとして追加しようとしましたが、機能しないようです。

.filter(((request, next) -> next.exchange(request).retryWhen(retryOnceOn401(tokenProvider))))

これに取り組む最善の方法の提案はありますか？よろしく

spring-webfluxproject-reactor
8
Kevin Hussey

これは、再試行が例外でのみ機能することを確認した後で明らかになりました。clientResponseオブジェクトは応答を保持するだけなので、bodyToが呼び出されたときにのみ、httpステータスで例外がスローされるため、これを修正します。 、この動作を模倣できます

@Bean(name = "retryWebClient")
    public WebClient retryWebClient(WebClient.Builder builder, TokenProvider tokenProvider) {
        return builder.baseUrl("http://localhost:8080")
                .filter((request, next) ->
                        next.exchange(request)
                            .doOnNext(clientResponse -> {
                                    if (clientResponse.statusCode() == HttpStatus.UNAUTHORIZED) {
                                        throw new RuntimeException();
                                    }
                            }).retryWhen(Retry.anyOf(RuntimeException.class)
                                .doOnRetry(objectRetryContext -> tokenProvider.expire())
                                .retryOnce())

                ).build();
    }

リピート/リトライで機能の1つを編集すると、元のリクエストは変更されません。私の場合、新しいOAuthトークンを取得する必要がありましたが、上記は同じ（期限切れ） ）トークン。交換フィルターを使用してこれを行う方法を考えました。OAuth password-flowがspring-security-2.0になったら、AccessTokensなどと統合できるはずですが、その間

ExchangeFilterFunction retryOn401Function(TokenProvider tokenProvider) {
        return (request, next) -> next.exchange(request)
                .flatMap((Function<ClientResponse, Mono<ClientResponse>>) clientResponse -> {
                    if (clientResponse.statusCode().value() == 401) {
                        ClientRequest retryRequest = ClientRequest.from(request).header("Authorization", "Bearer " + tokenProvider.getNewToken().toString()).build();
                        return next.exchange(retryRequest);
                    } else {
                        return Mono.just(clientResponse);
                    }
                });
    }
11
Kevin Hussey