【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次......