Merge pull request #18 from codingapi/tx-lcn-foxdd

Tx lcn foxdd

Author: xlorne

transaction-springcloud/src/main/java/com/codingapi/ribbon/loadbalancer/LcnLoadBalancerRule.java

@@ -0,0 +1,44 @@
+package com.codingapi.ribbon.loadbalancer;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.codingapi.tx.Constants;
+import com.codingapi.tx.aop.bean.TxTransactionLocal;
+import com.netflix.loadbalancer.Server;
+
+/**
+ * created by foxdd 2017-12-05
+ */
+public class LcnLoadBalancerRule {
+	
+	private Logger logger = LoggerFactory.getLogger(LcnLoadBalancerRule.class);
+	
+	public Server proxy(Server server){
+		logger.info("LCNloadBalancer proxy -> map-size -> " + Constants.cacheModelInfo.size());
+		logger.info("The selected server info, host:" + server.getHost() + ", port:" + server.getPort());
+		TxTransactionLocal txTransactionLocal = TxTransactionLocal.current();
+		if(txTransactionLocal == null){
+			return server;
+		}
+		
+		String groupId = txTransactionLocal.getGroupId();
+
+		//取出组件的appName
+		String appName = server.getMetaInfo().getAppName();
+		
+		String key = groupId + "_" + appName;
+		
+		Server cachedServer = (Server) Constants.cacheModelInfo.get(key);
+		if(cachedServer == null){
+			logger.info("The server of key:" + key + " has not been cached yet!");
+			Constants.cacheModelInfo.put(key, server);
+			return server;
+		} else{
+			logger.info("The cached server info, host:" + cachedServer.getHost() + ", port:" + cachedServer.getPort());
+			return cachedServer;
+		}
+		
+	}
+	
+}

transaction-springcloud/src/main/java/com/codingapi/ribbon/loadbalancer/LcnNoOpLoadBalancerProxy.java

@@ -0,0 +1,28 @@
+package com.codingapi.ribbon.loadbalancer;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.netflix.loadbalancer.NoOpLoadBalancer;
+import com.netflix.loadbalancer.Server;
+
+/**
+ * created by foxdd 2017-12-05
+ */
+public class LcnNoOpLoadBalancerProxy extends NoOpLoadBalancer {
+	
+	private Logger logger = LoggerFactory.getLogger(LcnNoOpLoadBalancerProxy.class);
+	
+	LcnLoadBalancerRule lcnLoadBalancerRule = new LcnLoadBalancerRule();
+	
+	public LcnNoOpLoadBalancerProxy(){
+		super();
+	}
+
+	@Override
+	public Server chooseServer(Object key){
+		logger.info("enter chooseServer method, key:" + key);
+		return lcnLoadBalancerRule.proxy(super.chooseServer(key));
+	}
+
+}

transaction-springcloud/src/main/java/com/codingapi/ribbon/loadbalancer/LcnRibbonConfiguration.java

@@ -0,0 +1,41 @@
+package com.codingapi.ribbon.loadbalancer;
+
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+import com.netflix.client.config.IClientConfig;
+import com.netflix.loadbalancer.ILoadBalancer;
+import com.netflix.loadbalancer.IPing;
+import com.netflix.loadbalancer.IRule;
+import com.netflix.loadbalancer.Server;
+import com.netflix.loadbalancer.ServerList;
+import com.netflix.loadbalancer.ServerListFilter;
+import com.netflix.loadbalancer.ServerListUpdater;
+
+@Configuration
+public class LcnRibbonConfiguration {
+
+	/**
+	 * 为ribbon的loadbalancer做代理，相比于重写IRULE,重写loadbalancer更有利于用户自选LB算法，而且有默认LB算法可用
+	 * @param config
+	 * @param serverList
+	 * @param serverListFilter
+	 * @param rule
+	 * @param ping
+	 * @param serverListUpdater
+	 * @return
+	 */
+	@Bean
+	public ILoadBalancer ribbonLoadBalancer(IClientConfig config,
+			ServerList<Server> serverList, ServerListFilter<Server> serverListFilter,
+			IRule rule, IPing ping, ServerListUpdater serverListUpdater) {
+		return new LcnZoneAwareLoadBalancerProxy<>(config, rule, ping, serverList,
+				serverListFilter, serverListUpdater);
+	}
+	
+	/*@Bean
+	public IRule ribbonRule(){
+		return new LcnRandomRuleProxy();
+	}*/
+
+}

transaction-springcloud/src/main/java/com/codingapi/ribbon/loadbalancer/LcnZoneAwareLoadBalancerProxy.java

@@ -0,0 +1,36 @@
+package com.codingapi.ribbon.loadbalancer;
+
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import com.netflix.client.config.IClientConfig;
+import com.netflix.loadbalancer.IPing;
+import com.netflix.loadbalancer.IRule;
+import com.netflix.loadbalancer.Server;
+import com.netflix.loadbalancer.ServerList;
+import com.netflix.loadbalancer.ServerListFilter;
+import com.netflix.loadbalancer.ServerListUpdater;
+import com.netflix.loadbalancer.ZoneAwareLoadBalancer;
+
+/**
+ * created by foxdd 2017-12-05
+ */
+public class LcnZoneAwareLoadBalancerProxy<T extends Server> extends ZoneAwareLoadBalancer<T> {
+	
+	private Logger logger = LoggerFactory.getLogger(LcnZoneAwareLoadBalancerProxy.class);
+	
+	LcnLoadBalancerRule lcnLoadBalancerRule = new LcnLoadBalancerRule();
+	
+	public LcnZoneAwareLoadBalancerProxy(IClientConfig clientConfig, IRule rule,
+            IPing ping, ServerList<T> serverList, ServerListFilter<T> filter,
+            ServerListUpdater serverListUpdater) {
+		super(clientConfig, rule, ping, serverList, filter, serverListUpdater);
+	}
+
+	@Override
+	public Server chooseServer(Object key){
+		logger.info("enter chooseServer method, key:" + key);
+		return lcnLoadBalancerRule.proxy(super.chooseServer(key));
+	}
+
+}

transaction-springcloud/src/main/java/com/codingapi/tx/TransactionConfiguration.java

@@ -1,14 +1,18 @@
 package com.codingapi.tx;
 
+import org.springframework.cloud.netflix.ribbon.RibbonClients;
 import org.springframework.context.annotation.ComponentScan;
 import org.springframework.context.annotation.Configuration;
 
+import com.codingapi.ribbon.loadbalancer.LcnRibbonConfiguration;
+
 /**
  * Created by lorne on 2017/6/26.
  */
 
 @Configuration
 @ComponentScan
+@RibbonClients(defaultConfiguration=LcnRibbonConfiguration.class)
 public class TransactionConfiguration {
 
 

tx-client/src/main/java/com/codingapi/tx/Constants.java

@@ -1,5 +1,8 @@
 package com.codingapi.tx;
 
+import java.util.Map;
+import java.util.concurrent.ConcurrentHashMap;
+
 import com.codingapi.tx.model.TxServer;
 
 /**
@@ -25,6 +28,11 @@ public class Constants {
      * 模块唯一标示
      */
     public static String uniqueKey;
+    
+    /**
+     * 用于优化ribbon负载
+     */
+    public static Map<String, Object> cacheModelInfo = new ConcurrentHashMap<>();
 
 
 }