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

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

需求：抢码功能

要求：

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

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

3、每个码不允许重复；

实现：

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

 1 function get_code($len){
 2 　　$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');
 3 　　$CHAR_ARR_LEN = count($CHAR_ARR) - 1;
 4 　　$code = '';
 5 　　while(--$len > 0){ 
　　　　　　$code .= $CHAR_ARR[rand(0,$CHAR_ARR_LEN)]; 
　　　　}
 6 　　return $code;
 7 }
 8 $pdo = new PDO('mysql:host=localhost;dbname=ci_test','root','root');
 9 //查询当前时间已发放验证码数量
10 $code_num_rs = $pdo->query("SELECT COUNT(*) as sum FROM code_test");
11 $code_num_arr = $code_num_rs->fetch(PDO::FETCH_ASSOC);
12 $code_num = $code_num_arr['sum'];
13 if($code_num < 1){
　　　　sleep(2); //暂停2秒
14 　　$code = get_code(6);
15 　　var_dump( $pdo->query("INSERT INTO code_test (code,create_time) VALUES ('$code',".time().")") );
16 }

 

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

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

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

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

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

2、并发情况下实现：

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

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

 

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

　　怎么避免这情况发生呢？　

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

 

 1 //生成码
 2 function get_code($len){
 3 　　$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');
 4 　　$CHAR_ARR_LEN = count($CHAR_ARR) - 1;
 5 　　$code = '';
 6 　　while(--$len > 0){ 
　　　　　　$code .= $CHAR_ARR[rand(0,$CHAR_ARR_LEN)]; 
　　　　}
 7 　　return $code;
 8 }
 9 $pdo = new PDO('mysql:host=localhost;dbname=ci_test','root','root');
10 $fp = fopen('lock.txt','r');
11 //通过排他锁 锁定该过程
12 if(flock($fp,LOCK_EX)){
13 　　//查询当前时间已发放验证码数量
14 　　$code_num_rs = $pdo->query("SELECT COUNT(*) as sum FROM code_test");
15 　　$code_num_arr = $code_num_rs->fetch(PDO::FETCH_ASSOC);
16 　　$code_num = $code_num_arr['sum'];
17 　　if($code_num < 1){
18 　　　　sleep(2);
19 　　　　$code = get_code(6);
20 　　　　var_dump( $pdo->query("INSERT INTO code_test (code,create_time) VALUES ('$code',".time().")") );
21 　　}
22 　　flock($fp,LOCK_UN);
23 　　fclose($fp);
24 }