Spring Cloud 手写负载均衡算法_手写spring cloud balance-优快云博客

本文链接：https://blog.youkuaiyun.com/qq_38152423/article/details/118996502

该博客介绍了如何在Spring Cloud中实现两种负载均衡策略：轮询和随机算法。`RotationLoadBalancer`实现了`LoadBalancer`接口，提供了`getSingleAddress`方法用于轮询算法，每次访问都会增加一个计数并取模确定服务实例；`getRandomAddress`方法则使用随机数选择服务实例。这两种方法有助于在分布式系统中有效地分配请求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package com.baojiaren.balancer;

import org.springframework.cloud.client.ServiceInstance;

import java.util.List;

/**
 * @author ✎ℳ๓₯㎕.倾心❀、
 * @create 2021-07-22-11:47
 */
public interface LoadBalancer {


    /**
     * 根据多个不同的地址 返回单个调用rpc地址 轮询算法
     * @param serviceInstances
     * @return
     */
    public ServiceInstance getSingleAddress(List<ServiceInstance> serviceInstances);

    /**
     * 根据多个不同的地址 返回单个调用rpc地址 随机算法
     * @param serviceInstances
     * @return
     */
    public ServiceInstance getRandomAddress(List<ServiceInstance> serviceInstances);
}

package com.baojiaren.balancer.impl;


import com.baojiaren.balancer.LoadBalancer;
import com.netflix.loadbalancer.RandomRule;
import org.springframework.cloud.client.ServiceInstance;
import org.springframework.stereotype.Service;

import java.lang.annotation.Annotation;
import java.util.List;
import java.util.Random;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * @author ✎ℳ๓₯㎕.倾心❀、
 * @create 2021-07-22-11:52
 * 负载均衡接口
 */
@Service
public class RotationLoadBalancer implements LoadBalancer {

    private AtomicInteger atomicInteger= new AtomicInteger(0);

    /**
     * 轮询算法
     * @param serviceInstances
     * @return
     */
    @Override
    public ServiceInstance getSingleAddress(List<ServiceInstance> serviceInstances) {
        int count = atomicInteger.incrementAndGet();//拿到访问次数
        int index=count%serviceInstances.size(); //取余  得到访问第几个
        /**
         * [192.168]
         * 第一次访问 1%2=1
         * 第二次访问 2%2=0
         * 第三次访问 3%2=1
         * 第四次访问 4%2=0
         */
        return serviceInstances.get(index);
    }

    /**
     * 随机算法
     * @param serviceInstances
     * @return
     */
    @Override
    public ServiceInstance getRandomAddress(List<ServiceInstance> serviceInstances) {
        //随机算法: 随机数最小0 最大值serviceInstances.size()
        Random random=new Random();
        return serviceInstances.get(random.nextInt(serviceInstances.size()));
    }
}