Merge pull request #686 from flystar32/master

fix loadbalance feignclient seata info

spring-cloud-alibaba-seata/src/main/java/org/springframework/cloud/alibaba/seata/feign/SeataFeignClient.java

@@ -35,49 +35,44 @@ import org.springframework.util.StringUtils;
  */
 public class SeataFeignClient implements Client {
 
-    private final Client delegate;
-    private final BeanFactory beanFactory;
-    private static final int MAP_SIZE = 16;
+	private final Client delegate;
+	private final BeanFactory beanFactory;
+	private static final int MAP_SIZE = 16;
 
-    SeataFeignClient(BeanFactory beanFactory) {
-        this.beanFactory = beanFactory;
-        this.delegate = new Client.Default(null, null);
-    }
+	SeataFeignClient(BeanFactory beanFactory) {
+		this.beanFactory = beanFactory;
+		this.delegate = new Client.Default(null, null);
+	}
 
-    SeataFeignClient(BeanFactory beanFactory, Client delegate) {
-        this.delegate = delegate;
-        this.beanFactory = beanFactory;
-    }
+	SeataFeignClient(BeanFactory beanFactory, Client delegate) {
+		this.delegate = delegate;
+		this.beanFactory = beanFactory;
+	}
 
-    @Override
-    public Response execute(Request request, Request.Options options) throws IOException {
+	@Override
+	public Response execute(Request request, Request.Options options) throws IOException {
 
-        Request modifiedRequest = getModifyRequest(request);
+		Request modifiedRequest = getModifyRequest(request);
+		return this.delegate.execute(modifiedRequest, options);
+	}
 
-        try {
-            return this.delegate.execute(modifiedRequest, options);
-        } finally {
+	private Request getModifyRequest(Request request) {
 
-        }
-    }
+		String xid = RootContext.getXID();
 
-    private Request getModifyRequest(Request request) {
+		if (StringUtils.isEmpty(xid)) {
+			return request;
+		}
 
-        String xid = RootContext.getXID();
+		Map<String, Collection<String>> headers = new HashMap<>(MAP_SIZE);
+		headers.putAll(request.headers());
 
-        if (StringUtils.isEmpty(xid)) {
-            return request;
-        }
+		List<String> fescarXid = new ArrayList<>();
+		fescarXid.add(xid);
+		headers.put(RootContext.KEY_XID, fescarXid);
 
-        Map<String, Collection<String>> headers = new HashMap<>(MAP_SIZE);
-        headers.putAll(request.headers());
-
-        List<String> fescarXid = new ArrayList<>();
-        fescarXid.add(xid);
-        headers.put(RootContext.KEY_XID, fescarXid);
-
-        return Request.create(request.method(), request.url(), headers, request.body(),
-            request.charset());
-    }
+		return Request.create(request.method(), request.url(), headers, request.body(),
+				request.charset());
+	}
 
 }

spring-cloud-alibaba-seata/src/main/java/org/springframework/cloud/alibaba/seata/feign/SeataFeignClientAutoConfiguration.java

@@ -65,7 +65,7 @@ public class SeataFeignClientAutoConfiguration {
 	protected static class FeignBeanPostProcessorConfiguration {
 
 		@Bean
-        SeataBeanPostProcessor seataBeanPostProcessor(
+		SeataBeanPostProcessor seataBeanPostProcessor(
 				SeataFeignObjectWrapper seataFeignObjectWrapper) {
 			return new SeataBeanPostProcessor(seataFeignObjectWrapper);
 		}

spring-cloud-alibaba-seata/src/main/java/org/springframework/cloud/alibaba/seata/feign/SeataFeignContext.java

@@ -32,7 +32,7 @@ public class SeataFeignContext extends FeignContext {
 	private final FeignContext delegate;
 
 	SeataFeignContext(SeataFeignObjectWrapper seataFeignObjectWrapper,
-					  FeignContext delegate) {
+			FeignContext delegate) {
 		this.seataFeignObjectWrapper = seataFeignObjectWrapper;
 		this.delegate = delegate;
 	}

spring-cloud-alibaba-seata/src/main/java/org/springframework/cloud/alibaba/seata/feign/SeataLoadBalancerFeignClient.java

@@ -17,7 +17,13 @@
 package org.springframework.cloud.alibaba.seata.feign;
 
 import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
+import io.seata.core.context.RootContext;
 import org.springframework.beans.factory.BeanFactory;
 import org.springframework.cloud.netflix.ribbon.SpringClientFactory;
 
@@ -26,23 +32,27 @@ import feign.Request;
 import feign.Response;
 import org.springframework.cloud.openfeign.ribbon.CachingSpringLoadBalancerFactory;
 import org.springframework.cloud.openfeign.ribbon.LoadBalancerFeignClient;
+import org.springframework.util.StringUtils;
 
 /**
  * @author xiaojing
  */
 public class SeataLoadBalancerFeignClient extends LoadBalancerFeignClient {
 
+	private static final int MAP_SIZE = 16;
+
 	private final BeanFactory beanFactory;
 
 	SeataLoadBalancerFeignClient(Client delegate,
-								 CachingSpringLoadBalancerFactory lbClientFactory,
-								 SpringClientFactory clientFactory, BeanFactory beanFactory) {
+			CachingSpringLoadBalancerFactory lbClientFactory,
+			SpringClientFactory clientFactory, BeanFactory beanFactory) {
 		super(wrap(delegate, beanFactory), lbClientFactory, clientFactory);
 		this.beanFactory = beanFactory;
 	}
 
 	@Override
 	public Response execute(Request request, Request.Options options) throws IOException {
+		Request modifiedRequest = getModifyRequest(request);
 		return super.execute(request, options);
 	}
 
@@ -50,4 +60,22 @@ public class SeataLoadBalancerFeignClient extends LoadBalancerFeignClient {
 		return (Client) new SeataFeignObjectWrapper(beanFactory).wrap(delegate);
 	}
 
+	private Request getModifyRequest(Request request) {
+
+		String xid = RootContext.getXID();
+
+		if (StringUtils.isEmpty(xid)) {
+			return request;
+		}
+
+		Map<String, Collection<String>> headers = new HashMap<>(MAP_SIZE);
+		headers.putAll(request.headers());
+
+		List<String> fescarXid = new ArrayList<>();
+		fescarXid.add(xid);
+		headers.put(RootContext.KEY_XID, fescarXid);
+
+		return Request.create(request.method(), request.url(), headers, request.body(),
+				request.charset());
+	}
 }