maxwaitmill redis_Redis限流实现

令牌桶算法(Token Bucket)和 Leaky Bucket 效果一样但方向相反的算法,更加容易理解.随着时间流逝,系统会按恒定1/QPS时间间隔(如果QPS=100,则间隔是10ms)往桶里加入Token(想象和漏洞漏水相反,有个水龙头在不断的加水),如果桶已经满了就不再加了.新请求来临时,会各自拿走一个Token,如果没有Token可拿了就阻塞或者拒绝服务.

令牌桶的另外一个好处是可以方便的改变速度. 一旦需要提高速率,则按需提高放入桶中的令牌的速率. 一般会定时(比如100毫秒)往桶中增加一定数量的令牌, 有些变种算法则实时的计算应该增加的令牌的数量。

令牌桶内令牌生成借鉴Guava-RateLimiter类的设计 ，每次getToken根据时间戳生成token，不超过最大值。

核心参数：

last_mill_second 最后时间毫秒

curr_permits 当前可用的令牌

max_burst 令牌桶最大值

rate 每秒生成几个令牌

app 应用名称local ratelimit_info=redis.pcall("HMGET",KEYS[1],"last_mill_second","curr_permits","max_burst","rate","app")

local last_mill_second=ratelimit_info[1]

local curr_permits=tonumber(ratelimit_info[2])

local max_burst=tonumber(ratelimit_info[3])

local rate=tonumber(ratelimit_info[4])

local app=tostring(ratelimit_info[5])

if app == nil then

return 0

end

local local_curr_permits=max_burst;

if(type(last_mill_second) ~='boolean' and last_mill_second ~=nil) then

local reverse_permits=math.floor((ARGV[2]-last_mill_second)/1000)*rate

if(reverse_permits>0) then

redis.pcall("HMSET",KEYS[1],"last_mill_second",ARGV[2])

end

local expect_curr_permits=reverse_permits+curr_permits

local_curr_permits=math.min(expect_curr_per