LUA+Reids实现库存秒杀预扣减记录流水以及自己的思考

news2025/6/10 13:06:27

lua脚本

记录流水

记录流水的作用

流水什么时候删除

我们在做库存扣减的时候，显示基于Lua脚本和Redis实现的预扣减

这样可以在秒杀扣减的时候保证操作的原子性和高效性

lua脚本

// ... 已有代码 ...

    @Override
    public InventoryResponse decrease(InventoryRequest request) {
        // 创建库存响应对象
        InventoryResponse inventoryResponse = new InventoryResponse();
        // 定义用于减少库存的Lua脚本
        String luaScript = """
                -- 检查哈希表 KEYS[2] 中是否已存在 ARGV[2] 对应的字段
                -- 如果存在，说明该操作已执行过，返回错误信息
                if redis.call('hexists', KEYS[2], ARGV[2]) == 1 then
                    return redis.error_reply('OPERATION_ALREADY_EXECUTED')
                end
                                
                -- 从 Redis 中获取 KEYS[1] 对应的值，即当前库存
                local current = redis.call('get', KEYS[1])
                -- 如果返回值为 false，说明键不存在，返回错误信息
                if current == false then
                    return redis.error_reply('KEY_NOT_FOUND')
                end
                -- 尝试将当前库存值转换为数字，如果转换失败，返回错误信息
                if tonumber(current) == nil then
                    return redis.error_reply('current value is not a number')
                end
                -- 如果当前库存为 0，返回库存为零的错误信息
                if tonumber(current) == 0 then
                    return redis.error_reply('INVENTORY_IS_ZERO')
                end
                -- 如果当前库存小于要减少的数量 ARGV[1]，返回库存不足的错误信息
                if tonumber(current) < tonumber(ARGV[1]) then
                    return redis.error_reply('INVENTORY_NOT_ENOUGH')
                end
                                
                -- 计算减少库存后的新库存值
                local new = tonumber(current) - tonumber(ARGV[1])
                -- 将新的库存值存储回 Redis 中
                redis.call('set', KEYS[1], tostring(new))
                                
                -- 获取 Redis 服务器的当前时间（秒和微秒）
                local time = redis.call("time")
                -- 将获取到的时间转换为毫秒级时间戳
                local currentTimeMillis = (time[1] * 1000) + math.floor(time[2] / 1000)
                                
                -- 使用哈希结构存储库存减少操作的日志
                -- 在 KEYS[2] 对应的哈希表中，以 ARGV[2] 为字段名，存储操作日志的 JSON 字符串
                redis.call('hset', KEYS[2], ARGV[2], cjson.encode({
                    action = "decrease",  -- 操作类型为减少库存
                    from = current,       -- 操作前的库存值
                    to = new,             -- 操作后的库存值
                    change = ARGV[1],     -- 减少的库存数量
                    by = ARGV[2],         -- 操作标识
                    timestamp = currentTimeMillis  -- 操作的时间戳
                }))
                                
                -- 返回更新后的库存值
                return new
                """;

        try {
            // ... 已有代码 ...

首先是合法性校验

然后是原子性扣减

最后记录一条库存扣减流水