php和redis设计秒杀活动

1 说明

前段时间面试的时候，一直被问到如何设计一个秒杀活动，但是无奈没有此方面的实际经验，所以只好凭着自己的理解和一些资料去设计这么一个程序
主要利用到了 redis 的 string 和 set,string 主要是利用它的 k - v 结构去对库存进行处理，也可以用 list 的数据结构来处理商品的库存，set 则用来确保用户进行重复的提交
其中我们最主要解决的问题是
- 防止并发产生超抢 / 超卖

2 流程设计

3 代码

3.1 服务端代码

class MiaoSha{

    const MSG_REPEAT_USER = '请勿重复参与';
    const MSG_EMPTY_STOCK = '库存不足';
    const MSG_KEY_NOT_EXIST = 'key 不存在';

    const IP_POOL = 'ip_pool';
    const USER_POOL = 'user_pool';

    /** @var Redis  */
    public $redis;
    public $key;

    public function __construct($key = '')
    {$this->checkKey($key);
        $this->redis = new Redis(); //todo  连接池
        $this->redis->connect('127.0.0.1');
    }

    public function checkKey($key = '')
    {if(!$key) {throw new Exception(self::MSG_KEY_NOT_EXIST);
        } else {$this->key = $key;}
    }

    public function setStock($value = 0)
    {if($this->redis->exists($this->key) == 0) {$this->redis->set($this->key,$value);
        }
    }

    public function checkIp($ip = 0)
    {
        $sKey = $this->key . self::IP_POOL;
        if(!$ip || $this->redis->sIsMember($sKey,$ip)) {throw new Exception(self::MSG_REPEAT_USER);
        }
    }

    public function checkUser($user = 0)
    {
        $sKey = $this->key . self::USER_POOL;
        if(!$user || $this->redis->sIsMember($sKey,$user)) {throw new Exception(self::MSG_REPEAT_USER);
        }
    }

    public function checkStock($user = 0, $ip = 0)
    {$num = $this->redis->decr($this->key);
        if($num < 0) {throw new Exception(self::MSG_EMPTY_STOCK);
        } else {$this->redis->sAdd($this->key . self::USER_POOL, $user);
            $this->redis->sAdd($this->key . self::IP_POOL, $ip);
            //todo add to mysql
            echo 'success' . PHP_EOL;
            error_log('success' . $user . PHP_EOL,3,'/var/www/html/demo/log/debug.log');
        }
    }

    /**
     * @note: 此种做法不能防止并发
     * @func checkStockFail
     * @param int $user
     * @param int $ip
     * @throws Exception
     */
    public function checkStockFail($user = 0,$ip = 0) {$num = $this->redis->get($this->key);
        if($num > 0){$this->redis->sAdd($this->key . self::USER_POOL, $user);
            $this->redis->sAdd($this->key . self::IP_POOL, $ip);
            //todo add to mysql
            echo 'success' . PHP_EOL;
            error_log('success' . $user . PHP_EOL,3,'/var/www/html/demo/log/debug.log');
            $num--;
            $this->redis->set($this->key,$num);
        } else {throw new Exception(self::MSG_EMPTY_STOCK);
        }
    }
}

3.2 客户端测试代码

function test()
{
    try{
        $key = 'cup_';
        $handler = new MiaoSha($key);
        $handler->setStock(10);
        $user = rand(1,10000);
        $ip = $user;
        $handler->checkIp($ip);
        $handler->checkUser($user);
        $handler->checkStock($user,$ip);
    } catch (\Exception $e) {echo $e->getMessage() . PHP_EOL;
        error_log('fail' . $e->getMessage() .PHP_EOL,3,'/var/www/html/demo/log/debug.log');
    }
}

function test2()
{
    try{
        $key = 'cup_';
        $handler = new MiaoSha($key);
        $handler->setStock(10);
        $user = rand(1,10000);
        $ip = $user;
        $handler->checkIp($ip);
        $handler->checkUser($user);
        $handler->checkStockFail($user,$ip); // 不能防止并发的
    } catch (\Exception $e) {echo $e->getMessage() . PHP_EOL;
        error_log('fail' . $e->getMessage() .PHP_EOL,3,'/var/www/html/demo/log/debug.log');
    }
}

4 测试

测试环境说明

• ubantu16.04
• redis2.8.4
• php5.5

在服务端代码里面我们有两个函数分别是 checkStock 和 checkStockFail, 其中 checkStockFail 不能在高并发的情况下效果很差，不能在 redis 层面保证库存为 0 的时候终止操作。
我们利用 ab 工具进行测试
其中 www.hello.com 是配置的虚拟主机名称 flash-sale.php 是我们脚本的名称

 #第 1 种情况 500 并发下 用客户端的 test2()去执行
 ab -n 500 -c 100 www.hello.com/flash-sale.php 

log 日志的记录结果:

 #第 2 种情况 5000 并发下 用客户端的 test2()去执行
 ab -n 5000 -c 1000 www.hello.com/flash-sale.php 

log 日志的记录结果:

 #第 3 种情况 500 并发下 用客户端的 test()去执行
 ab -n 500 -c 100 www.hello.com/flash-sale.php 

log 日志的记录结果:

 #第 4 种情况 5000 并发下 用客户端的 test()去执行
 ab -n 5000 -c 1000 www.hello.com/flash-sale.php 

log 日志的记录结果:

5 总结

我们从日志中可以很明显的看出第 3、4 中情况下，可以保证商品的数量总是我们设置的库存值 10，但是在情况 1、2 下，则产生了超卖的现象
redis 来控制并发主要是利用了其 api 都是原子性操作的优势，从 checkStock 和 checkStockFail 中可以看出，一个是直接 decr 对库存进行减一操作，所以不存在并发的情况，但是另一个方法是将库存值先取出做减一操作然后再重新赋值，这样的话，在并发下，多个进程会读取到多个库存为 1 的值，因此会产生超卖的情况