Ribbon负载均衡机制及自定义规则

Spring Cloud Ribbon是一个基于HTTP 和TCP的客户端负载均衡工具，他基于NetflixRibbon实现。通过SpringCloud的封装，可以让我们轻松的将面向服务的Rest模板请求自动转换成客户端负载均衡的服务调用。
负载均衡是对系统的高可用、网络压力的环节和处理能力扩容的重要手段之一。我们常说的负载均衡都指的是服务端负载均衡，其中分为硬件负载均衡和软件负载均衡。硬件负载均衡主要通过在服务器节点之间安装专门用于复杂均衡的设备，比如F5。软件负载均衡则是通过在服务器上安装一些具有负载均衡功能或模板的软件来完成请求分发工作，比如Nginx。
服务提供者只需要启动多个服务实例并注册到一个注册中心或是多个相关的服务注册中心。
服务消费者直接通过调用被@LoadBalanced注解修饰过的RestTemplate来实现面向服务的接口调用。

一个负载均衡器，至少提供以下功能：
要维护各个服务器的IP等信息
根据特定逻辑选取服务器
为了实现基本的负载均衡功能，Ribbon的负载均衡器由三大子模块
Rule
Ping
ServerList

自定义的规则类：

public class MyRule implements IRule{
	private ILoadBalancer lb;

	public Server choose(Object key) {
		Random r = new Random();
		int rNum = r.nextInt(10);
		
		List<Server> servers = lb.getAllServers();
		
		if(rNum > 7) {
			return getServerByPort(servers, 8085);
		}
		return getServerByPort(servers, 8080);
	}
	
	private Server getServerByPort(List<Server> servers, int port) {
		for(Server s : servers) {
			if(s.getPort() == port) {
				return s;
			}
		}
		return null;
	}

	public void setLoadBalancer(ILoadBalancer lb) {
		this.lb = lb;
	}

	public ILoadBalancer getLoadBalancer() {
		return this.lb;
	}

	
}

测试类：

public class LBMain {
	public static void main(String[] args) {
		BaseLoadBalancer lb=new BaseLoadBalancer();		
		MyRule rule=new MyRule();
		rule.setLoadBalancer(lb);
		lb.setRule(rule);
		List<Server>servers=new ArrayList<Server>();
		servers.add(new Server("localhost",8080));
		servers.add(new Server("localhost",8085));
		lb.addServers(servers);
		for(int i=0;i<10;i++){
			Server s=lb.chooseServer(null);
			System.out.println(s);
		
		}
	}

}

更改上篇文章中客户端的类，使其使用自定义规则类：

public class TestRibbon {
	public static void main(String[] args) throws Exception{
		ConfigurationManager.getConfigInstance().setProperty("my-client.ribbon.listOfServers", "localhost:8080,localhost:8085");
		ConfigurationManager.getConfigInstance().setProperty(
	      		"my-client.ribbon.NFLoadBalancerRuleClassName", MyRule.class.getName());
		RestClient client=(RestClient) ClientFactory.getNamedClient("my-client");
		HttpRequest request=HttpRequest.newBuilder().uri("/person").build();
		for(int i=0;i<10;i++){
			HttpResponse response=client.executeWithLoadBalancer(request);
			String json=response.getEntity(String.class);
			System.out.println(json);
		}
	}

}