Apache Thrift 初学小讲(七)【负载均衡】

本节在《代理》和《连接池》的基础上加上负载均衡,主要的思路是在代理上加上集群服务的n个连接池,客户端连接到代理上,代理转发时根据负载均衡算法选择一个合适连接池中的连接:

 

Apache Thrift 初学小讲(七)【负载均衡】_第1张图片
 

一般的负载均衡算法有:

1 [加权]轮询法([Weight] Round Robin),

2 [加权]随机法([Weight] Random),

3 源地址哈希法(Hash),

4 [加权]最小连接数法([Weight] Least Connections)。

 

这里只实现加权最小连接数法:least_conn算法很简单,首选遍历后端集群,比较每个后端的conns/weight,选取该值最小的后端。如果有多个后端的conns/weight值同为最小的,那么对它们采用加权轮询算法。

 

首先在ConnectionPoolFactory.java加上计算该连接池的权重值的方法:

//取得权重值
public double getWeight() {
	//活动的连接数 除以 最大连接数
	double numActive = pool.getNumActive(); 
	double maxTotal = pool.getMaxTotal();
	System.out.println(numActive + "**" + maxTotal); 
	return numActive / maxTotal;
}

 

然后在ThirftProxy.java加上如下代码,其中getLeastConnPoolFactory就是least_conn算法的实现:

private static List poolFactorys = new ArrayList();
	
static {
	GenericObjectPoolConfig config = new GenericObjectPoolConfig();
	//链接池中最大连接数,默认为8
	config.setMaxTotal(2); 
	//当连接池资源耗尽时,调用者最大阻塞的时间,超时将跑出异常。单位:毫秒;默认为-1,表示永不超时
	config.setMaxWaitMillis(3000);
	ConnectionPoolFactory poolFactory = new ConnectionPoolFactory(config, "127.0.0.1", 9090); 
	poolFactorys.add(poolFactory);
	poolFactory = new ConnectionPoolFactory(config, "127.0.0.1", 9091);
	poolFactorys.add(poolFactory);
}

public static ConnectionPoolFactory getLeastConnPoolFactory() {
	ConnectionPoolFactory selectPoolFactory = poolFactorys.get(0);
	double selectWeight = selectPoolFactory.getWeight();
	
	for (int i = 1; i < poolFactorys.size(); i++) {
		ConnectionPoolFactory poolFactory = poolFactorys.get(i);
		double weight = poolFactory.getWeight();
		System.out.println(selectWeight + "--" + weight);
		//这里为了简单,没有实现:如果有多个后端的conns/weight值同为最小的,那么对它们采用加权轮询算法
		if ( weight < selectWeight ) {
			selectPoolFactory = poolFactory;
			selectWeight = weight;
		}
	}
	
	return selectPoolFactory;
}

 

最后在ProxyProcessor的process要转发的时候用getLeastConnPoolFactory取得连接:

//转发请求
ConnectionPoolFactory poolFactory = getLeastConnPoolFactory();
TTransport transport = poolFactory.getConnection();

 

工程结构图:

Apache Thrift 初学小讲(七)【负载均衡】_第2张图片
 

启动两个服务TestServer,端口分别是9090和9091,并发的执行TestClient,结果如预期。

 

几种简单的负载均衡算法及其Java代码实现:http://www.cnblogs.com/szlbm/p/5588555.html

 

附件是整个eclipse工程。

你可能感兴趣的:(thrift)