SpringCloud/spring-cloud-commons
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

In LoadBalancerInterceptor add ServiceInstance to static ThreadLocal and provide static getter.

Browse Source
This commit is contained in:
Spencer Gibb
2015-03-25 09:53:48 -06:00
parent 9e9e30fd70
commit 56fbc5e970
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
spring-cloud-commons/src/main/java/org/springframework/cloud/client/loadbalancer/LoadBalancerInterceptor.java
View File

@@ -34,6 +34,8 @@ public class LoadBalancerInterceptor implements ClientHttpRequestInterceptor {
private LoadBalancerClient loadBalancer;
private static final ThreadLocal<ServiceInstance> instanceHolder = new ThreadLocal<>();
public LoadBalancerInterceptor(LoadBalancerClient loadBalancer) {
this.loadBalancer = loadBalancer;
}
@@ -43,12 +45,14 @@ public class LoadBalancerInterceptor implements ClientHttpRequestInterceptor {
final ClientHttpRequestExecution execution) throws IOException {
final URI originalUri = request.getURI();
String serviceName = originalUri.getHost();
request.getHeaders().add("X-SpringCloud-Service-Id", serviceName);
return this.loadBalancer.execute(serviceName,
new LoadBalancerRequest<ClientHttpResponse>() {
@Override
public ClientHttpResponse apply(final ServiceInstance instance)
throws Exception {
instanceHolder.set(instance);
HttpRequest serviceRequest = new ServiceRequestWrapper(request,
instance);
return execution.execute(serviceRequest, body);
@@ -75,4 +79,8 @@ public class LoadBalancerInterceptor implements ClientHttpRequestInterceptor {
}
public static ServiceInstance getThreadLocalServiceInstance() {
return instanceHolder.get();
}
}