SpringCloud学习总结（三）：Ribbon

什么是Ribbon

Ribbon是Netflix发布的开源顶目，主要功能是提供客户端的软件负载均衡算法，将Netflix的中间层服务连接在一起。Ribbon客户端组件提供一系列完善配置项如连接超时，重试等。简单的说，就是在配置文件中列出LoadBalancer〔简称LB）后面所有的机器，Ribbon会自动的帮助你基于某种规则（筒单轮洵，随机连接等）去连接这些机器。我们也很容易使Ribbon实现自定义负载均衡算法.

启动两个服务实例

参考SpringCloud学习总结（二）：Eureka入门及详解，启动两个user-service实例：
具体代码在SpringCloud学习总结（二）中创建服务提供者并注册服务到Eureka下

开启负载均衡

在服务消费方中spring提供的rest模板中开启，因为Eureka中已经集成了Ribbon，所以我们无需引入新的依赖。具体服务消费者代码参考SpringCloud学习总结（二）中从Eureka获取服务下
在RestTemplate的配置方法上添加@LoadBalanced注解：

@SpringBootApplication
@EnableDiscoveryClient // 开启Eureka客户端
public class ServerConsumerApplication {

	@Bean
	@LoadBalanced
	public RestTemplate restTemplate() {
		// 这次我们使用了OkHttp客户端,只需要注入工厂即可
		return new RestTemplate(new OkHttp3ClientHttpRequestFactory());
	}

	public static void main(String[] args) {
		SpringApplication.run(ServerConsumerApplication.class, args);
	}
}

之前我们在SpringCloud学习总结（二）中调用服务使用discoveryClient，手动获取ip和端口，
现在我们不用，我们直接通过服务名称调用：

@Component
public class UserDao {

    @Autowired
    private RestTemplate restTemplate;

   /* public User queryUserById(Long id){
        String url = "http://localhost:8081/user/" + id;
        return this.restTemplate.getForObject(url, User.class);
    }*/

    @Autowired
    private DiscoveryClient discoveryClient; // eureka客户端，可以获取到eureka中服务的信息

/*    public User queryUserById(Long id){
        // 根据服务名称，获取服务实例。有可能是集群，所以是service实例集合
        List<ServiceInstance> instances = discoveryClient.getInstances("user-service");
        // 因为只有一个UserService。所以获取第一个实例
        ServiceInstance instance = instances.get(0);
        // 获取ip和端口信息，拼接成服务地址
        String baseUrl = "http://" + instance.getHost() + ":" + instance.getPort() + "/user/" + id;
        return this.restTemplate.getForObject(baseUrl, User.class);
    }*/

    public List<User> queryUserByIds(List<Long> ids) {
        List<User> users = new ArrayList<>();
        // 地址直接写服务名称即可
        String baseUrl = "http://user-service/user/";
        ids.forEach(id -> {
            // 我们测试多次查询，
            users.add(this.restTemplate.getForObject(baseUrl + id, User.class));
            // 每次间隔500毫秒
            try {
                Thread.sleep(500);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        });
        return users;
    }
}

访问页面，查看结果：访问成功

源码跟踪

为什么我们只输入了service名称就可以访问了呢？之前还要获取ip和端口。
显然有人帮我们根据service名称，获取到了服务实例的ip和端口。它就是LoadBalancerInterceptor
代码上打断点：

一路源码跟踪：RestTemplate.getForObject --> RestTemplate.execute --> RestTemplate.doExecute：

点击进入AbstractClientHttpRequest.execute --> AbstractBufferingClientHttpRequest.executeInternal --> InterceptingClientHttpRequest.executeInternal --> InterceptingClientHttpRequest.execute:

继续跟入：LoadBalancerInterceptor.intercept方法

继续跟入execute方法：发现获取了8082端口的服务

再跟下一次，发现获取的是8081：

负载均衡策略

Ribbon默认的负载均衡策略是简单的轮询，我们可以测试一下：
编写测试类，在刚才的源码中我们看到拦截中是使用RibbonLoadBalanceClient来进行负载均衡的，其中有一个choose方法，找到choose方法的接口方法，是这样介绍的：

这个就是负载均衡获取实例的方法。
我们对注入这个类的对象，然后对其测试：

@RunWith(SpringRunner.class)
@SpringBootTest(classes = ServerConsumerApplication.class)
public class LoadBalanceTest {

    @Autowired
    private RibbonLoadBalancerClient client;

    @Test
    public void testLoadBalance(){
        for (int i = 0; i < 100; i++) {
            ServiceInstance instance = this.client.choose("user-service");
            System.out.println(instance.getHost() + ":" +instance.getPort());
        }
    }
}

最后的结果确实是轮询。

如何修改负载均衡的策略

查看源码：

这个rule是谁：

这里的rule默认值是一个RoundRobinRule，看类的介绍：

就是轮询的意思

我们注意到，这个类其实是实现了接口IRule的，查看一下：定义负载均衡的规则接口。

有以下实现：

SpringBoot也帮我们提供了修改负载均衡规则的配置入口，在server-consumer的application.yml中添加如下配置：

user-service:
  ribbon:
    NFLoadBalancerRuleClassName: com.netflix.loadbalancer.RandomRule

格式是：{服务名称}.ribbon.NFLoadBalancerRuleClassName，值就是IRule的实现类
这样负载均衡策略就变成随机了