解決Spring Cloud Gateway獲取body內(nèi)容,不影響GET請(qǐng)求的操作
廢話
這幾天換了新工作,需要重新開發(fā)一套系統(tǒng),技術(shù)選用Spring Cloud。在對(duì)接終端接口的時(shí)候要做驗(yàn)簽,就涉及到在網(wǎng)關(guān)做攔截器,然后取出BODY里面的數(shù)據(jù)。
網(wǎng)上找了幾個(gè)方法,有的拿不到數(shù)據(jù),有的拿到數(shù)據(jù)之后不支持GET請(qǐng)求了。沒有一個(gè)合理的解決辦法,最后想到在動(dòng)態(tài)路由構(gòu)建的時(shí)候可以指定METHOD,于是有了如下解決辦法
解決
@Bean
public RouteLocator vmRouteLocator(RouteLocatorBuilder builder) {
return builder.routes()
.route(r -> r.method(HttpMethod.POST).and()
.readBody(Object.class, requestBody -> {
//相當(dāng)于緩存了body信息,在filter 中可以這么獲取 exchange.getAttribute("cachedRequestBodyObject");
log.info("requestBody is {}", requestBody);
return true;
})
.and().path("/terminal/**")
.filters(f -> f.filter(terminalSignFilter()))
.uri("lb://TERMINAL-SERVICE")
.order(0)
.id("terminal-service")
)
.route(r -> r.method(HttpMethod.GET).and()
.path("/terminal/**")
.filters(f -> f.filter(terminalSignFilter()))
.uri("lb://TERMINAL-SERVICE")
.order(1)
.id("terminal-service")
)
.build();
}
關(guān)鍵代碼:
r.method(HttpMethod.POST)
r.method(HttpMethod.GET)
分別指定了不同請(qǐng)求METHOD對(duì)應(yīng)的路由策略
在POST請(qǐng)求中需要緩存BODY信息,在Filter中便可以獲取到
GET請(qǐng)求因?yàn)闆]有BODY,所以如果不指定GET的路由便會(huì)報(bào)錯(cuò)
可能會(huì)有更通用的方法,但是目前只想到這么多,以后有好的解決辦法會(huì)繼續(xù)更新
補(bǔ)充知識(shí):Spring Cloud Gateway 2.x 獲取body中的數(shù)據(jù)并緩存在請(qǐng)求中
場(chǎng)景
因?yàn)閔ttp請(qǐng)求中的body,讀取過一次后就無法重新再讀,但是我們希望網(wǎng)關(guān)項(xiàng)目中可以在所有filter中共享body中的內(nèi)容。
寫法
import lombok.extern.slf4j.Slf4j;
import org.springframework.cloud.gateway.filter.GatewayFilterChain;
import org.springframework.cloud.gateway.filter.GlobalFilter;
import org.springframework.core.Ordered;
import org.springframework.core.io.buffer.DataBuffer;
import org.springframework.core.io.buffer.DataBufferUtils;
import org.springframework.http.server.reactive.ServerHttpRequest;
import org.springframework.http.server.reactive.ServerHttpRequestDecorator;
import org.springframework.stereotype.Component;
import org.springframework.web.reactive.function.server.HandlerStrategies;
import org.springframework.web.reactive.function.server.ServerRequest;
import org.springframework.web.server.ServerWebExchange;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
@Component
@Slf4j
public class CacheBodyParamsFilter implements GlobalFilter, Ordered {
@Override
public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
if (ParamsUtil.logBody(exchange)) {
return DataBufferUtils.join(exchange.getRequest().getBody())
.flatMap(dataBuffer -> {
byte[] bytes = new byte[dataBuffer.readableByteCount()];
dataBuffer.read(bytes);
DataBufferUtils.release(dataBuffer);
Flux<DataBuffer> cachedFlux = Flux.defer(() -> {
DataBuffer buffer = exchange.getResponse().bufferFactory().wrap(bytes);
DataBufferUtils.retain(buffer);
return Mono.just(buffer);
});
ServerHttpRequest mutatedRequest = new ServerHttpRequestDecorator(exchange.getRequest()) {
@Override
public Flux<DataBuffer> getBody() {
return cachedFlux;
}
};
ServerWebExchange mutatedExchange = exchange.mutate().request(mutatedRequest).build();
return ServerRequest.create(mutatedExchange, HandlerStrategies.withDefaults().messageReaders())
.bodyToMono(String.class)
.doOnNext(objectValue -> {
//在此處,將body中的params值獲取到,并存放在本次請(qǐng)求的attributes屬性中,這樣就可以在本次請(qǐng)求中的所有地方進(jìn)行使用了
mutatedExchange.getAttributes().put(CommonConstant.PARAMS, ParamsUtil.buildParams(mutatedRequest,objectValue));
}).then(chain.filter(mutatedExchange));
});
}
return chain.filter(exchange);
}
@Override
public int getOrder() {
return Ordered.HIGHEST_PRECEDENCE;
}
}
以上這篇解決Spring Cloud Gateway獲取body內(nèi)容,不影響GET請(qǐng)求的操作就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Java畢業(yè)設(shè)計(jì)實(shí)戰(zhàn)之仿小米電子產(chǎn)品售賣商城系統(tǒng)的實(shí)現(xiàn)
這是一個(gè)使用了java+SpringBoot+Vue+MySQL+Redis+ElementUI開發(fā)的仿小米商城系統(tǒng),是一個(gè)畢業(yè)設(shè)計(jì)的實(shí)戰(zhàn)練習(xí),具有小米商城該有的所有基礎(chǔ)功能,感興趣的朋友快來看看吧2022-01-01
詳解Spring Cloud Gateway 數(shù)據(jù)庫(kù)存儲(chǔ)路由信息的擴(kuò)展方案
這篇文章主要介紹了詳解Spring Cloud Gateway 數(shù)據(jù)庫(kù)存儲(chǔ)路由信息的擴(kuò)展方案,小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧2018-11-11
Java微信二次開發(fā)(一) Java微信請(qǐng)求驗(yàn)證功能
這篇文章主要為大家詳細(xì)介紹了Java微信二次開發(fā)第一篇,Java微信請(qǐng)求驗(yàn)證功能,具有一定的參考價(jià)值,感興趣的小伙伴們可以參考一下2017-04-04
IntelliJ IDEA 2020.3 重大特性(新功能一覽)
這篇文章主要介紹了IntelliJ IDEA 2020.3 重大特性(新功能一覽),本文通過圖文并茂的形式給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-12-12

