基于Redis的限流器的实现

2017-12-27 19:14:37来源:CSDN作者:wang258533488人点击

分享

1 概述

系统中的接口通常都有限流,比如 70次/秒 ,如何保证我们的接口的调用次数在超过第三方接口限流的时候快速失败呢?这时候就需要限流器了。下面是笔者用redis实现限流器的流程图。
这里写图片描述

2 代码

/** * 获取限流权限 * @param key * @param millisecond 毫秒数 * @param limitCount 限流次数 * @return */public static boolean getCurrentLimiting(String key, Long millisecond, Integer limitCount){    try {        boolean currentLimitingLock = RedisDistributeLockUtil.lock(key);        if(currentLimitingLock){            Long llen = RedisClient.llen(getCurrentLimitingRedisKey(key));            if(llen < limitCount){                RedisClient.lpush(getCurrentLimitingRedisKey(key), DateUtil.current(false) + "");                return true;            }else{                Long lastTime = Long.parseLong(RedisClient.lindex(getCurrentLimitingRedisKey(key), -1));                if((DateUtil.current(false) - lastTime) >= millisecond){                    RedisClient.lpush(getCurrentLimitingRedisKey(key), DateUtil.current(false) + "");                    RedisClient.ltrim(getCurrentLimitingRedisKey(key), 0, limitCount);                    return true;                }            }        }    }finally {        RedisDistributeLockUtil.unlock(key);    }    return false;}

最新文章

123

最新摄影

闪念基因

微信扫一扫

第七城市微信公众平台