您的位置:首页 > 数据库 > Redis

基于Redis的限流器的实现

2020-06-09 05:20 381 查看

1.概述

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


2.代码

[code]/**
 * 获取限流权限
 * @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;
}


--------------------- 
作者:秦霜 
来源:CSDN 
原文:https://blog.csdn.net/wang258533488/article/details/78913827 
版权声明:本文为博主原创文章,转载请附上博文链接!

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: