淺談併發加鎖

luler發表於2020-01-05

在我們的工作中,常常會出現一些對數量控制有精確要求的需求,比如商品庫存量、獎品數量、報名人數限制等等,這些應用場景往往都存在高併發可能,比較容易出現資料量超量問題。以下做一下示例探索:

首先設計一個存量表

CREATE TABLE `product` (
  `id` int(11) NOT NULL AUTO_INCREMENT,
  `product_name` varchar(255) NOT NULL DEFAULT '',
  `count` int(10) NOT NULL DEFAULT '0',
  PRIMARY KEY (`id`)
) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8mb4;

新增一行資料如下,設定基礎庫存量為10

淺談併發加鎖

問題程式碼如下:

        $process_num = 50; //開50個程式,模擬50個使用者
        for ($i = 0; $i < $process_num; $i++) {
            MultiProcessHelper::instance($process_num)->multiProcessTask(function () use ($i) {
                if (Db::name('product')->where('id', 1)->value('count') > 0) {
                    $res = Db::name('product')->where('id', 1)->setDec('count');
                    if ($res) {
                        dump('獲取到更新資源許可權:' . $i);
                    }
                }
            });
        }

執行結果,50個使用者都獲取到了更新資源的許可權

淺談併發加鎖

資料庫相應資料存量變成了-40

淺談併發加鎖

這顯然不是我們所期待的,這就是高併發帶來的問題,同一時刻有多個程式讀取同一條資料,同一時刻有多個程式更新同一條資料

資料庫自有的鎖機制解決

  1. 要進行DML層面的限制(最後關卡安全,報錯總比出現資料問題產生的影響小),主要的修改是將count的型別改成了無符號整數,這樣該值就不可能再出現負數值
    CREATE TABLE `product` (
    `id` int(11) NOT NULL AUTO_INCREMENT,
    `product_name` varchar(255) NOT NULL DEFAULT '',
    `count` int(10) unsigned NOT NULL DEFAULT '0',
    PRIMARY KEY (`id`)
    ) ENGINE=InnoDB AUTO_INCREMENT=2 DEFAULT CHARSET=utf8mb4;

執行以下程式碼,出現資料庫報錯

淺談併發加鎖

  1. mysql提供的行級鎖select ... lock in share mode(阻塞寫),select ... for update(阻塞讀寫,悲觀鎖),所以for update機制能滿足我們的原子要求。編輯程式碼如下:
        $process_num = 50; //開50個程式,模擬50個使用者
        for ($i = 0; $i < $process_num; $i++) {
            MultiProcessHelper::instance($process_num)->multiProcessTask(function () use ($i) {
                Db::startTrans(); //行級鎖必須在事務中才能生效
                //設定for update,程式會阻塞在這裡,只能允許一個程式獲取到行鎖,其他等待獲取
                if (Db::name('product')->where('id', 1)->lock('for update')->value('count') > 0) { 
                    $res = Db::name('product')->where('id', 1)->setDec('count');
                    if ($res) {
                        dump('獲取到更新資源許可權:' . $i);
                    }
                }
                Db::commit();
            });
        }

只有十個程式獲取到了更新許可權,消費正常

淺談併發加鎖

淺談併發加鎖

  1. 將條件語句放到update上,保持語句執行的原子性,杜絕併發幻讀
    修改程式碼如下:
        $process_num = 50; //開50個程式,模擬50個使用者
        for ($i = 0; $i < $process_num; $i++) {
            MultiProcessHelper::instance($process_num)->multiProcessTask(function () use ($i) {
                    //合併兩條語句為一條更新語句
                    $res = Db::name('product')->where('id', 1)->where('count', '>', 0)->setDec('count');
                    if ($res) {
                        dump('獲取到更新資源許可權:' . $i);
                    }
            });
        }

只有十個程式獲取到了更新許可權,消費正常

淺談併發加鎖

淺談併發加鎖

檔案鎖機制解決

編輯程式碼如下:

        $process_num = 50; //開50個程式,模擬50個使用者
        for ($i = 0; $i < $process_num; $i++) {
            MultiProcessHelper::instance($process_num)->multiProcessTask(function () use ($i) {
                $filename = app()->getRootPath() . 'runtime/lock';
                $file = fopen($filename, 'w'); //開啟檔案
                $lock = flock($file, LOCK_EX);
                // $lock=flock($handle, LOCK_EX|LOCK_NB); (非同步非阻塞,所有程式如果出現獲取不到鎖,不等待跳過,加鎖失敗)
                //獲取檔案排他鎖:LOCK_EX(非同步阻塞,只有一個程式獲得鎖,其他競爭程式等待)
                //還有一種共享鎖:LOCK_SH(所有程式都可以獲取共享鎖,讀取檔案,當且只有一個鎖時,才允許寫操作,否則操作失敗,容易出現死鎖問題)
                if ($lock) {
                    try {
                        if (Db::name('product')->where('id', 1)->lock('for update')->value('count') > 0) {
                            $res = Db::name('product')->where('id', 1)->setDec('count');
                            if ($res) {
                                dump('獲取到更新資源許可權:' . $i);
                            }
                        }
                    } catch (\Exception $e) {
                        dump($e->getMessage());
                    } finally {
                        flock($file, LOCK_UN); //無論如何都要釋放鎖
                    }
                }
                fclose($file); //關閉檔案控制程式碼
            });
        }

只有十個程式獲取到了更新許可權,消費正常

淺談併發加鎖

淺談併發加鎖

分散式鎖機制解決

以上檔案鎖,只適應於單體架構的需求,在叢集架構、分散式等多機聯網結構中就是掩耳盜鈴了,所以適應性更好地鎖機制還是要使用分散式鎖,分散式鎖最常用和最易用就是redis的setnx鎖了。
編輯程式碼如下:

        $process_num = 50; //開50個程式,模擬50個使用者
        for ($i = 0; $i < $process_num; $i++) {
            MultiProcessHelper::instance($process_num)->multiProcessTask(function () use ($i) {
                //獲取redis鎖
                //關於CacheHelper::getRedisLock是怎樣獲取鎖的,注意幾個點就行:1.如何避免死鎖;2.如何設定過期時間;3.如何設定搶佔條件;4.如何迴圈等待判斷。這些不在本文討論範圍,可自行研究,以後有空我也可以寫一篇博文
                $lock = CacheHelper::getRedisLock('redis_lock');
                if ($lock) {
                    try {
                        if (Db::name('product')->where('id', 1)->lock('for update')->value('count') > 0) {
                            $res = Db::name('product')->where('id', 1)->setDec('count');
                            if ($res) {
                                dump('獲取到更新資源許可權:' . $i);
                            }
                        }
                    } catch (\Exception $e) {
                        dump($e->getMessage());
                    }
                } else {
//                    dump('獲取redis鎖失敗');
                }
            });
        }

只有十個程式獲取到了更新許可權,消費正常

淺談併發加鎖

淺談併發加鎖

加鎖並不是最好的解決方案,只能達到資料安全的要求,同時效能會很大的損耗,可能出現死鎖或者請求長期等待返回5XX的問題,要根據場景和需求來做選擇和設計。

本作品採用《CC 協議》,轉載必須註明作者和本文連結

相關文章