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

Posted 小熊猫小

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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

  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

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

以上是关于PHP通过加锁实现并发情况下抢码实现的主要内容,如果未能解决你的问题,请参考以下文章

php 使用redis锁限制并发访问类

MySQL 多版本并发控制

高并发环境下,Redisson实现redis分布式锁

mysql优化之加锁

Java并发系列ReentrantLock源码分析

Go Map 为啥是非线程安全的?