我正在尝试使用
WebClient
来调用我的restServices。之前在 RestTemplate
上,我们定义了 ClientHttpRequestInterceptor
并将其附加到 RestTemplate
来拦截和修改请求。对于WebClient
,有没有办法做到同样的事情?
谢谢,
-斯雷尼
当您使用 WebClient Builder 时,您可以使用
ExchangeFilterFunction
方法传入 filter()
接口的实现。这相当于 ClientHttpRequestInterceptor
的 RestTemplate
。
ExchangeFilterFunction 文档:https://docs.spring.io/spring-framework/docs/current/javadoc-api/org/springframework/web/reactive/function/client/ExchangeFilterFunction.html
例如:
WebClient webClient = WebClient.builder()
.baseUrl("http://localhost:8080|)
.filter(logFilter())
.build();
private ExchangeFilterFunction logFilter() {
return (clientRequest, next) -> {
logger.info("External Request to {}", clientRequest.url());
return next.exchange(clientRequest);
};
}
就我而言,我需要从传入请求中获取一些标头并将它们放入我的请求中。我在这里找到了我需要的东西。
首先需要一个过滤器
/**
* ReactiveRequestContextFilter
*
* @author L.cm
*/
@Configuration
@ConditionalOnWebApplication(type = ConditionalOnWebApplication.Type.REACTIVE)
public class ReactiveRequestContextFilter implements WebFilter {
@Override
public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain) {
ServerHttpRequest request = exchange.getRequest();
return chain.filter(exchange)
.subscriberContext(ctx -> ctx.put(ReactiveRequestContextHolder.CONTEXT_KEY, request));
}
}
还有ReactiveRequestContextHolder
/**
* ReactiveRequestContextHolder
*
* @author L.cm
*/
public class ReactiveRequestContextHolder {
static final Class<ServerHttpRequest> CONTEXT_KEY = ServerHttpRequest.class;
/**
* Gets the {@code Mono<ServerHttpRequest>} from Reactor {@link Context}
* @return the {@code Mono<ServerHttpRequest>}
*/
public static Mono<ServerHttpRequest> getRequest() {
return Mono.subscriberContext()
.map(ctx -> ctx.get(CONTEXT_KEY));
}
}
最后就像 Michael McFadyen 说的,你需要配置一个 ExchangeFilterFunction
,就我而言,我需要身份验证和来源:
private ExchangeFilterFunction headerFilter() {
return (request, next) -> ReactiveRequestContextHolder.getRequest()
.flatMap(r -> {
ClientRequest clientRequest = ClientRequest.from(request)
.headers(headers -> {
headers.set(HttpHeaders.ORIGIN, r.getHeaders().getFirst(HttpHeaders.ORIGIN));
headers.set(HttpHeaders.AUTHORIZATION, r.getHeaders().getFirst(HttpHeaders.AUTHORIZATION));
})
.build();
return next.exchange(clientRequest);
});
}
编辑:在项目reactor的较新版本中Mono.subscriberContext()
不存在。必须用
Mono.deferContextual(Mono::just)
更改,并且
.subscriberContext(ctx -> ...)
与
.contextWrite(ctx -> ...)
更改