自定义Ribbon规则--客户端写法

ゝ一纸荒年。 2022-10-13 15:58 170阅读 0赞

### 自定义Ribbon规则--客户端写法 ###

*   *  代码结构
 *  controller的代码
 *   *  自定义配置文件
     *  ==自定义的规则==:
 *  application.yml
 *   *  效果如下:

## 代码结构 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70]

# controller的代码 #

package com.demo.springcloud.controller;
    
    import com.demo.springcloud.pojo.Dept;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.web.bind.annotation.*;
    import org.springframework.web.client.RestTemplate;
    
    import java.util.List;
    
    @RestController
    public class DeptConsumerController { 
    
        @Autowired
        private RestTemplate restTemplate;
    
    // private static final String REST_URI_PREFIX = "http://localhost:8001";
        private static final String REST_URI_PREFIX = "http://springcloud-provider-dept";
    
    
        @GetMapping("/consumer/dept/queryAll")
        public List<Dept> queryAll(){ 
            return restTemplate.getForObject(REST_URI_PREFIX+"/dept/queryAll", List.class);
        }
    
    }

springcloud-provider-dep是服务提供者的id:  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 1]

## 自定义配置文件 ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 2]

## 自定义的规则: ##

//
    // Source code recreated from a .class file by IntelliJ IDEA
    // (powered by FernFlower decompiler)
    //
    package com.demo.springcloud.myRule;
    import com.netflix.client.config.IClientConfig;
    import com.netflix.loadbalancer.AbstractLoadBalancerRule;
    import com.netflix.loadbalancer.ILoadBalancer;
    import com.netflix.loadbalancer.Server;
    import java.util.List;
    
    /** * 自定义的规则: * 每个服务走5次，然后才切换到另一个服务上 */
    public class HflRandomRule extends AbstractLoadBalancerRule { 
    
        private int total;
        private int curentIndex;
        public HflRandomRule() { 
        }
    
    // @SuppressWarnings({"RCN_REDUNDANT_NULLCHECK_OF_NULL_VALUE"})
        public Server choose(ILoadBalancer lb, Object key) { 
            if (lb == null) { 
                return null;
            } else { 
                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;
                    }
    
                    if(total<5){ 
                        server = upList.get(curentIndex);
                    }else{ 
                        total = 0;
                        curentIndex ++;
                        if(curentIndex > upList.size()){ 
                            curentIndex = 0;
                        }
                        server = upList.get(curentIndex);
                    }
                    total++;
    
    
    
                    if (server == null) { 
                        Thread.yield();
                    } else { 
                        if (server.isAlive()) { 
                            return server;
                        }
    
                        server = null;
                        Thread.yield();
                    }
                }
    
                return server;
            }
        }
    
    
        public Server choose(Object key) { 
            return this.choose(this.getLoadBalancer(), key);
        }
    
        public void initWithNiwsConfig(IClientConfig clientConfig) { 
        }
    }

# application.yml #

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 3]

## 效果如下: ##

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 4]

--------------------

完

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70]: /images/20221005/446044eaa2ad4cdfbf7b549cb253ca4c.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 1]: /images/20221005/26092fb57fd1448cbfa5f3a606a389bb.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 2]: /images/20221005/18adb1124fe24ffa9ed20c72c9fa4144.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 3]: /images/20221005/a58c587b1f50495a8822f8da5b379dcf.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2JhaWR1XzIxMzQ5NjM1_size_16_color_FFFFFF_t_70 4]: /images/20221005/66ded96dd25e44e9bbd050a608b5fe50.png