微服务架构之均衡组件 Ribbon解析：LoadBalancerClient实现原理解析

LoadBalancerClient

通过之前两篇文章的讲述，相信读者已经对Ribbon与RestTemplate和Feign之间的工作原理有了一定的了解。接下来，本小节会对LoadBalancerClient进行负载均衡的具体原理和实现。

LoadBalancerClient是Ribbon项目的核心类之一，可以在RestTemplate发送网络请求时替代RestTemplate进行网络调用。

/**
 * 客户端负载均衡器的接口
 */
public interface LoadBalancerClient extends ServiceInstanceChooser {

	/**
	 * 从serviceId所代表的服务列表中选择一个服务器来发送网络请求
	 */
	<T> T execute(String serviceId, LoadBalancerRequest<T> request) throws IOException;

	/**
     * 同上。
	 */
	<T> T execute(String serviceId, ServiceInstance serviceInstance, LoadBalancerRequest<T> request) throws IOException;

	/**
     * 构建网络请求URI
	 */
	URI reconstructURI(ServiceInstance instance, URI original);
}

LoadBalancerClient接口继承了ServiceInstanceChooser来
从服务器列表中依据负载均衡策略选出一个服务器。

//实现该类来选择一个服务器来发送请求。
public interface ServiceInstanceChooser {
    /**
     * 根据serviceId从服务器列表中选择一个ServiceInstance
     **/
    ServiceInstance choose(String serviceId);
}

RibbonLoadBalancerClient是LoadBalancerClient的实现类之一，它的execute会首先使用ILoadBalancer来选择Server，然后将Server封装成RibbonServer对象，最后再调用LoadBalancerRequest的apply函数进行网络请求的处理。

@Override
public <T> T execute(String serviceId, LoadBalancerRequest<T> request) throws IOException {
    //每次发送请求都会创建一个ILoadBalancer,会涉及到选举规则(IRule)，服务器列表集群(ServerList)和检验服务是否存在(IPing)等细节实现
    ILoadBalancer loadBalancer = getLoadBalancer(serviceId);
    //选举一个具体的server，会涉及到选举规则，服务器列表和服务器是否存在等问题。
    Server server = getServer(loadBalancer);
    if (server == null) {
        throw new IllegalStateException("No instances available for " + serviceId);
    }
    RibbonServer ribbonServer = new RibbonServer(serviceId, server, isSecure(server,
            serviceId), serverIntrospector(serviceId).getMetadata(server));

    return execute(serviceId, ribbonServer, request);
}

ILoadBalancer的初始化过程，getLoadBalancer函数直接调用了SpringClientFactory的getLoadBalancer函数。SpringClientFactory是NamedContextFactory的实现类，关于NamedContextFactory的机制我们在Feign章节中已经详细讲解过了，通过它可以实现多套Bean实例的管理。

public ILoadBalancer getLoadBalancer(String name) {
		return getInstance(name, ILoadBalancer.class);
}

getServer则是直接调用了ILoadBalancer的chooseServer函数来使用负载均衡策略从已知的服务器列表中选出一个服务器。

protected Server getServer(ILoadBalancer loadBalancer) {
    if (loadBalancer == null) {
        return null;
    }
    return loadBalancer.chooseServer("default");
}

execute函数中使用LoadBalancerRequest实例的apply函数，将之前根据负载均衡策略选择出来的服务器作为参数传递进去，进行真正的HTTP请求的发送。

public <T> T execute(String serviceId, ServiceInstance serviceInstance, LoadBalancerRequest<T> request) throws IOException {
    Server server = null;
    if(serviceInstance instanceof RibbonServer) {
        server = ((RibbonServer)serviceInstance).getServer();
    }

    RibbonLoadBalancerContext context = this.clientFactory
            .getLoadBalancerContext(serviceId);
    RibbonStatsRecorder statsRecorder = new RibbonStatsRecorder(context, server);

    try {
        T returnVal = request.apply(serviceInstance);
        statsRecorder.recordStats(returnVal);
        return returnVal;
    }
    catch (IOException ex) {
        statsRecorder.recordStats(ex);
        throw ex;
    }
    catch (Exception ex) {
        statsRecorder.recordStats(ex);
        ReflectionUtils.rethrowRuntimeException(ex);
    }
    return null;
}

LoadBalancerRequest的apply函数的具体实现不再详细讲解，因为Ribbon最为重要的部分就是使用负载均衡策略来选择服务器，也就是ILoadBalancer的chooseServer函数的实现，在接下来的文章里对其进行详细的讲解。