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

Myth丶恋晨 2022-05-17 02:23 237阅读 0赞

> 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次

如有错误，请留言指出。