首先来看一下Ribbon的作用:
ribbon 是一个客户端负载均衡器，主要是实现类似于 nginx的负载均衡的功能。
SpringCloud提供了7种负载均衡的算法，如下表所示，ribbon 默认使用轮询算法(RoundRobinRule)来实现负载均衡，
| 策略名 | 策略声明 | 策略描述 | 实现说明 |
| BestAvailableRule | public class BestAvailableRule extends ClientConfigEnabledRoundRobinRule | 选择一个最小的并发请求的server | 逐个考察Server，如果Server被tripped了，则忽略，在选择其中ActiveRequestsCount最小的server |
| AvailabilityFilteringRule | public class AvailabilityFilteringRule extends PredicateBasedRule | 过滤掉那些因为一直连接失败的被标记为circuit tripped的后端server，并过滤掉那些高并发的的后端server（active connections 超过配置的阈值） | 使用一个AvailabilityPredicate来包含过滤server的逻辑，其实就就是检查status里记录的各个server的运行状态 |
| WeightedResponseTimeRule | public class WeightedResponseTimeRule extends RoundRobinRule | 根据响应时间分配一个weight，响应时间越长，weight越小，被选中的可能性越低。 | 一个后台线程定期的从status里面读取评价响应时间，为每个server计算一个weight。Weight的计算也比较简单responsetime 减去每个server自己平均的responsetim |
| RetryRule | public class RetryRule extends AbstractLoadBalancerRule | 对选定的负载均衡策略机上重试机制。 | 在一个配置时间段内当选择server不成功，则一直尝试使用subRule的方式选择一个可用的server |
| RoundRobinRule | public class RoundRobinRule extends AbstractLoadBalancerRule | roundRobin方式轮询选择server | 轮询index，选择index对应位置的server |
| RandomRule | public class RandomRule extends AbstractLoadBalancerRule | 随机选择一个server | 在index上随机，选择index对应位置的server |
| ZoneAvoidanceRule | public class ZoneAvoidanceRule extends PredicateBasedRule | 复合判断server所在区域的性能和server的可用性选择server | 使用ZoneAvoidancePredicate和AvailabilityPredicate来判断是否选择某个server，前一个判断判定一个zone的运行性能是否可用，剔除不可用的zone（的所有server） |
下面我们通过代码来实现Ribbon的负载均衡功能；不同于dubbo的rpc调用，SpringCloud采用的rest方式进行服务调用，创建一个RemoteController类来模拟调用：
@RestController
@RequestMapping("/remote")
public class RemoteController {
private static final String URL_PREFIX = "http://SERVICE01";
@Autowired
private RestTemplate restTemplate;
@GetMapping("/getService")
public Object getRemoteService() {
return restTemplate.getForObject(URL_PREFIX + "/local/getServiceInfo", Object.class);
}
}其中的RestTemplate类需要手动的加载到IOC容器中，在启动类中添加如下的代码:
@Bean
@LoadBalanced
public RestTemplate getRestTemplate() {
return new RestTemplate();
}接着我们创建service01的多个实例:
通过postman模拟请求，分别会调用这四个服务中的一个，并且调用的服务顺序是有序的。
那么如何调用其他的6种负载均衡算法呢?
很简单，我们只要在启动类种添加如下的代码即可:
@Bean
public IRule getRule(){
return new RandomRule();
}这时候可能又有一个疑问了，这个负载均衡的算法只能使用SpringCloud提供的7种算法吗？如果想要自己定义一个算法又该如何实现呢？
假如这边有一个比较变态的需求，我只要service06给我提供服务那怎么操作呢?
依然简单，我们可以新建一个配置类:
@Configuration
public class CustomRule {
@Bean
public IRule getRule() {
return new SingleRule();
}
}
自定义负载均衡算法:
public class SingleRule extends AbstractLoadBalancerRule {
public Server choose(ILoadBalancer lb, Object key) {
if (lb == null) {
return null;
}
Server server = null;
while (server == null) {
if (Thread.interrupted()) {
return null;
}
List<Server> upList = lb.getReachableServers();
List<Server> allList = lb.getAllServers();
int serverCount = allList.size();
if (serverCount == 0) {
return null;
}
server = upList.get(0);
if (server == null) {
Thread.yield();
continue;
}
if (server.isAlive()) {
return (server);
}
// Shouldn't actually happen.. but must be transient or a bug.
server = null;
Thread.yield();
}
return server;
}
@Override
public Server choose(Object o) {
return choose(getLoadBalancer(), o);
}
@Override
public void initWithNiwsConfig(IClientConfig iClientConfig) {
}
}
这里有一点需要注意:
也就是说，这个配置类不能和之前的启动类放到启动类的同一个包下面，不然所有的ribbon客户端都会采用这个自定义的负载均衡算法。
至此，一个自定义的负载均衡算法就实现啦，当然实际中估计没有人会这么用，具体的算法还是需要依据不同的业务来确定。