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

基于php语言使用加锁实现并发情况下抢码功能，特定时间段开放抢码并不允许开放的码重复：

需求：抢码功能

要求：

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。

这样就是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);
}

来源：oschina

链接：https://my.oschina.net/u/4479011/blog/4297997

标签

apache

flock

php

mysql

Benchmark