【Ribbon篇四】自定义负载均衡策略（4）

官方文档特别指出：自定义的负载均衡配置类不能放在 @componentScan 所扫描的当前包下及其子包下，否则我们自定义的这个配置类就会被所有的Ribbon客户端所共享，也就是说我们达不到特殊化定制的目的了。

要求自定义的算法：依旧是轮询策略，但是每个服务器被调用5次后轮到下一个服务，即以前是每个服务被调用1次，现在是每个被调用5次。

1. 打开microservicecloud-consumer-dept-80工程，自定义算法类必须继承 AbstractLoadBalanceRule 类，启动类所在的包是com.linhw.demo，我们新建com.linhw.rule包，并在该包下新建一个类CustomeRule

public class CustomeRule extends AbstractLoadBalancerRule {

    /*

     * total = 0 // 当total==5以后，我们指针才能往下走， index = 0 // 当前对外提供服务的服务器地址，

     * total需要重新置为零，但是已经达到过一个5次，我们的index = 1

     */

    private int total = 0; // 总共被调用的次数，目前要求每台被调用5次

    private int currentIndex = 0; // 当前提供服务的机器号

    public Server choose(ILoadBalancer lb, Object key) {

        if (lb == null) {

            return null;

        }

        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;

            }

            // int index = rand.nextInt(serverCount);

            // server = upList.get(index);

            if (total < 5) {

                server = upList.get(currentIndex);

                total++;

            } else {

                total = 0;

                currentIndex++;

                if (currentIndex >= upList.size()) {

                    currentIndex = 0;

                }

            }

            if (server == null) {

                Thread.yield();

                continue;

            }

            if (server.isAlive()) {

                return (server);

            }

            // Shouldn't actually happen.. but must be transient or a bug.

            server = null;

            Thread.yield();

        }

        return server;

    }

    @Override

    public Server choose(Object key) {

        return choose(getLoadBalancer(), key);

    }

    @Override

    public void initWithNiwsConfig(IClientConfig clientConfig) {

    }

}

2. 在配置类中自定义规则

@Configuration

public class ConfigBean {

    @Bean

    @LoadBalanced

    public RestTemplate getRestTemplate() {

        return new RestTemplate();

    }

    //配置负载均衡的策略为随机，默认算法为轮询算法

    @Bean

    public IRule lbRule() {

        return new CustomeRule(); //自定义负载均衡规则

    }

}

3. 在主启动类添加 @RibbonClient 注解

@RibbonClient(name="microservicecloud-dept", configuration=ConfigBean.class)

name指定针对哪个服务进行负载均衡，而configuration指定负载均衡的算法具体实现类。

4. 先启动eureka集群，再启动服务提供者，最后启动消费者，访问：http://localhost/consumer/dept/get/1，可以看到先访问生产者1服务5次，然后访问生产者2服务5次......

巴特西

【Ribbon篇四】自定义负载均衡策略（4）

最新文章

热门文章