Merge branch '5.1.x'

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/ClientResponse.java

@@ -36,6 +36,7 @@ import org.springframework.http.ResponseEntity;
 import org.springframework.http.client.reactive.ClientHttpResponse;
 import org.springframework.http.codec.HttpMessageReader;
 import org.springframework.http.codec.HttpMessageWriter;
+import org.springframework.lang.Nullable;
 import org.springframework.util.MultiValueMap;
 import org.springframework.web.reactive.function.BodyExtractor;
 
@@ -63,8 +64,9 @@ public interface ClientResponse {
 	 * Return the status code of this response.
 	 * @return the status as an HttpStatus enum value
 	 * @throws IllegalArgumentException in case of an unknown HTTP status code
-	 * @see HttpStatus#valueOf(int)
+	 * @see HttpStatus#resolve(int)
 	 */
+	@Nullable
 	HttpStatus statusCode();
 
 	/**
@@ -203,6 +205,17 @@ public interface ClientResponse {
 		return new DefaultClientResponseBuilder(strategies).statusCode(statusCode);
 	}
 
+	/**
+	 * Create a response builder with the given raw status code and strategies for reading the body.
+	 * @param statusCode the status code
+	 * @param strategies the strategies
+	 * @return the created builder
+	 * @since 5.1.9
+	 */
+	static Builder create(int statusCode, ExchangeStrategies strategies) {
+		return new DefaultClientResponseBuilder(strategies).rawStatusCode(statusCode);
+	}
+
 	/**
 	 * Create a response builder with the given status code and message body readers.
 	 * @param statusCode the status code
@@ -268,6 +281,14 @@ public interface ClientResponse {
 		 */
 		Builder statusCode(HttpStatus statusCode);
 
+		/**
+		 * Set the raw status code of the response.
+		 * @param statusCode the new status code.
+		 * @return this builder
+		 * @since 5.1.9
+		 */
+		Builder rawStatusCode(int statusCode);
+
 		/**
 		 * Add the given header value(s) under the given name.
 		 * @param headerName  the header name

spring-webflux/src/main/java/org/springframework/web/reactive/function/client/DefaultClientResponseBuilder.java

@@ -67,7 +67,7 @@ final class DefaultClientResponseBuilder implements ClientResponse.Builder {
 
 	private ExchangeStrategies strategies;
 
-	private HttpStatus statusCode = HttpStatus.OK;
+	private int statusCode = 200;
 
 	private final HttpHeaders headers = new HttpHeaders();
 
@@ -87,7 +87,7 @@ final class DefaultClientResponseBuilder implements ClientResponse.Builder {
 	public DefaultClientResponseBuilder(ClientResponse other) {
 		Assert.notNull(other, "ClientResponse must not be null");
 		this.strategies = other.strategies();
-		statusCode(other.statusCode());
+		this.statusCode = other.rawStatusCode();
 		headers(headers -> headers.addAll(other.headers().asHttpHeaders()));
 		cookies(cookies -> cookies.addAll(other.cookies()));
 		if (other instanceof DefaultClientResponse) {
@@ -101,7 +101,12 @@ final class DefaultClientResponseBuilder implements ClientResponse.Builder {
 
 	@Override
 	public DefaultClientResponseBuilder statusCode(HttpStatus statusCode) {
-		Assert.notNull(statusCode, "HttpStatus must not be null");
+		return rawStatusCode(statusCode.value());
+	}
+
+	@Override
+	public DefaultClientResponseBuilder rawStatusCode(int statusCode) {
+		Assert.isTrue(statusCode >= 100 && statusCode < 600, "StatusCode must be between 1xx and 5xx");
 		this.statusCode = statusCode;
 		return this;
 	}
@@ -179,7 +184,7 @@ final class DefaultClientResponseBuilder implements ClientResponse.Builder {
 
 	private static class BuiltClientHttpResponse implements ClientHttpResponse {
 
-		private final HttpStatus statusCode;
+		private final int statusCode;
 
 		private final HttpHeaders headers;
 
@@ -187,7 +192,7 @@ final class DefaultClientResponseBuilder implements ClientResponse.Builder {
 
 		private final Flux<DataBuffer> body;
 
-		public BuiltClientHttpResponse(HttpStatus statusCode, HttpHeaders headers,
+		public BuiltClientHttpResponse(int statusCode, HttpHeaders headers,
 				MultiValueMap<String, ResponseCookie> cookies, Flux<DataBuffer> body) {
 
 			this.statusCode = statusCode;
@@ -198,12 +203,12 @@ final class DefaultClientResponseBuilder implements ClientResponse.Builder {
 
 		@Override
 		public HttpStatus getStatusCode() {
-			return this.statusCode;
+			return HttpStatus.resolve(this.statusCode);
 		}
 
 		@Override
 		public int getRawStatusCode() {
-			return this.statusCode.value();
+			return this.statusCode;
 		}
 
 		@Override

spring-webflux/src/test/java/org/springframework/web/reactive/function/client/DefaultClientResponseBuilderTests.java

@@ -100,5 +100,18 @@ public class DefaultClientResponseBuilderTests {
 				.verifyComplete();
 	}
 
+	@Test
+	public void fromCustomStatus() {
+
+		ClientResponse other = ClientResponse.create(499, ExchangeStrategies.withDefaults())
+				.build();
+
+		ClientResponse result = ClientResponse.from(other)
+				.build();
+
+		assertThat(result.rawStatusCode()).isEqualTo(499);
+		assertThat(result.statusCode()).isNull();
+	}
+
 
 }