成人在线亚洲_国产日韩视频一区二区三区_久久久国产精品_99国内精品久久久久久久

您的位置:首頁技術(shù)文章
文章詳情頁

如何在Spring WebFlux的任何地方獲取Request對象

瀏覽:98日期:2023-07-25 11:51:14
1 不一樣的世界

在常規(guī)的Spring Web項(xiàng)目中,我們要獲取Request對象是非常方便的,不少庫都提供了靜態(tài)方法來獲取。獲取代碼如下:

ServletRequestAttributes requestAttributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes();// get the requestHttpServletRequest request = requestAttributes.getRequest();

在類RequestContextHolder提供了靜態(tài)方法,也就意味著你可以在任何地方調(diào)用。而它使用了ThreadLocal來保存Request對象,也就是不同線程是可以獲取各自的Request對象。

但在響應(yīng)式WebFlux的世界里,并沒有提供類似的Holder類,而WebFlux是無法感知線程的,任何一個(gè)線程可以在任何時(shí)候處理任何請求,如果它覺得切換當(dāng)前線程更有效率,它就會(huì)這么做。但在Servlet Based的應(yīng)用里,它會(huì)為某個(gè)請求安排一個(gè)線程去處理完整個(gè)過程。

這個(gè)巨大的差別,意味著不能簡單地通過ThreadLocal來保存和獲取Request了。

2 先保存,再獲取

為了在后面可以方便獲得Request對象,我們就需要在開始的時(shí)候把它存在一個(gè)可以使用、并且是相同scope的容器里。這里需要解決兩個(gè)關(guān)鍵問題:

(1)Request對象從何而來;

(2)存在哪里?

針對問題(1), 我們可以回想什么時(shí)候會(huì)出現(xiàn)Request對象,最容易想得到的就是WebFilter了,它的方法簽名如下:

public Mono<Void> filter(ServerWebExchange exchange, WebFilterChain chain);

我們可以通過ServerWebExchange直接獲取到Request對象:

ServerHttpRequest request = exchange.getRequest();

而因?yàn)镕ilter是可以先于應(yīng)用邏輯執(zhí)行的,所以滿足要求,問題(1)解決。

針對問題(2),需要一個(gè)與Reavtive請求相同范圍的容器,reactor.util.context.Context可以滿足需求。查看reactor的官方文檔(https://projectreactor.io/docs/core/release/reference/#context )可見下面這段話:

Since version 3.1.0, Reactor comes with an advanced feature that is somewhat comparable to ThreadLocal but can be applied to a Flux or a Mono instead of a Thread. This feature is called Context.

并且官網(wǎng)也給出了為何ThreadLocal在某些場景不適用的解釋,有興趣可以看看。

3 代碼實(shí)現(xiàn)3.1 WebFilter獲取并保存

首先,在WebFilter中獲取Request對象并保存,代碼如下:

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

從ServerWebExchange中獲取到ServerHttpRequest對象,再通過put方法把它放進(jìn)Context里。

3.2 工具類Holder

實(shí)現(xiàn)一個(gè)工具類來提供靜態(tài)方法,在Filter后的任何場景都可以使用:

public class ReactiveRequestContextHolder { public static final Class<ServerHttpRequest> CONTEXT_KEY = ServerHttpRequest.class; public static Mono<ServerHttpRequest> getRequest() { return Mono.subscriberContext().map(ctx -> ctx.get(CONTEXT_KEY)); }}3.3 在Controller中使用

我們嘗試在Controller中使用ReactiveRequestContextHolder來獲取Request:

@RestControllerpublic class GetRequestController { @RequestMapping('/request') public Mono<String> getRequest() { return ReactiveRequestContextHolder.getRequest().map(request -> request.getHeaders().getFirst('user')); }}

上面方法獲取了Request對象,然后再獲取了Request中的Header。

啟動(dòng)應(yīng)用,測試如下:

$ curl http://localhost:8088/request -H ’user: pkslow’pkslow$ curl http://localhost:8088/request -H ’user: larry’larry$ curl http://localhost:8088/request -H ’user: www.pkslow.com’www.pkslow.com

可以成功獲取請求頭user。

4 總結(jié)

代碼請查看:https://github.com/LarryDpk/pkslow-samples

以上就是如何在Spring WebFlux的任何地方獲取Request對象的詳細(xì)內(nèi)容,更多關(guān)于Spring WebFlux獲取Request對象的資料請關(guān)注好吧啦網(wǎng)其它相關(guān)文章!

標(biāo)簽: Spring
相關(guān)文章: