Redis脚本Lua实现访问控制频率

Redis 脚本功能可以自定义新的命令,并且也是原子执行。
本篇和上一篇是相对应的。

一、简单介绍下脚本:

redis2.6推出脚本功能,允许开发者使用 Lua语言编写脚本传到Redis中执行,在Lua脚本中可以调用大部分redis命令。

  • 使用脚本优点:
    • 减少网络开销:执行一次脚本只需要发送一次请求,减少网络往返时延。如果不使用脚本,可能多个命令要发送多个请求,还会出现竞态情况。
    • 原子操作:Redis会将脚本作为一个整体执行,中间不会被其他事务插入。编写脚本过程中无须考虑竞态条件,也无须使用事务。事务可以完成的所有功能都可以通过脚本实现。
    • 复用:客户端发送的脚本会永久存储在Redis 中,意味着其他客户端可以复用这段脚本。

二、使用脚本实现访问频率限制,

  • 方案一:
Lua代码如下:
    -- Lua 使用 redis.call()调用redis命令
    local times = redis.call('incr', KEYS[1])

    if times == 1 then
        -- KEYS[1] 键刚刚创建,所以为其设置生存时间
        redis.call('expire', KEYS[1], ARGV[1])
    end 
    if times > tonumber(ARGV[2]) then   -- ARGV[2]为字符串,需要转为number
        return 0
    end
    return 1

执行脚本:
$ redis-cli –eval /path/to/ratelimiting.lua rate.limiting:127.0.0.1 , 10 3


参数说明:
–eval:告诉redis-cli 读取并运行后面的Lua脚本。
/path/to/ratelimiting.lua : 文件位置
rate.limiting:userId 要操作的key,在脚本中用 KEYS[1]获取
10 3 :两个参数,在脚本中用 ARGV[1]和ARGV[2] 获取
注意:”,” 两边的空格不能省。

  • 方案二:
Lua代码如下:
    local limitLen = redis.call('llen', KEYS[1])
    if limitLen < tonumber(ARGV[2]) then
        redis.call('lpush', KEYS[1], os.time())
    else 
        local times = redis.call('lindex', KEYS[1], -1)
        if os.time() - times < ARGV[1] then
            return 0
        else 
            redis.call('lpush', KEYS[1], os.time())
            redis.call('ltrim', KEYS[1], 0, ARGV[2])
        end
    end 
    return 1

执行脚本:
$ redis-cli –eval /path/to/ratelimiting.lua rate.limiting:127.0.0.1 , 10 3


rate.limiting:127.0.0.1 这个key 每10 秒最多访问3次

如有错误,请留言指出。

猜你喜欢

转载自blog.csdn.net/annotation_yang/article/details/81430440