基于Redis-Lua脚本的抢购方案

Lua 脚本功能是 Reids的一大亮点，通过内嵌对 Lua 环境的支持，
Redis 解决了长久以来不能高效地处理 CAS （check-and-set） 命令的缺点。
并且可以通过组合使用多个命令，轻松实现以前很难实现或者不能高效实现的模式。

该抢购方案基于Lua脚本，实际项目中可根据需要扩展命令。

伪代码

是否已抢购(是否达到抢购上限)
是否还有库存
如果未抢购、有库存则扣减，否则返回0
加入已抢购集合
加入购买处理队列
返回1

脚本

编写脚本

script.lua

-- variable
local stockKey = KEYS[1]
local purchasedKey = KEYS[2]
local handlerKey = KEYS[3]
local userId = ARGV[1]

-- check purchased set
if redis.call("SISMEMBER", purchasedKey, userId) ~= 0 then return 0 end

-- check stock
local stock = redis.call("GET", stockKey)
if not stock or tonumber(stock) <= 0 then return 0 end

-- stock decrease
redis.call("DECR", stockKey)
-- add purchased set
redis.call("SADD", purchasedKey, userId)
-- add purchased handler queue
redis.call("LPUSH", handlerKey, userId)
return 1

调试脚本

又是个坑：参数的分隔用逗号 , 而且空格不能少

redis-cli --ldb --eval script.lua apple_stock apple_purchased_set apple_purchased_queue , 711

测试脚本

redis-cli

文本转换为命令：

script load "-- variable\nlocal stockKey = KEYS[1]\nlocal purchasedKey = KEYS[2]\nlocal handlerKey = KEYS[3]\nlocal userId = ARGV[1]\n\n-- check purchased set\nif redis.call(\"SISMEMBER\", purchasedKey, userId) ~= 0 then return 0 end\n\n-- check stock\nlocal stock = redis.call(\"GET\", stockKey)\nif not stock or tonumber(stock) <= 0 then return 0 end\n\n-- stock decrease\nredis.call(\"DECR\", stockKey)\n-- add purchased set\nredis.call(\"SADD\", purchasedKey, userId)\n-- add purchased handler queue\nredis.call(\"LPUSH\", handlerKey, userId)\nreturn 1"

得到sha1，执行命令：