[#2720]Improve ReactiveLoadBalancerClientFilter implementation so tha…

…t we can pass attribute to ReactiveLoadBalancer (#2721)
spring-cloud · Mar 8, 2024 · e6d237c · e6d237c
1 parent 688391a
commit e6d237c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.../main/java/org/springframework/cloud/gateway/filter/ReactiveLoadBalancerClientFilter.java b/.../main/java/org/springframework/cloud/gateway/filter/ReactiveLoadBalancerClientFilter.java
@@ -114,7 +114,7 @@ public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
 				.getSupportedLifecycleProcessors(clientFactory.getInstances(serviceId, LoadBalancerLifecycle.class),
 						RequestDataContext.class, ResponseData.class, ServiceInstance.class);
 		DefaultRequest<RequestDataContext> lbRequest = new DefaultRequest<>(
-				new RequestDataContext(new RequestData(exchange.getRequest()), getHint(serviceId)));
+				new RequestDataContext(new RequestData(exchange.getRequest(), exchange.getAttributes()), getHint(serviceId)));
 		return choose(lbRequest, serviceId, supportedLifecycleProcessors).doOnNext(response -> {
 
 			if (!response.hasServer()) {
@@ -154,7 +154,7 @@ public Mono<Void> filter(ServerWebExchange exchange, GatewayFilterChain chain) {
 						.onComplete(new CompletionContext<ResponseData, ServiceInstance, RequestDataContext>(
 								CompletionContext.Status.SUCCESS, lbRequest,
 								exchange.getAttribute(GATEWAY_LOADBALANCER_RESPONSE_ATTR),
-								new ResponseData(exchange.getResponse(), new RequestData(exchange.getRequest()))))));
+								new ResponseData(exchange.getResponse(), new RequestData(exchange.getRequest(), exchange.getAttributes()))))));
 	}
 
 	protected URI reconstructURI(ServiceInstance serviceInstance, URI original) {