Ribbon本地负载均衡轮询算法实现

最新推荐文章于 2023-07-21 10:26:17 发布

昕er

最新推荐文章于 2023-07-21 10:26:17 发布

阅读量327

点赞数 2

文章标签： ribbon 负载均衡算法

本文链接：https://blog.csdn.net/weixin_45638884/article/details/122848358

版权

前提：80服务可以调用8888服务 Rinbon下实现

在80配置包有如下配置可以实现服务调用

@Configuration
public class ApplicationContextConfig {

    @Bean
    @LoadBalanced
    public RestTemplate getRestTemplate(){
        return  new RestTemplate();
    }
}

通常Ribbon可以实现轮询负载均衡

实现本地：

去掉@LoadBalanced

在80服务com.atguigu.springcloud.lb中创建LoadBalancer 接口

public interface LoadBalancer {
     //收集服务器总共有多少台能够提供服务的机器，并放到list里面
    ServiceInstance instances(List<ServiceInstance> serviceInstances);

}

在 com.atguigu.springcloud.lb下实现

  @Component
  public class MyLB implements LoadBalancer {
  
      private AtomicInteger atomicInteger = new AtomicInteger(0);
  
      //坐标
      private final int getAndIncrement(){
          int current;
          int next;
          do {
              current = this.atomicInteger.get();
              next = current >= 2147483647 ? 0 : current + 1;
          }while (!this.atomicInteger.compareAndSet(current,next));  //第一个参数是期望值，第二个参数是修改值是
          System.out.println("*******第几次访问，次数next: "+next);
          return next;
      }
  
      @Override
      public ServiceInstance instances(List<ServiceInstance> serviceInstances) {  //得到机器的列表
         int index = getAndIncrement() % serviceInstances.size(); //得到服务器的下标位置
          return serviceInstances.get(index);
      }
  }

在controller中

package com.atguigu.springcloud.controller;

public class OrderController {

   // public static final String PAYMENT_URL = "http://localhost:8001";
    public static final String PAYMENT_URL = "http://CLOUD-PAYMENT-SERVICE";

    @Resource
    private RestTemplate restTemplate;

    @Resource
    private LoadBalancer loadBalancer;

    @Resource
    private DiscoveryClient discoveryClient;

..........

    @GetMapping(value = "/consumer/payment/lb")
     public String getPaymentLB(){
        List<ServiceInstance> instances = discoveryClient.getInstances("CLOUD-PAYMENT-SERVICE");
        if (instances == null || instances.size() <= 0){
            return null;
        }
        ServiceInstance serviceInstance = loadBalancer.instances(instances);
        URI uri = serviceInstance.getUri();
        return restTemplate.getForObject(uri+"/payment/lb",String.class);
    }
}