Redis秒杀lua脚本

 int result = stringRedisTemplate.execute(SECKILL_SCRIPT, //seckill.lua
                Collections.emptyList(),
                voucherId, userId, orderId
        ).intValue();
-- seckill.lua
-- 创建消息组    XGROUP CREATE  stream.orders g1 0 MKSTREAM
-- 参数列表
local voucherId = ARGV[1] -- 优惠卷Id
local userId = ARGV[2] -- 用户ID
local orderId = ARGV[3] -- 订单ID
-- key列表
local stockKey = 'seckill:stock:' .. voucherId; -- 库存key
local orderKey = 'seckill:order' .. voucherId; -- 订单key
-- 脚本业务
-- 判断库存是否充足 get stockKey
if (tonumber(redis.call('get', stockKey)) >= 0) then
    return 1; -- 库存不足返回1
end
-- 判断用户是否已下单 get orderKey
if (redis.call('sismember', orderKey, userId) == 1) then
    return 2; -- 已下单返回2
end
-- 扣库存 incrby stockKey -1
redis.call('incrby', stockKey, -1);
-- 下单 sadd orderKey userId
redis.call('sadd', orderKey, userId);
-- 发送消息到队列中,xadd stream.orders * k1 v1 ……
redis.call('xadd', 'stream.orders', '*', 'userId', userId, 'voucherId', voucherId, 'id', orderId);
return 0;
-- unlock.lua
-- 比较线程标识与锁中的标识是否一致
if(redis.call('GET',KEYS[1]) == ARGV[1]) then
    --释放锁
    return redis.call('DEL',KEYS[1])
end
return 0
-- 可重入锁getlock.lua,单体架构
local key = KEYS[1];
local threadId = ARGV[1];
local releaseTime = ARGV[2]; --自动过期时间
--判断锁是否存在
if (redis.call('exists', key) == 0) then
    -- 创建锁
    redis.call('hset', key, threadId, '1');
    -- 设置存活时间
    redis.call('expire', key, releaseTime);
    return 1;
end ;
-- 存在则判断threadId是否是自己
if (redis.call('hexists', key, threadId) == 1) then
    -- 获取锁,锁重入次数+1
    redis.call('hincrby', key, threadId, 1);
    -- 重新设置存活时间
    redis.call('expire', key, releaseTime);
    return 1;
end ;

return 0; --说明锁不是自己的,获取锁失败
-- 可重入解锁unlock.lua
local key = KEYS[1];
local threadId = ARGV[1];
local releaseTime = ARGV[2]; --自动过期时间
-- 判断锁是不是自己的
if (redis.call('hexists', key, threadId) == 0) then
    return null;
end ;
-- 锁是自己的,锁重入次数减一
local count = redis.call('hincrby', key, threadId, -1);
-- 判断锁重入次数是否已经为0
if (count > 0) then
    -- 大于0说明不能释放锁,重新设置存活时间
    redis.call('expire', key, releaseTime);
    return nil;
else
    --可以释放锁
    redis.call('del', key);
    return nil;
