springcloud 负载均衡优化

Author: xlorne

transaction-dubbo/pom.xml

@@ -74,7 +74,7 @@
         <dependency>
             <groupId>com.alibaba</groupId>
             <artifactId>dubbo</artifactId>
-            <version>2.5.3</version>
+            <version>2.5.7</version>
             <exclusions>
                 <exclusion>
                     <groupId>org.springframework</groupId>

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

@@ -1,11 +1,13 @@
 package com.codingapi.ribbon.loadbalancer;
 
+import com.codingapi.tx.aop.bean.TxTransactionLocal;
+import com.lorne.core.framework.utils.encode.MD5Util;
+import com.netflix.loadbalancer.Server;
+import org.apache.commons.lang.StringUtils;
 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;
+import java.util.List;
 
 /**
  * created by foxdd 2017-12-05
@@ -14,31 +16,71 @@ 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());
+	public Server proxy(List<Server> servers,Server server){
+
 		TxTransactionLocal txTransactionLocal = TxTransactionLocal.current();
-		if(txTransactionLocal == null){
+		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);
+
+		try{
+			logger.info("LCNBalanceProxy - > start");
+
+			String groupId = txTransactionLocal.getGroupId();
+
+			//取出组件的appName
+			String appName = server.getMetaInfo().getAppName();
+
+
+			String key = MD5Util.md5((groupId + "_" + appName).getBytes());
+
+			Server oldServer =getServer(txTransactionLocal,servers,key);
+			if(oldServer == null){
+				logger.info("The server of key:" + key + " has not been cached yet!");
+				return server;
+			}
+
+			putServer(key, txTransactionLocal, server);
+			logger.info("LCNBalanceProxy - > load new server ");
+
 			return server;
-		} else{
-			logger.info("The cached server info, host:" + cachedServer.getHost() + ", port:" + cachedServer.getPort());
-			return cachedServer;
+		}finally {
+			logger.info("LCNBalanceProxy - > end");
+		}
+	}
+
+
+
+	private void putServer(String key,TxTransactionLocal txTransactionLocal,Server server){
+		String serviceName =  server.getMetaInfo().getAppName();
+		String address = server.getHostPort();
+
+		String md5 = MD5Util.md5((address+serviceName).getBytes());
+
+		logger.info("putServer->address->"+address+",md5-->"+md5);
+
+		txTransactionLocal.putLoadBalance(key,md5);
+	}
+
+
+	private Server getServer(TxTransactionLocal txTransactionLocal, List<Server> servers, String key){
+		String val = txTransactionLocal.getLoadBalance(key);
+		if(StringUtils.isEmpty(val)){
+			return null;
+		}
+		for(Server server:servers){
+			String serviceName =  server.getMetaInfo().getAppName();
+			String address = server.getHostPort();
+
+			String md5 = MD5Util.md5((address+serviceName).getBytes());
+
+			logger.info("getServer->address->"+address+",md5-->"+md5);
+
+			if(val.equals(md5)){
+				return server;
+			}
 		}
-		
+		return null;
 	}
 
 }

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

@@ -22,7 +22,7 @@ public LcnNoOpLoadBalancerProxy(){
 	@Override
 	public Server chooseServer(Object key){
 		logger.info("enter chooseServer method, key:" + key);
-		return lcnLoadBalancerRule.proxy(super.chooseServer(key));
+		return lcnLoadBalancerRule.proxy(getAllServers(),super.chooseServer(key));
 	}
 
 }

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

@@ -30,7 +30,7 @@ public LcnZoneAwareLoadBalancerProxy(IClientConfig clientConfig, IRule rule,
 	@Override
 	public Server chooseServer(Object key){
 		logger.info("enter chooseServer method, key:" + key);
-		return lcnLoadBalancerRule.proxy(super.chooseServer(key));
+		return lcnLoadBalancerRule.proxy(getAllServers(),super.chooseServer(key));
 	}
 
 }

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

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