【五】Ribbon负载均衡

<!-- Ribbon相关  -->
<!--Ribbon  需要 eureka -->
<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-eureka</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-ribbon</artifactId>
</dependency>
<dependency>
    <groupId>org.springframework.cloud</groupId>
    <artifactId>spring-cloud-starter-config</artifactId>
</dependency>

eureka:
  client:
    register-with-eureka: false  #自己不能注册
    service-url: 
      defaultZone: http://eureka7001.com:7001/eureka/,http://eureka7002.com:7002/eureka/,http://eureka7003.com:7003/eureka/  

@Configuration
public class ConfigBean
{ 
    @Bean
    @LoadBalanced   //Spring Cloud Ribbon是基于Netflix Ribbon实现的一套客户端  负载均衡的工具。
    public RestTemplate getRestTemplate()
    {
        return new RestTemplate();
    }
    
}

//把下面这行 
private static final String REST_URL_PREFIX = "http://localhost:8001";
//改成下面这行
private static final String REST_URL_PREFIX = "http://MICROSERVICECLOUD-DEPT";

#记得修改数据库名

CREATE SCHEMA `springclouddb01` DEFAULT CHARACTER SET utf8 ;

create table springclouddb01.dept
(
    deptno bigint not null primary key auto_increment,
    dname varchar(60),
    db_source varchar(60)
);

insert into dept(dname , db_source) values(‘开发部‘ , database());
insert into dept(dname , db_source) values(‘人事部‘ , database());
insert into dept(dname , db_source) values(‘财务部‘ , database());
insert into dept(dname , db_source) values(‘市场部‘ , database());
insert into dept(dname , db_source) values(‘运维部‘ , database());

select * from springclouddb01.dept;

* 切换 访问的算法 很简单只需要换成我们要返回算法的实例即可
* 默认有七个算法，可以也自定义自己的算法

package cn.hfbin.myrule;

/**
 * 该配置类不可以放在与注解 @ComponentScan 的同包或者子包下，否则不起作用 （自定义算法也是一样）
 */
@Configuration
public class MySelfRule
{
    /*
    * 切换 访问的算法 很简单只需要换成我们要返回算法的实例即可
    * 默认有七个算法，可以自定义自己的算法
    * */
    @Bean
    public IRule myRule()
    {
        //如果 突然间一个服务挂了 访问带挂的服务器会报错，出现错误页面
        //return new RoundRobinRule();
        //return new RandomRule(); //达到的目的，用我们重新选择的随机算法替代默认的轮询。
        //如果 突然间一个服务挂了 访问带挂的服务器会报错，出现错误页面，但是过一下子他不会再访问挂的机器，不会显示出错误的页面。
        return new RetryRule();
    }
}

@Configuration
public class MySelfRule
{
    @Bean
    public IRule myRule()
    {
        //return new RandomRule();// Ribbon默认是轮询，我自定义为随机
        //return new RoundRobinRule();// Ribbon默认是轮询，我自定义为随机
        //return new RetryRule();
        return new RandomRule();// 我自定义为每台机器5次
    }
}

package cn.hfbin.myrule;

/*
* 参考随机数的源码来修改
*
* https://github.com/Netflix/ribbon/blob/master/ribbon-loadbalancer/src/main/java/com/netflix/loadbalancer/RandomRule.java
* */
public class RandomRule extends AbstractLoadBalancerRule {

    // total = 0 // 当total==5以后，我们指针才能往下走，
    // index = 0 // 当前对外提供服务的服务器地址，
    // total需要重新置为零，但是已经达到过一个5次，我们的index = 1
    // 分析：我们5次，但是微服务只有8001 8002 8003 三台，OK？
    //


    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) {
                /*
                 * No servers. End regardless of pass, because subsequent passes only get more
                 * restrictive.
                 */
                return null;
            }

//            int index = rand.nextInt(serverCount);// java.util.Random().nextInt(3);
//            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) {
                /*
                 * The only time this should happen is if the server list were somehow trimmed.
                 * This is a transient condition. Retry after yielding.
                 */
                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) {
        // TODO Auto-generated method stub

    }

}