SpringCloud/spring-cloud-gateway
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Short circuit caching request body if already done.

Browse Source 
Fixes gh-2969
This commit is contained in:
sgibb
2024-03-20 23:00:27 -04:00
parent c79e9ad2b3
commit cc62626997
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
spring-cloud-gateway-server/src/main/java/org/springframework/cloud/gateway/support/ServerWebExchangeUtils.java
View File

@@ -361,6 +361,14 @@ public final class ServerWebExchangeUtils {
*/
private static <T> Mono<T> cacheRequestBody(ServerWebExchange exchange, boolean cacheDecoratedRequest,
Function<ServerHttpRequest, Mono<T>> function) {
// don't cache if body is already cached
Object cachedDataBuffer = exchange.getAttribute(CACHED_REQUEST_BODY_ATTR);
if (cachedDataBuffer instanceof DataBuffer) {
if (log.isTraceEnabled()) {
log.trace("body already in exchange attribute, short circuiting");
}
return Mono.just(exchange.getRequest()).flatMap(function);
}
ServerHttpResponse response = exchange.getResponse();
DataBufferFactory factory = response.bufferFactory();
// Join all the DataBuffers so we have a single DataBuffer for the body