Redis系列 | 阿里云 Redis 版数据库Lua脚本支持与限制

Lua使用限制

为了保证脚本里面的所有操作都在相同slot进行，云数据库Redis集群版本会对Lua脚本做如下限制：

所有key都应该由KEYS数组来传递，redis.call/pcall中调用的redis命令，key的位置必须是KEYS array（不能使用Lua变量替换KEYS），否则直接返回错误信息：

原生的Lua脚本编写

            $script = <<<luascript
			local rate_key = KEYS[1]
			local limit = tonumber(ARGV[1]) -- 最大限制
			local expire_time = ARGV[2] -- 过期时间
			local result = redis.call('SETNX',rate_key,1); -- SETNX 只有不存在的时候才设置,返回1。否则返回0
			if result == 1 then
				redis.call('expire',rate_key,expire_time)
				return 1
			else
				if tonumber(redis.call("GET", rate_key)) >= limit then
					return 0
				else
					redis.call("INCR", rate_key)
					return 1
				end
			end
luascript;
            $scriptSha = $redis->script('load', $script);

 以上脚本在原生的Redis是可以运行的，但是在阿里云Redis则不行，直接返回false

解释一下官方的限制：

key的位置必须是KEYS array（不能使用Lua变量替换KEYS），而上面的脚本则是把KEYS[1] 赋值给了Lua变量 rate_key ，从而导致不能使用

最终修改兼容版本

        local limit = tonumber(ARGV[1]) -- 最大限制
        local expire_time = ARGV[2] -- 过期时间
        local result = redis.call('SETNX',KEYS[1],1); -- SETNX 只有不存在的时候才设置,返回1。否则返回0
        if result == 1 then
            redis.call('expire',KEYS[1],expire_time)
            return 1
        else
            if tonumber(redis.call("GET", KEYS[1])) >= limit then
                return 0
            else
                redis.call("INCR", KEYS[1])
                return 1
            end
        end
luascript;
        $scriptSha = $redis->script('load', $script);