PHP通过加锁实现并发情况下抢码实现

Posted 2020-06-26 小熊猫小

需求：抢码功能

要求：

1、特定时间段才开放抢码；

2、每个时间段放开的码是有限的；

3、每个码不允许重复；

 

实现：

1、在不考虑并发的情况下实现：

function get_code($len){
    $CHAR_ARR = array(‘1‘,‘2‘,‘3‘,‘4‘,‘5‘,‘6‘,‘7‘,‘8‘,‘9‘,‘A‘,‘B‘,‘C‘,‘D‘,‘E‘,‘F‘,‘G‘,‘H‘,‘I‘,‘J‘,‘K‘,‘L‘,‘M‘,‘N‘,‘O‘,‘P‘,‘Q‘,‘X‘,‘Y‘,‘Z‘,‘W‘,‘S‘,‘R‘,‘T‘);
    $CHAR_ARR_LEN = count($CHAR_ARR) - 1;
    $code = ‘‘;
    while(--$len > 0){ $code .= $CHAR_ARR[rand(0,$CHAR_ARR_LEN)]; }
    return $code;
}


$pdo = new PDO(‘mysql:host=localhost;dbname=ci_test‘,‘root‘,‘root‘);

//查询当前时间已发放验证码数量
$code_num_rs = $pdo->query("SELECT COUNT(*) as sum FROM code_test");
$code_num_arr = $code_num_rs->fetch(PDO::FETCH_ASSOC);
$code_num = $code_num_arr[‘sum‘];

if($code_num < 1){
　　 sleep(2); //暂停2秒
    $code = get_code(6);
    var_dump( $pdo->query("INSERT INTO code_test (code,create_time) VALUES (‘$code‘,".time().")") );
}

　　上述代码默认满足当前是开放时间，和码是不重复的；

　　在不考虑并发情况下流程：

　　1）选查询当前数据库发放的验证码数量；

　　2）如果还有名额，则生成验证码，插入到数据库，返回验证码到客户端；

　　3）如果已满；则返回提示，已无名额；

2、并发情况下实现：

　　那么看下上面代码在并发情况下得到的结果：

　　测试并发，可以使用apache benchmark来测试，apache benchmark是APACHE旗下的HTTP SERVER的性能评测工具，通过cmd进入到apche的bin目录下，通过ab命令调用，如：ab -c 并发数量 -n 总访问量 url

　　cb -c 100 -n 100 http://localhost/php_mulit.php

　　这样就是100个用户同事去抢1个名额，在查询的时候，每个用户都查询到还有一个名额，则会去生成验证码，插入数据库，返回验证码；这样就造成了验证码发多了。事实上，运行完该命令，数据库多了13条记录，而不是一条。

　　怎么避免这情况发生呢？

　　可以通过加排他锁来锁定判断到插入这个过程，保证这个判断流程任意一时间只有一个进程在运行。实现如下：

　　

//生成码
function get_code($len){
    $CHAR_ARR = array(‘1‘,‘2‘,‘3‘,‘4‘,‘5‘,‘6‘,‘7‘,‘8‘,‘9‘,‘A‘,‘B‘,‘C‘,‘D‘,‘E‘,‘F‘,‘G‘,‘H‘,‘I‘,‘J‘,‘K‘,‘L‘,‘M‘,‘N‘,‘O‘,‘P‘,‘Q‘,‘X‘,‘Y‘,‘Z‘,‘W‘,‘S‘,‘R‘,‘T‘);
    $CHAR_ARR_LEN = count($CHAR_ARR) - 1;
    $code = ‘‘;
    while(--$len > 0){ $code .= $CHAR_ARR[rand(0,$CHAR_ARR_LEN)]; }
    return $code;
}


$pdo = new PDO(‘mysql:host=localhost;dbname=ci_test‘,‘root‘,‘root‘);

$fp = fopen(‘lock.txt‘,‘r‘);
//通过排他锁 锁定该过程
if(flock($fp,LOCK_EX)){
    //查询当前时间已发放验证码数量
    $code_num_rs = $pdo->query("SELECT COUNT(*) as sum FROM code_test");
    $code_num_arr = $code_num_rs->fetch(PDO::FETCH_ASSOC);
    $code_num = $code_num_arr[‘sum‘];
    if($code_num < 1){
        sleep(2);
        $code = get_code(6);
        var_dump( $pdo->query("INSERT INTO code_test (code,create_time) VALUES (‘$code‘,".time().")") );
    }
    flock($fp,LOCK_UN);
    fclose($fp);
}

 

通过flock函数来锁定该过程。

更多flock信息可以参考php手册：http://php.net/manual/zh/function.flock.php

再次运行 

　　cb -c 100 -n 100 http://localhost/php_mulit.php

数据库只增加了一条记录，保证了并发情况下数据的正确。