Redis 实现 访问频率限制

redis如何解决“单位时间内只能n次操作”这样的问题?
比如说1分钟内同一用户只能访问100次

方案1:string

每个用户使用一个times:{userId}的字符串类型作为键。每次操作都递增这个键,如果递增后键返回值为1,则需要设置过期时间为60秒。这样每次访问,都需要判断该值是否为100,如果是,则限制该用户的操作。
伪代码

function time_limit(userId) {
if (EXISTS times:{userId} == 1)
var times = INCR times:{userId}
if times > 99
return -1
else
MULTI
INCR times:{userId}
//此处,如果不加事务,竞态条件可能出现
EXPIRE times:{userId},60
EXEC
return 1
}

上面为什么要用MULTI,那是因为如果在执行完INCR times:{userId}之后,如果(出现故障)没有设置过期时间,那么该键将永远存在,所以需要加上事务。

但是上述的方案可能会出现一个问题,比如说一个用户,在第一分钟的后5秒访问了99次,在第二分钟的前五秒又访问了99次,那么就违背了我们的初衷,尽管这样的情况比较极端,但是我们需要更加精确的控制方式。

方案2:list

使用times:{userId}作为key生成一个list,判断list元素个数,如果小于100,则添加当前时间。否则将判断当前时间和最早的时间的差值,如果小于60,则限制,否则,将最早时间弹出,并将当前时间放入list。

function time_limit(userId) {
var times = LEN times:{userId}
if (times < 100)
LPUSH times:{userId}, now()
return 1
else
var early = LINDEX times:{userId}, -1
if (now() - early < 60) 
return -1
else 
LPUSH times:{userId}, now()
RPOP times:{userId}
return 1
}

细心地人会发现,这里也会出现竞态条件,如果最后一个RPOP没有执行,那么获得的early,和当前时间差值永远小于60秒,所以,这里也要加事务。

但是在高并发的缓存系统中,大量使用事务是非常糟糕的,可以用redis自带的lua脚本功能实现多个操作的“原子性”

方案3:使用lua脚本实现频率限制

local times = redis.call('incr', KEYS[1])

if times == 1 then
redis.call('expire', KEYS[1], ARGV[1])
end

if times > tonumber(ARGV[2]) then
return 0
end
return 1
```javascript
  • 0
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
可以使用 Redislua 脚本来实现访问频率限制。以下是一个简单的示例: ``` -- KEYS[1] 表示限制的键名 -- ARGV[1] 表示时间窗口内允许的最大请求数 -- ARGV[2] 表示时间窗口的大小(单位秒) local count = redis.call("incr", KEYS[1]) if tonumber(count) == 1 then -- 如果键不存在,则设置过期时间 redis.call("expire", KEYS[1], ARGV[2]) end if tonumber(count) > tonumber(ARGV[1]) then -- 如果请求数超过限制,则返回 0 return 0 end -- 返回剩余可用的请求数 return tonumber(ARGV[1]) - tonumber(count) ``` 使用方法: 1. 定义一个 Lua 脚本,将上面的代码保存到一个文件中,比如 `rate_limit.lua`。 2. 将 Lua 脚本加载到 Redis 中: ``` $ redis-cli script load "$(cat rate_limit.lua)" ``` 这将返回一个 SHA1 值,可以使用该值来调用脚本。 3. 在需要进行频率限制的地方,使用以下命令来调用 Lua 脚本: ``` $ redis-cli EVALSHA <sha1> 1 <key> <max_requests> <window_size> ``` 其中 `<sha1>` 是脚本的 SHA1 值,`<key>` 是限制的键名,`<max_requests>` 是时间窗口内允许的最大请求数,`<window_size>` 是时间窗口的大小(单位秒)。 例如,要对 IP 地址为 `127.0.0.1` 的客户端进行每分钟最多允许 10 次请求的限制,可以使用以下命令: ``` $ redis-cli EVALSHA <sha1> 1 "rate_limit:127.0.0.1" 10 60 ``` 如果返回值为 0,则表示已经超过了限制,否则表示剩余可用的请求次数。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值