SpringCloud Ribbon（二）之自定义负载均衡策略IRule

小咪咪 2022-10-29 11:19 320阅读 0赞

### 一、Ribbon负载均衡策略 ###

一个服务对应一个LoadBalancer，一个LoadBalancer只有一个Rule，LoadBalancer记录服务的注册地址，Rule提供从服务的注册地址中找出一个地址的规则。

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21vYWt1bg_size_16_color_FFFFFF_t_70][]

Ribbon提供七种负载均衡策略，默认的负载均衡策略是轮训策略。

<table> 
 <tbody> 
  <tr> 
   <td>名称</td> 
   <td>解释</td> 
  </tr> 
  <tr> 
   <td>RoundRobinRule</td> 
   <td>轮训策略</td> 
  </tr> 
  <tr> 
   <td>RandomRule</td> 
   <td>随机策略</td> 
  </tr> 
  <tr> 
   <td>BestAvailableRule</td> 
   <td>过滤出故障服务器后，选择一个并发量最小的</td> 
  </tr> 
  <tr> 
   <td>WeightedResponseTimeRule</td> 
   <td>针对响应时间加权轮询</td> 
  </tr> 
  <tr> 
   <td>AvailabilityFilteringRule</td> 
   <td>可用过滤策略，先过滤出故障的或并发请求大于阈值的一部分服务实例，然后再以线性轮询的方式从过滤后的实例清单中选出一个;</td> 
  </tr> 
  <tr> 
   <td>ZoneAvoidanceRule</td> 
   <td>从最佳区域实例集合中选择一个最优性能的服务实例</td> 
  </tr> 
  <tr> 
   <td>RetryRule</td> 
   <td>选择一个Server，如果失败，重新选择一个Server重试</td> 
  </tr> 
 </tbody> 
</table>

### 二、 自定义负载均衡策略 ###

本文自定义负载均衡规则提供根据指定URI优先转发到特定的机器的功能。

### 三、实现代码 ###

（1）负载均衡策略实现类

public class MicroServiceLoadBalanceRule extends AbstractLoadBalancerRule {
    
        @Override
        public void initWithNiwsConfig(IClientConfig iClientConfig) {
    
        }
    
        @Override
        public Server choose(Object key) {
    
            ILoadBalancer lb = getLoadBalancer();
            if (lb == null) {
                return null;
            }
    
            List<Server> servers = lb.getReachableServers();
            if (servers.isEmpty()){
                return null;
            }
    
            RequestContext ctx = RequestContext.getCurrentContext();
            HttpServletRequest request = ctx.getRequest();
            String uri = request.getRequestURI();
            boolean specialApi = uri.startsWith("/api") ;
            // /api路径优先转发到指定的机器
            if(specialApi){
                String ip = "192.168.2.3";
                Server chooseServer = null;
                for (Server server : servers) {
                    if (server.getHost().equals(ip)) {
                        chooseServer = server;
                    }
                }
    
                if(chooseServer == null){
                    chooseServer = servers.stream().findFirst().orElse(null);
                }
                return chooseServer;
            }
    
            return servers.get(new Random().nextInt(servers.size()));
        }
    }

（2）application.yaml文件配置

micro-service:
      ribbon:
        NFLoadBalancerRuleClassName: com.mk.springcloud.gateway.loadbalance.MicroServiceLoadBalanceRule

（3） 代码配置

public class MicroRibbonConfig {
    
        @Bean
        public IRule microIRule(){
    
            return new MicroServiceLoadBalanceRule();
        }
    
    }
    
    @RibbonClient(name = "micro-service", configuration = MicroRibbonConfig.class)
    public class RibbonClientConfig {
    
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L21vYWt1bg_size_16_color_FFFFFF_t_70]: /images/20221024/3999bd7b2760400786392b956ea45959.png