Ribbon自定义负载均衡策略

骑猪看日落 2021-10-18 07:24 302阅读 0赞

### 搭建注册中心和服务 ###

参考：[搭建注册中心][Link 1]\`  
[服务提供者创建][Link 2]  
[Ribbon消费者服务创建][Ribbon]

### 自定义负载均衡 ###

在ribbon项目配置文件类RibbonConfig简单修改即可：

1 创建类MyRule 实现接口 IRule 实现方法

public class MyRule implements IRule {
        ILoadBalancer balancer = new BaseLoadBalancer();
        @Override
        public Server choose(Object o) {
            List<Server> allServers = balancer.getAllServers();
            //返回8092端口服务器
            return findServer(allServers,8092);
        }
    
        private Server findServer(List<Server> allServers, int port) {
            for (Server server : allServers) {
                if (server.getPort() == port) {
                    return server;
                }
            }
            System.out.println("NULL port="+port);
            return null;
        }
    
        @Override
        public void setLoadBalancer(ILoadBalancer iLoadBalancer) {
            this.balancer = iLoadBalancer;
    
        }
    
        @Override
        public ILoadBalancer getLoadBalancer() {
            return balancer;
        }
    }

自己定义了只调用 localhost:8092服务提供者。

2 修改配置文件类RibbonConfig：

@Configuration
    public class RibbonConfig {
        @Bean
        @LoadBalanced
        public RestTemplate restTemplate() {
            return new RestTemplate();
        }
    
        @Bean
        public IRule ribbonRandomRule() {
            return new MyRule();
        }
    }

在返回Rule的时候使用自定义的MyRule。

启动所有项目测试一下：访问路径localhost:8084/test  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTQ0OTI2NjU_size_16_color_FFFFFF_t_70]  
刷新页面也是返回结果 success 92 说明自定义负载均衡策略生效。

[Link 1]: https://blog.csdn.net/u014492665/article/details/99084429
[Link 2]: https://blog.csdn.net/u014492665/article/details/99086434
[Ribbon]: https://blog.csdn.net/u014492665/article/details/99087843
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3UwMTQ0OTI2NjU_size_16_color_FFFFFF_t_70]: /images/20211018/ec3c053773f040eaa3a83df1614ee664.png