Redis Lua 脚本怎么同时校验库存和扣减数量

库存扣减最怕“先读后写”被并发请求插进来：两个请求都读到库存为 1，随后各自扣减，结果就可能出现超卖。Redis Lua 的做法是把读取、判断和扣减放进同一次脚本执行，并明确区分 KEYS 与 ARGV。库存键放在 KEYS[1]，扣减数量放在 ARGV[1]，校验通过才执行 DECRBY。

这个场景可以直接用 Lua 脚本解决：一次 EVAL 中先读取库存，库存不足返回失败码，库存足够才扣减并返回新库存。不要把业务值拼进脚本文本，也不要在脚本里动态拼接未声明的键名。

一、先把 KEYS 和 ARGV 分工固定下来

Redis 的 EVAL 语法是 EVAL script numkeys [key ...] [arg ...]。numkeys 决定前几个输入是键名；这些键名会进入 Lua 的 KEYS，后面的普通参数进入 ARGV。库存键属于 Redis 键，扣减数量只是业务值，不能混在一起。

Go 客户端调用时也要保持这个顺序：

script := `return {KEYS[1], ARGV[1]}`
// 第三个参数 1 表示后面的第一个值是 Redis 键。
result, err := rdb.Eval(ctx, script, []string{"stock:sku-100"}, "2").Result()
if err != nil {
    // 网络错误或脚本错误交给上层处理，不把它当成库存不足。
    return err
}
_ = result

二、把校验、扣减和返回值放进一次脚本

下面的脚本只处理一个库存键。它先把 ARGV[1] 转成数字，拒绝非正数；然后读取库存。库存不足时返回 {0, stock}，成功时执行 DECRBY 并返回 {1, left}。状态码由业务自己约定，数组结构则保持稳定。

local amount = tonumber(ARGV[1])
-- 扣减数量必须是正整数，避免把加库存误当成扣库存。
if not amount or amount 

这里的关键不是 Lua 代码有多长，而是检查和写入没有回到应用层之间。脚本执行期间 Redis 会连续处理脚本中的命令，所以并发请求不会在“读到库存”和“扣减库存”之间插入另一段业务逻辑。

三、客户端只按返回结构处理结果

不要让脚本返回“库存不足”这样的展示文案。返回码更适合日志、重试和多语言接口：1 表示成功，0 表示库存不足，-2 表示参数错误。Go 客户端拿到数组后，根据类型断言成整数，再映射成自己的业务错误。

values, err := rdb.Eval(ctx, script, []string{"stock:sku-100"}, 2).Int64Slice()
if err != nil {
    // 脚本执行失败与业务失败分开记录，便于区分 Redis 故障。
    return err
}
if len(values) != 2 {
    // 返回结构变化应尽快暴露，不要静默当成成功。
    return errors.New("unexpected redis lua result")
}
switch values[0] {
case 1:
    fmt.Printf("扣减成功，剩余库存：%d\n", values[1])
case 0:
    fmt.Printf("库存不足，当前库存：%d\n", values[1])
default:
    return errors.New("invalid stock amount")
}

四、别忽略脚本的键名边界

Redis 文档要求脚本访问的键都显式作为输入键传入，尤其是集群环境。不要在 Lua 中用商品 ID 拼出另一个键，也不要从某个集合里读出键名后再访问它。多个键参与同一脚本时，要让它们落在同一个集群哈希槽，通常使用相同的哈希标签，例如 stock:{sku-100} 和 order:{sku-100}。

另外，脚本文本应保持稳定，把商品编号、数量、阈值放到参数里。这样更容易复用脚本，也不会因为每次把值拼进源码而不断制造新的脚本文本。

常见问题

库存键不存在时会怎样？

示例把不存在的键按 0 处理，因此会返回库存不足。若业务需要区分“未初始化”和“库存为 0”，可以先用 EXISTS 判断并约定新的状态码。

Redis Lua 返回数字为什么要统一结构？

数组结构能让客户端稳定解码。状态码与剩余库存固定在第 1、2 个位置，日志和接口层可以各自生成文案。

可以用 MULTI 代替 Lua 吗？

如果判断结果决定是否写入，Lua 更适合把条件和写入放在一起；MULTI/EXEC 适合已经确定的一组命令，不能自动把应用层的条件判断搬进事务。