如何在WebFilter中调用反应性endpoint

Question

我想实现一个WebFilter，它读取传入请求的特定头，用这个头的值调用一个GET请求到另一个反应RESTendpoint，然后用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()
}

这显然现在不起作用。我的网络客户端正在返回单声道，所以我不能在我的Mutate（）调用中直接使用它，因为这需要一个字符串。出于显而易见的原因，我也不能真正使网络客户端调用成为阻塞操作。

有人知道我怎么解决这个问题吗？

Answer 1

我不使用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的响应后映射标题。