关于redis实现限速器有哪几种方式

5e7ea7600dae8805

 

redis 实现限速器的几种方式。

GET + INCR + EXPIRE

先获取 key 的当前值,如果没有超出限制再执行 INCR 增1,如果 key 不存在,使用 redis 的事务初始化 key 和过期时间。

(推荐:redis视频教程)

伪代码:

count = redis.GET(key)
if redis return nil {
  redis.MULTI
  	redis.INCR(key)
  	redis.EXPIRE(key, expire_time)
  redis.EXEC
  count = 1
}
if count > limit {
  return 超出限制
} else {
  redis.INCR(key)
}

高并发下的问题:

如果同时10个并发程序运行GET返回nil,则10个并发程序执行redis中的事务,将密钥增加1,但每个程序的count值为1。如果limit设置为小于10的值,则实际运行的程序将超过限制。事务执行完成后,如果再次将redis分配分配给count,则每个程序可能返回10,因此程序无法继续。

key 已经存在的情况下,先 GET 后 INCR 的逻辑也可能会出现实际执行的程序数多于 limit 的情况。

INCR + EXPIRE

先 INCR, 如果值为1说明是 key 刚设置的,此时再执行 EXPIRE

伪代码:

count = redis.INCR(key)
if count == 1 {
  redis.EXPIRE(key, expire_time)
}
if count > limit {
  return 超出限制
}

慎用

如果 INCR 之后程序挂掉了,没有执行 EXPIRE, 那么这个 key 就没有过期时间了,具体的影响视需求而定。

lua脚本

local current
current = redis.call("incr",KEYS[1])
if tonumber(current) == 1 then
    redis.call("expire",KEYS[1],1)
end
© 版权声明
THE END
喜欢就支持一下吧
点赞15 分享
评论 抢沙发
头像
欢迎您留下宝贵的见解!
提交
头像

昵称

取消
昵称表情代码图片