SpringCloud(四)Ribbon自定义负载均衡

上篇文章我们已经完成了Ribbon负载均衡的功能。做法很简单，只需要在RestTemplate添加@LoanBalanced 的注解。默认情况下，Ribbon的负载均衡策略是RoundRobbin(轮训)的方式，可很多时候在特定场景下需要不同的策略，这个时候就需要自定义Ribbon策略了。看下面代码：

[java]  view plain  copy

1. package com.zhuyang.cloud.controller;  
2. import org.springframework.beans.factory.annotation.Autowired;  
3. import org.springframework.cloud.client.loadbalancer.LoadBalanced;  
4. import org.springframework.cloud.netflix.ribbon.RibbonClient;  
5. import org.springframework.context.annotation.Bean;  
6. import org.springframework.web.bind.annotation.PathVariable;  
7. import org.springframework.web.bind.annotation.RequestMapping;  
8. import org.springframework.web.bind.annotation.RequestMethod;  
9. import org.springframework.web.bind.annotation.RestController;  
10. import org.springframework.web.client.RestTemplate;  
11. import com.zhuyang.cloud.entity.User;  
12. import com.zhuyang.config.RibbonConfiguration;  
13. @RestController  
14. @RibbonClient(name = "microservice-provider", configuration = RibbonConfiguration.class)//name是provider的服务名  <span style="font-family: Arial, Helvetica, sans-serif;">RibbonConfiguration为自定义配置</span>  
15. public class MovieController {  
16.     @Bean  
17.     @LoadBalanced  
18.     public RestTemplate restTemplate() { // equals to RestTemplate  
19.         // restTemplate=new RestTemplate();  
20.         return new RestTemplate();  
21.     }  
22.     @Autowired  
23.     private RestTemplate restTemplate;  
24.     @RequestMapping(value = "/movie/{id}", method = RequestMethod.GET)  
25.     public User findById(@PathVariable Long id) {  
26.         // return restTemplate.getForEntity("http://localhost:8000/service/"+id,  
27.         // User.class).getBody();  
28.         return restTemplate.getForEntity("http://microservice-provider/provider/service/" + id, User.class).getBody();  
29.     }  
30. }  

[html]  view plain  copy

1. server:   
2.   port: 8001   
3. eureka:   
4.   client:   
5.     serviceUrl:   
6.       defaultZone: http://user:[email protected]:8761/eureka/    # 指定注册中心的地址   
7.   instance:   
8.     preferIpAddress: true   
9. spring:   
10.   application:   
11.     name: microservice-consumer   
12. microservice-provider:  ##config ribbon   
13.   ribbon:   
14.     eureka:   
15.       enabled: false   
16.     listOfServers: localhost:8000, localhost:8002,localhost:8003 ##假设provider有3台instance端口分别是8000 8002 8003  
17.     ServerListRefreshInterval: 15000  

[java]  view plain  copy

1. package com.zhuyang.config;  
2. import org.springframework.beans.factory.annotation.Autowired;  
3. import org.springframework.context.annotation.Bean;  
4. import com.netflix.client.config.IClientConfig;  
5. import com.netflix.loadbalancer.IPing;  
6. import com.netflix.loadbalancer.IRule;  
7. import com.netflix.loadbalancer.PingUrl;  
8. import com.netflix.loadbalancer.ZoneAvoidanceRule;  
9. public class RibbonConfiguration {  
10.     @Autowired  
11.     private IClientConfig ribbonClientConfig;  
12.     @Bean  
13.     public IPing ribbonPing(IClientConfig config) {  
14.         // ping url will try to access http://microservice-provider/provider/ to  
15.         // see if reponse code is 200 . check PingUrl.isAlive()  
16.         // param /provider/ is the context-path of provider service  
17.         return new PingUrl(false, "/provider/");  
18.     }  
19.     @Bean  
20.     public IRule ribbonRule(IClientConfig config) {  
21.         // return new AvailabilityFilteringRule();  
22.          return new RandomRule();//  
23.         // return new BestAvailableRule();  
24.         // return new RoundRobinRule();//轮询  
25.         // return new WeightedResponseTimeRule();  
26.         // return new RetryRule();  
27.         // return new ZoneAvoidanceRule();  
28.     }  
29. }  

在RibbonConfiguration中的ribbonRule方法就是用来定义不用的策略，每种策略所对应的实现类和描述 都已经添加了注释。例如我们返回的是RandomRule策略，那么我们在多次请求provider的时候就不再是轮训的方式进行命中，而是随机方式。。下面是RandomRule的代码实现

[java]  view plain  copy

1. public Server choose(ILoadBalancer lb, Object key) {  
2.         if (lb == null) {  
3.             return null;  
4.         }  
5.         Server server = null;  
6.         while (server == null) {  
7.             if (Thread.interrupted()) {  
8.                 return null;  
9.             }  
10.             List<Server> upList = lb.getReachableServers();//get all reachable server .list  listOfServers: localhost:8000, localhost:8002,localhost:8003  
11.             List<Server> allList = lb.getAllServers();  
12.             int serverCount = allList.size();  
13.             if (serverCount == 0) {  
14.                 return null;  
15.             }  
16.             int index = rand.nextInt(serverCount);//get random index  
17.             server = upList.get(index);//get specified server eg:<span style="font-family: Arial, Helvetica, sans-serif;">localhost:8000</span>  
18.             if (server == null) {  
19.                 Thread.yield();  
20.                 continue;  
21.             }  
22.             if (server.isAlive()) {  
23.                 return (server);  
24.             }  
25.             // Shouldn't actually happen.. but must be transient or a bug.  
26.             server = null;  
27.             Thread.yield();  
28.         }  
29.         return server;//return selected server  
30.     }