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);