基于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);
}