redis秒杀之lua脚本

Lua 脚本核心原理：

1.单线程模型：Redis 使用单线程处理命令，所有命令按顺序执行。Lua 脚本会被视为一个整体任务，执行期间不会被其他命令中断。

2.原子性保证：将库存检查、扣减、订单记录等多个操作放在一个脚本中，会连续执行，中间不会有其他客户端操作插入。保证了脚本的原子性。

3.无需锁机制：由于 Redis 的单线程特性，Lua 脚本天然避免了并发冲突，无需额外加锁。

实现流程：

1.首先库存预热：活动开始前将对应商品的库存存redis；

2.然后实现lua脚本（可以将库存检查、扣减、订单记录等多个操作放在一个脚本中）。

下述的字段备注

-- 秒杀Lua脚本
-- KEYS $1$ : 库存key
-- KEYS $2$ : 订单集合key(用于去重)
-- ARGV $1$ : 用户ID
-- ARGV $2$ : 当前时间戳(可选)

-- 返回: 1-秒杀成功; 0-秒杀失败

php具体实现如下：

php 复制代码

class SeckillService
{
    private $redis;
    
    public function __construct()
    {
        $this->redis = new Redis();
        $this->redis->connect('127.0.0.1', 6379);
    }
    
    /**
     * 执行秒杀
     * @param int $productId 商品ID
     * @param int $userId 用户ID
     * @return array ['success'=>bool, 'msg'=>string]
     */
    public function seckill($productId, $userId)
    {
        // 定义Redis key
        $stockKey = "product:{$productId}:stock";
        $orderKey = "product:{$productId}:orders";
        
        // Lua脚本
        $luaScript = <<<LUA
local stockKey = KEYS[1]
local orderKey = KEYS[2]
local userId = ARGV[1]

local stock = tonumber(redis.call('GET', stockKey) or 0)
if stock <= 0 then
    return 0
end

if redis.call('SISMEMBER', orderKey, userId) == 1 then
    return 0
end

redis.call('DECR', stockKey)
redis.call('SADD', orderKey, userId)
return 1
LUA;
        
        try {
            // 执行Lua脚本
            $result = $this->redis->eval(
                $luaScript, 
                [$stockKey, $orderKey, $userId], 
                2 // KEYS的数量
            );
            
            if ($result == 1) {
                // 秒杀成功，异步创建实际订单到数据库
                $this->createOrderAsync($productId, $userId);
                return ['success' => true, 'msg' => '秒杀成功'];
            }
            
            return ['success' => false, 'msg' => '秒杀失败，可能已售罄或您已参与过'];
            
        } catch (Exception $e) {
            // 记录日志
            error_log("秒杀异常: " . $e->getMessage());
            return ['success' => false, 'msg' => '系统繁忙，请稍后再试'];
        }
    }
    
    /**
     * 异步创建订单(实际项目中可用消息队列实现)
     */
    private function createOrderAsync($productId, $userId)
    {
        // 这里可以写入消息队列或直接处理
        // 示例: 写入文件日志，实际项目中不要这样做
        $orderData = json_encode([
            'product_id' => $productId,
            'user_id' => $userId,
            'time' => time()
        ]);
        file_put_contents('orders.log', $orderData . "\n", FILE_APPEND);
        
        // 实际项目建议使用消息队列如RabbitMQ、Kafka等
        // $this->mq->publish('order_queue', $orderData);
    }
    
    /**
     * 初始化商品库存(仅在活动前调用)
     */
    public function initStock($productId, $quantity)
    {
        $stockKey = "product:{$productId}:stock";
        return $this->redis->set($stockKey, $quantity);
    }
}