抢票或商品订单防并发、防超卖

分享一下自己项目开发过程中的设计方案，涉及到订单防并发、防超卖的需求，兼容类似抢票，订单需要锁定商品序号，付款后成交，超时或取消后需要将序号放回可以重新下单的需求。

目录

函数整体思路

代码实现 

优化思路

---

函数整体思路

• 使用redis列表分发购买令牌序号tokenId，分页加载防超内存，而且可以丢失重建。
• mysql联合唯一索引做最后异常兜底。
• 为防止幻读，获取令牌序号tokenId函数不能包含在事务内。

代码实现 

<?php
/*
* 基于Yii集成，Query::table是yii\db\Query->from的简单二次封装
* */

class AppOrder
{
    /*
     * 创建订单
     * */
    public static function createOrder($userId, $productId)
    {
        $product = ProProduct::findOne(['id' => $productId]);
        //获取tokenId
        $tokenId = self::getTokenId($product);
        if ($tokenId == 0) {
            return ['status' => 'err', 'message' => '商品已售罄'];
        }

        $transaction = \Yii::$app->db->beginTransaction();
        try {
            //创建主订单
            $order = [
                'orderNo' => (APP_ENV == 'prod' ? '' : (APP_ENV . '_')) . date('YmdHis') . rand(1000, 9999),
                'userId' => $userId,
                'price' => $product->price,
            ];
            $orderList['orderId'] = Query::table('app_order')->insert($order, true);
            $orderList['tokenId'] = $tokenId;
            $orderList['price'] = $product->price;
            $orderList['productId'] = $productId;
            //创建商品订单
            Query::table('app_order_list')->insert($order + $orderList);
            //更新销量
            \Yii::$app->db->createCommand("UPDATE `pro_product` SET sellNum=sellNum+1 WHERE id = $productId ")->queryAll();
            $transaction->commit();
            return ['status' => 'ok', 'orderNo' => $order['orderNo']];
        } catch (\Exception $e) {
            $transaction->rollBack();
            return ['status' => 'err', 'message' => '下单失败请重试'];
        }
    }

    /*
     * 获取一个可用tokenId，重点强调一下，该方法不能被包含在事务内，防止查询幻读
     * */
    public static function getTokenId($product)
    {
        $rKeyQueue = 'productQueue:id' . $product->id;  //保存可用的tokenid队列
        $rKeyStatus = 'productStatus:id' . $product->id;//标记商品状态，【1在售，2已售罄】
        $rKeyLock = 'productQLock:id' . $product->id;//标记队列正在创建中
        $rKeyToken = 'productToken:id' . $product->id . '-';  //标记token正在使用中
        //判断队列是否存在或为空
        $tokenId = redis()->rpop($rKeyQueue);
        if (!$tokenId) {
            //判断是否是因为商品真实已售罄
            if (redis()->get($rKeyStatus) == 2) return 0;

            //tokenid创建队列，第一个人负责创建队列，其余排队循环获取
            $res = self::lock($rKeyLock, false);
            if (!$res) goto Loop;

            //全局查询是否有使用中的token，防止有tokenId被错误的重新加入了队列
            $res = self::redisScan($rKeyToken . "*", 1);
            if (count($res)) {
                self::unlock($rKeyLock);
                goto Loop;
            }

            /*   一、获取关闭的订单中可用的历史tokenid */
            $where = ['productId' => $product->id, 'isDel' => 0];
            $subSql = Query::table('app_order_list')->select('tokenId')
                ->where($where + ['closeTimeStamp' => 0]);
            $closesTokenIds = Query::table('app_order_list')
                ->select('tokenId')
                ->where($where + [
                        ['>', 'closeTimeStamp', 0],
                        ['not in', 'tokenId', $subSql]
                    ])->distinct("tokenId")->limit(100)->all();
            $tokenIds = array_column($closesTokenIds, 'tokenId');

            /*   二、可用tokenId一次最多取100个，防止发行量设置过大，redis队列内存超载   */
            $maxTokenId = Query::table('app_order_list')->where($where)->max('tokenId') ?: 0;
            for ($i = $maxTokenId + 1; $i <= $product->issuesNum; $i++) {
                if (count($tokenIds) >= 100) break;
                $tokenIds[] = $i;
            }

            if (count($tokenIds) == 0) goto Sellout;//已售罄
            $tokenId = array_shift($tokenIds);//自己拿一个
            if (count($tokenIds) == 0) {
                redis()->setex($rKeyStatus, 60, 2);//1分钟的缓冲
            } else {
                redis()->lpush($rKeyQueue, $tokenIds);
                redis()->expire($rKeyQueue, 3600);//没有抢购完就删除掉队列，需要的时候重新创建吧
            }
            self::unlock($rKeyLock);
        }

        //获取tokenId成功，并标记当前tokenid使用中
        self::lock($rKeyToken . $tokenId);
        return $tokenId;

        //循环获取
        Loop:
        sleep(1);
        return self::getTokenId($product);

        //商品已售罄，并标记已售罄
        Sellout:
        redis()->setex($rKeyStatus, 60, 2);//1分钟的缓冲
        return 0;
    }

    /*
     * redis实现防并发锁
     * @param bool $stop 是否抛错阻止上下文继续执行
     * @param bool $ex 锁有效时间
     * @return bool 是否加锁成功
     * */
    public static $lock = [];

    public static function lock($key, $stop = true, $ex = 60)
    {
        $val = uniqid();
        if (redis()->set($key, $val, 'NX', 'EX', $ex)) {
            self::$lock[$key] = $val;
            register_shutdown_function(function () use ($key) {
                self::unlock($key);
            });
            return true;//上锁成功，返回true
        } else {
            if (!$stop) return false;//上锁失败，返回false
            throw new \Exception('操作进行中，请勿重复请求');
        }
    }

    /*
     * 解锁
     * */
    public static function unLock($key)
    {
        $releaseLuaScript = <<<LUA
if redis.call("GET",KEYS[1])==ARGV[1] then
    return redis.call("DEL",KEYS[1])
else
    return 0
end
LUA;
        return redis()->eval($releaseLuaScript, 1, $key, self::$lock[$key]);
    }
    
    /*
     * 用scan实现redis的keys操作避免redis卡顿现象
     * $limit 获取数量，0不限
     * $prefix是否需要连接配置前缀
     * */
    public static function redisScan($pattern, $limit = 0, $prefix = true)
    {
        $data = [];
        $cursor = 0;
        if ($prefix === true) $pattern = redis()->options['prefix'] . $pattern;//是否添加前缀
        do {
            list($cursor, $keys) = redis()->scan($cursor, 'MATCH', $pattern);
            $cursor = (int)$cursor;
            if (!empty($keys)) {
                $data = array_merge($data, $keys);
                if (count($keys) >= $limit and $limit) break;
            }
        } while ($cursor !== 0);
        return $limit ? array_slice($data, 0, $limit) : $data;
    }

}

优化思路

• tokenId队列重建查询逻辑可以使用redis的Bitmap实现，其实就是类似tokenId回收站功能，效果也会不错。