负载均衡轮询逻辑

背景

在有些业务场景中,需要轮流的取出某个集合中的其中一个数据,这个时候就需要一个轮询逻辑,比较优秀的是ribbon+feign中的轮询逻辑,轮流取出服务器列表中的地址。

这里就模仿ribbon的轮询负载做了一个轮询的策略方法

实现

每次使用的时候,就是传入要负载的数据总数,会返回一个本次应该取第几条

java 复制代码
    /**
     * 轮询
     * @param staffNumber 总数
     * @return 应取第几条
     */
    private int getPolling(int staffNumber){
        try {
            Long next = pollingConfig.getNextAndSet();
            long index = next % staffNumber;
            return (int)index;
        }catch (Exception e){
            log.error("轮询策略取用户号码失败,默认取第一个号码", e);
            return 0;
        }

    }

数据次数查询,不断的+1很简单的逻辑

这里的+1和更新逻辑是用redis来做的,毕竟redis的increment的+1操作并且更新是通过lua脚本实现的,是属于原子性的操作,不会出现+1 但是没有更新的情况。

如果自己通过本地内存实现这个+1更新缓存操作的的话,一定要注意线程问题。

java 复制代码
public class PollingConfig {


    @Resource
    private RedisCacheHelper redisCacheHelper;

    private static long MAX = 1000000L;

    /**
     * 计数器,获取下一个数据
     * @return 结果
     */
    public Long getNextAndSet(){
        Long increment = redisCacheHelper.increment(RedisKeyConstants.WHATS_APP_NUMBER_POLLING, 1);
        if (increment > MAX) {
            return incrementZero();
        }
        return increment;
    }

    //这个方法是为了防止数值过大
    public synchronized Long incrementZero(){
        Long increment = redisCacheHelper.increment(RedisKeyConstants.WHATS_APP_NUMBER_POLLING, 1);
        if (increment > MAX) {
            redisCacheHelper.set(RedisKeyConstants.WHATS_APP_NUMBER_POLLING, "1");
            return 1L;
        }
        return increment;
    }

}
相关推荐
程序猿小D2 小时前
[附源码+数据库+毕业论文]基于Spring+MyBatis+MySQL+Maven+jsp实现的个人财务管理系统,推荐!
java·数据库·mysql·spring·毕业论文·ssm框架·个人财务管理系统
永日456703 小时前
学习日记-spring-day42-7.7
java·学习·spring
二十雨辰3 小时前
[尚庭公寓]07-Knife快速入门
java·开发语言·spring
NE_STOP5 小时前
SpringBoot--简单入门
java·spring
张小洛7 小时前
Spring AOP 设计解密:代理对象生成、拦截器链调度与注解适配全流程源码解析
java·后端·spring·spring aop·aop
Wyc724098 小时前
SpringBoot
java·spring boot·spring
Orlando cron9 小时前
源哈希(sh)解析
负载均衡·哈希算法
neoooo10 小时前
别慌,Java只有值传递——一次搞懂“为啥我改了它还不变”!
java·后端·spring
GJCTYU11 小时前
spring中@Transactional注解和事务的实战理解附代码
数据库·spring boot·后端·spring·oracle·mybatis
linweidong11 小时前
七牛云Java开发面试题及参考答案(60道面试题汇总)
spring·读写分离·aop·cap·java开发·spring ioc·java面经