end ;
<think>好的,用户想知道如何在Redis中使用Lua脚本来实现秒杀功能。首先,我得回忆一下秒杀场景的特点,比如高并发、库存扣减的原子性需求,以及如何防止超卖。RedisLua脚本正好能保证操作的原子性,所以很适合这个场景。 接下来,需要确定具体的实现步骤。首先,用户需要设置商品库存,比如用字符串或者哈希结构来存储。然后,在秒杀时,每个请求都会触发Lua脚本的执行,检查库存是否足够,如果足够就扣减库存并记录用户ID。这里的关键是确保这两个操作(检查和扣减)是原子性的,避免竞态条件。 然后,考虑到性能,可能需要用哈希结构来存储库存和已购用户,这样查询和更新会更高效。例如,用商品ID作为键,库存和用户列表作为字段。另外,超卖问题必须处理,Lua脚本中的逻辑需要先判断库存是否大于零,再进行扣减,确保不会出现负数。 还有,用户可能需要示例代码。所以得写一个Lua脚本的例子,用KEYS和ARGV来接收参数,比如商品键和用户ID。脚本里先获取当前库存,然后判断,如果足够就执行hincrby扣减库存,并hset记录用户。最后返回不同的状态码表示成功或失败。 另外,需要注意的一些问题,比如脚本执行时间不能太长,否则会影响Redis性能。所以脚本要尽量简洁,避免复杂计算。同时,分布式锁可能不需要,因为Lua本身原子执行。还要提到集群环境下,要确保所有操作在同一个节点上,可能要用hash tag来保证。 最后,应该给出优化建议,比如库存预热、限流措施、异步处理等,帮助用户全面了解如何在实际中应用。还要提醒他们错误处理和重试机制,比如用Redis的事务或者重试策略来应对失败情况。 检查一下引用的资料,比如引用2和5提到Lua脚本的原子性适合秒杀和分布式锁,引用4提到原子性适合批量操作,这些都要在回答中引用。确保示例代码正确,使用正确的Redis命令,比如hget、hincrby、hset等,并且返回合适的值。 可能还需要考虑其他方面,比如库存预热到Redis,避免每次从数据库读取。以及如何防止同一用户重复购买,可以通过检查用户ID是否已存在来实现。这些细节在示例代码中应该体现出来。 总结一下,步骤应该是:1. 设计数据结构存储库存和用户;2. 编写Lua脚本实现原子操作;3. 处理脚本执行中的注意事项;4. 提供优化和错误处理建议。这样用户就能有一个清晰的实现路径了。</think>### 使用 Lua 脚本Redis 中实现秒杀功能的最佳实践 #### 1. 核心设计原则 - **原子性操作**:通过 Lua 脚本确保库存检查与扣减、用户记录等操作在单次脚本执行中完成,避免并发竞争问题[^2][^5]。 - **高性能设计**:减少网络往返,将关键逻辑封装在脚本中一次性执行[^4]。 - **防止超卖**:通过条件判断确保库存不为负值。 #### 2. 数据结构设计 - **库存存储**:使用 `HASH` 结构存储商品库存和已购买用户列表: ```bash HSET seckill:goods:1001 stock 100 # 商品ID 1001,初始库存100 ``` - **用户记录**:同一 `HASH` 中记录购买用户: ```bash HSET seckill:goods:1001 users 0 # 初始用户列表(用位图或JSON存储) ``` #### 3. Lua 脚本实现示例 ```lua -- KEYS[1]: 商品键(如 seckill:goods:1001) -- ARGV[1]: 用户ID -- 返回值:1(成功)/ 0(库存不足)/ -1(重复购买) local stock_key = KEYS[1] local user_id = ARGV[1] -- 检查用户是否已购买 if redis.call('HEXISTS', stock_key, 'user:'..user_id) == 1 then return -1 end -- 获取当前库存 local stock = tonumber(redis.call('HGET', stock_key, 'stock')) -- 库存不足时直接返回 if stock <= 0 then return 0 end -- 扣减库存并记录用户 redis.call('HINCRBY', stock_key, 'stock', -1) redis.call('HSET', stock_key, 'user:'..user_id, 1) return 1 ``` #### 4. 操作流程 ```python import redis r = redis.Redis() script = """ -- 上述 Lua 脚本内容 """ seckill_script = r.register_script(script) def handle_seckill_request(goods_key, user_id): result = seckill_script(keys=[goods_key], args=[user_id]) if result == 1: print("秒杀成功") elif result == 0: print("库存不足") elif result == -1: print("请勿重复购买") ``` #### 5. 关键优化措施 - **集群模式适配**:使用 `{}` 定义哈希标签确保所有操作在同一个分片: ```lua local stock_key = "{seckill}:"..KEYS[1] ``` - **库存预热**:提前将数据库库存加载到 Redis - **限流保护**:结合 `redis.call('CL.THROTTLE')` 进行请求限流[^4] - **异步落库**:成功后通过消息队列异步更新数据库 #### 6. 注意事项 - **脚本复杂度**:保持脚本简单,避免循环和复杂计算(执行时间应 < 1ms)[^1] - **错误处理**:添加 `pcall` 异常捕获: ```lua local ok, ret = pcall(function() -- 业务逻辑 end) if not ok then return redis.error_reply("执行错误") end ``` - **版本兼容**:Redis 2.6+ 支持完整 Lua 功能 : Lua 脚本的原子性特性可以完美解决秒杀场景的并发控制问题 : 脚本执行期间会阻塞其他操作,需严格控制执行时间 : 建议将库存和用户记录合并存储以提升操作效率
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值