负载均衡轮询逻辑

avatar
作者
筋斗云
阅读量:1

背景

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

实现

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

    /**      * 轮询      * @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更新缓存操作的的话,一定要注意线程问题。

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;     }  } 

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!