Spring RequestContextHolder 和 WebTestClient
[英]Spring RequestContextHolder and WebTestClient
问题描述
I'm using Spring RequestContextHolder in the controller and it works fine.
我在 controller 中使用 Spring RequestContextHolder ，它工作正常。 But in the unit test I get java.lang.IllegalStateException using WebTestClient .但在单元测试中，我使用WebTestClient java.lang.IllegalStateException Here is an example:这是一个例子：
package demo.reactive.controller;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.context.request.RequestContextHolder;
import reactor.core.publisher.Mono;
@RestController
public class FooController {
@GetMapping("/foo")
public ResponseEntity<Mono<String>> foo() {
String sessionId = RequestContextHolder.currentRequestAttributes().getSessionId();
return ResponseEntity.ok(Mono.just(sessionId));
}
}
package demo.reactive;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.reactive.WebFluxTest;
import org.springframework.test.web.reactive.server.WebTestClient;
import demo.reactive.controller.FooController;
@WebFluxTest(FooController.class)
class DemoReactiveApplicationTests {
@Autowired private WebTestClient client;
@Test
void contextLoads() {
client.get().uri("/foo").exchange().expectStatus().isOk();
}
}
java.lang.IllegalStateException: No thread-bound request found: Are you referring to request attributes outside of an actual web request, or processing a request outside of the originally receiving thread?
How can I get the unit test to work.我怎样才能让单元测试工作。 Or it's not even suppose to use RequestContextHolder with WebFlux?或者甚至不打算将RequestContextHolder与 WebFlux 一起使用？
2 个解决方案
解决方案1
1 已采纳 2020-05-11 00:32:31
Spring RequestContextHolder cannot be used with Spring Webflux. Spring RequestContextHolder不能与 Spring Webflux 一起使用。 RequestContextHolder primary works using java.lang.ThreadLocal . RequestContextHolder主要使用java.lang.ThreadLocal工作。 This clearly doesn't fit with Reactor architecture.这显然不适合 Reactor 架构。
Having said that, We can implement an alternative using reactor's Mono#subscriberContext() operator.话虽如此，我们可以使用反应器的Mono#subscriberContext()运算符实现替代方案。 The best way to create a reusable solution is to implement spring WebFilter to get hold of request attributes and set it in the reactor context.创建可重用解决方案的最佳方法是实现WebFilter WebFilter 以获取请求属性并将其设置在反应器上下文中。
public class WebRequestAttributesContextFilter implements WebFilter {
public static final String WEB_REQUEST_ATTRIBUTES = "WebRequestAttributes";
@Override
public Mono<Void> filter(ServerWebExchange serverWebExchange, WebFilterChain webFilterChain) {
return webFilterChain.filter(serverWebExchange)
.subscriberContext(context -> context.put(WEB_REQUEST_ATTRIBUTES, serverWebExchange.getAttributes()));
}
}
The attributes (sessionId in your case) can then be read from the reactor context anywhere in the request processing pipeline as follows:然后可以从请求处理管道中的任何位置的反应器上下文中读取属性（在您的情况下为 sessionId），如下所示：
@GetMapping("/foo")
public Mono<String> helloWorld() {
return Mono.subscriberContext().map(context -> (String) context.<Map<String, Object>>get(
WebRequestAttributesContextFilter.WEB_REQUEST_ATTRIBUTES
).get("session"));
}
解决方案2
0 2020-05-10 15:26:19
You should rather use subscriberContext from Mono/Flux API.您应该使用Mono/Flux API 中的subscriberContext 。
Take a look on docs: https://projectreactor.io/docs/core/release/api/reactor/core/publisher/Mono.html#subscriberContext--查看文档： https://projectreactor.io/docs/core/release/api/reactor/core/publisher/Mono.html#subscriberContext--
声明:本站的技术帖子网页，遵循CC BY-SA 4.0协议，如果您需要转载，请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.