Add ServerHttpRequestContext. (#857)

spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/ServerHttpRequestContext.java

@@ -0,0 +1,44 @@
+/*
+ * Copyright 2012-2020 the original author or authors.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.springframework.cloud.client.loadbalancer;
+
+import org.springframework.http.HttpMethod;
+import org.springframework.http.server.reactive.ServerHttpRequest;
+
+/**
+ * @author Olga Maciaszek-Sharma
+ * @since 3.0.0
+ */
+public class ServerHttpRequestContext extends DefaultRequestContext {
+
+	public ServerHttpRequestContext(ServerHttpRequest serverHttpRequest) {
+		this(serverHttpRequest, "default");
+	}
+
+	public ServerHttpRequestContext(ServerHttpRequest serverHttpRequest, String hint) {
+		super(serverHttpRequest, hint);
+	}
+
+	public ServerHttpRequest getClientRequest() {
+		return (ServerHttpRequest) super.getClientRequest();
+	}
+
+	public HttpMethod method() {
+		return ((ServerHttpRequest) super.getClientRequest()).getMethod();
+	}
+
+}

spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/reactive/ReactorLoadBalancerExchangeFilterFunction.java

@@ -107,8 +107,8 @@ public class ReactorLoadBalancerExchangeFilterFunction implements LoadBalancedEx
 							lifecycle -> lifecycle.onComplete(new CompletionContext<ClientResponse, ServiceInstance>(
 									CompletionContext.Status.FAILED, throwable, lbResponse))))
 					.doOnSuccess(clientResponse -> supportedLifecycleProcessors.forEach(
-							lifecycle -> lifecycle.onComplete(new CompletionContext<ClientResponse, ServiceInstance>(
-									CompletionContext.Status.SUCCESS, lbResponse, clientResponse))));
+							lifecycle -> lifecycle.onComplete(new CompletionContext<>(CompletionContext.Status.SUCCESS,
+									lbResponse, clientResponse))));
 		});
 	}