首頁 > 文章 > 後端開發 > 基於PHP使用加鎖實作並發情況下搶碼功能的方法

基於PHP使用加鎖實作並發情況下搶碼功能的方法

墨辰丷原創: 2018-06-01 15:44:181275瀏覽

本文基於php語言使用加鎖實現並發情況下搶碼功能，特定時段開放搶碼並不允許開放的碼重複，本文介紹的非常詳細，需要的朋友參考下

#需求：搶碼功能

要求：

1、特定時段才開放搶碼；

2、每個時段放開的碼是有限的；

3、每個碼不允許重複；

實作：

1、在不考慮並發的情況下實作：

function get_code($len){
$CHAR_ARR = array(&#39;1&#39;,&#39;2&#39;,&#39;3&#39;,&#39;4&#39;,&#39;5&#39;,&#39;6&#39;,&#39;7&#39;,&#39;8&#39;,&#39;9&#39;,&#39;A&#39;,&#39;B&#39;,&#39;C&#39;,&#39;D&#39;,&#39;E&#39;,&#39;F&#39;,&#39;G&#39;,&#39;H&#39;,&#39;I&#39;,&#39;J&#39;,&#39;K&#39;,&#39;L&#39;,&#39;M&#39;,&#39;N&#39;,&#39;O&#39;,&#39;P&#39;,&#39;Q&#39;,&#39;X&#39;,&#39;Y&#39;,&#39;Z&#39;,&#39;W&#39;,&#39;S&#39;,&#39;R&#39;,&#39;T&#39;);
$CHAR_ARR_LEN = count($CHAR_ARR) - 1;
$code = &#39;&#39;;
while(--$len > 0){ $code .= $CHAR_ARR[rand(0,$CHAR_ARR_LEN)]; }
return $code;
}
$pdo = new PDO(&#39;mysql:host=localhost;dbname=ci_test&#39;,&#39;root&#39;,&#39;root&#39;);
//查询当前时间已发放验证码数量
$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[&#39;sum&#39;];
if($code_num < 1){<br>　　 sleep(2); //暂停2秒
$code = get_code(6);
var_dump( $pdo->query("INSERT INTO code_test (code,create_time) VALUES (&#39;$code&#39;,".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(&#39;1&#39;,&#39;2&#39;,&#39;3&#39;,&#39;4&#39;,&#39;5&#39;,&#39;6&#39;,&#39;7&#39;,&#39;8&#39;,&#39;9&#39;,&#39;A&#39;,&#39;B&#39;,&#39;C&#39;,&#39;D&#39;,&#39;E&#39;,&#39;F&#39;,&#39;G&#39;,&#39;H&#39;,&#39;I&#39;,&#39;J&#39;,&#39;K&#39;,&#39;L&#39;,&#39;M&#39;,&#39;N&#39;,&#39;O&#39;,&#39;P&#39;,&#39;Q&#39;,&#39;X&#39;,&#39;Y&#39;,&#39;Z&#39;,&#39;W&#39;,&#39;S&#39;,&#39;R&#39;,&#39;T&#39;);
$CHAR_ARR_LEN = count($CHAR_ARR) - 1;
$code = &#39;&#39;;
while(--$len > 0){ $code .= $CHAR_ARR[rand(0,$CHAR_ARR_LEN)]; }
return $code;
}
$pdo = new PDO(&#39;mysql:host=localhost;dbname=ci_test&#39;,&#39;root&#39;,&#39;root&#39;);
$fp = fopen(&#39;lock.txt&#39;,&#39;r&#39;);
//通过排他锁 锁定该过程
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[&#39;sum&#39;];
if($code_num < 1){
sleep(2);
$code = get_code(6);
var_dump( $pdo->query("INSERT INTO code_test (code,create_time) VALUES (&#39;$code&#39;,".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使用加鎖實作並發情況下搶碼功能的方法

相關文章