Spring Webflux - как вызвать реактивную конечную точку в WebFilter - PullRequest
Новая
       4

Spring Webflux - как вызвать реактивную конечную точку в WebFilter

0 голосов
/ 12 апреля 2020

Я хочу реализовать WebFilter, который считывает указанный c заголовок входящего запроса, вызывает запрос GET к другой реактивной конечной точке REST со значением этого заголовка, а затем изменяет исходный запрос на значение ответа GET. .

Я хочу реализовать это в WebFilter, потому что я не хочу добавлять этот вызов функции к каждой функции в моем @RestController.

В настоящее время у меня есть это: 

@Component
class ExampleWebFilter(val webClients: WebClients) : WebFilter {
    override fun filter(exchange: ServerWebExchange, chain: WebFilterChain): Mono<Void> {
        println(exchange.request.headers)
        println(exchange.request.path)
        println(exchange.response)

        val test = webClients.internalAuthServiceClient.get()
                .uri("/api/authorisation/v1/test")
                .header("authToken", "authToken123")
                .retrieve().bodyToMono(String::class.java)

        println(test)

        exchange.mutate().request(
                exchange.request.mutate().header("newheader", test).build()
        )
        return chain.filter(exchange)
    }
}

@Component
class WebClients() {
    val internalAuthServiceClient = WebClient.builder()
            .baseUrl("lb://auth-service")
            .build()
}

Сейчас это явно не работает. Мой WebClient возвращает Mono, поэтому я не могу использовать это непосредственно в моем вызове mutate(), потому что для этого требуется строка. Я также не могу заставить WebClient вызывать операцию блокировки по понятным причинам.

Кто-нибудь знает, как я мог решить эту проблему?

1 Ответ

1 голос
/ 12 апреля 2020

Я не использую kotlin, поэтому вам придется конвертировать, но именно так вы и сделаете это в java. Я думаю, что все будет примерно так же. 

@Override
public Mono<Void> filter(ServerWebExchange serverWebExchange,
                         WebFilterChain webFilterChain) {
    return webClients
            .internalAuthServiceClient
            .get()
            .uri("/api/authorisation/v1/test")
            .retrieve()
            .bodyToMono(String.class)
            //Gonna assume you tested the above and all works
            //If get bad response or any error really
            // then print error + return empty mono
            .onErrorResume(e -> {
                e.printStackTrace();
                return Mono.empty();
            })
            //Map the header AFTER a response
            //only maps if Mono is not empty
            .map(header -> {
                serverWebExchange
                        .getRequest()
                        .mutate()
                        .header("web-filter", header);
                return serverWebExchange;
            })
            //return unchanged serverWebExchange if empty Mono
            .defaultIfEmpty(serverWebExchange)
            //Flatmap since filter returns Mono to prevent returning Mono<Mono<void>>
            .flatMap(webFilterChain::filter);
}

Проблема, с которой вы сталкиваетесь, связана с тем, что вы пытаетесь сделать это синхронно, тогда как вам нужно отобразить заголовок после получил ответ от WebClient.

...