From 5a3c2b73644c614208437d6c6f02dffaebfb98d5 Mon Sep 17 00:00:00 2001
From: Olga Maciaszek-Sharma <olga.maciaszek@gmail.com>
Date: Tue, 1 Dec 2020 10:08:33 -0600
Subject: [PATCH] Add ServerHttpRequestContext. (#857)

---
 .../ServerHttpRequestContext.java             | 44 +++++++++++++++++++
 ...torLoadBalancerExchangeFilterFunction.java |  4 +-
 2 files changed, 46 insertions(+), 2 deletions(-)
 create mode 100644 spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/ServerHttpRequestContext.java

diff --git a/spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/ServerHttpRequestContext.java b/spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/ServerHttpRequestContext.java
new file mode 100644
index 00000000..99f0d2be
--- /dev/null
+++ b/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();
+	}
+
+}
diff --git a/spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/reactive/ReactorLoadBalancerExchangeFilterFunction.java b/spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/reactive/ReactorLoadBalancerExchangeFilterFunction.java
index 6b2c903f..ec2e150b 100644
--- a/spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/reactive/ReactorLoadBalancerExchangeFilterFunction.java
+++ b/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))));
 		});
 	}