From bc8f8f0bf976ae0b1a0156ccc0ba81622e601709 Mon Sep 17 00:00:00 2001 From: Spencer Gibb Date: Wed, 18 Dec 2019 22:59:14 -0500 Subject: [PATCH] Adds missing method implementations --- .../support/DefaultClientResponse.java | 19 +++++++++++++++++++ .../support/DefaultServerResponse.java | 5 +++++ 2 files changed, 24 insertions(+) diff --git a/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultClientResponse.java b/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultClientResponse.java index 29819529..fd4140ee 100644 --- a/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultClientResponse.java +++ b/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultClientResponse.java @@ -40,6 +40,7 @@ import org.springframework.web.reactive.function.BodyExtractor; import org.springframework.web.reactive.function.BodyExtractors; import org.springframework.web.reactive.function.client.ClientResponse; import org.springframework.web.reactive.function.client.ExchangeStrategies; +import org.springframework.web.reactive.function.client.WebClientResponseException; /** * Default implementation of {@link ClientResponse}. @@ -120,6 +121,24 @@ public class DefaultClientResponse implements ClientResponse { } } + @Override + public Mono releaseBody() { + return body(BodyExtractors.toDataBuffers()) + .map(DataBufferUtils::release) + .then(); + } + + @Override + public Mono> toBodilessEntity() { + return releaseBody() + .then(toEntityInternal(Mono.empty())); + } + + @Override + public Mono createException() { + throw new UnsupportedOperationException(); + } + @SuppressWarnings("unchecked") private Mono consumeAndCancel() { return (Mono) this.response.getBody().map(buffer -> { diff --git a/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultServerResponse.java b/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultServerResponse.java index 84ea7382..d9f175ad 100644 --- a/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultServerResponse.java +++ b/spring-cloud-gateway-core/src/main/java/org/springframework/cloud/gateway/support/DefaultServerResponse.java @@ -64,6 +64,11 @@ public class DefaultServerResponse implements ServerResponse { return HttpStatus.valueOf(response().getStatusCode().value()); } + @Override + public int rawStatusCode() { + return response().getStatusCode().value(); + } + @Override public final HttpHeaders headers() { return response().getHeaders();