C++讀寫鎖(stl,boost)
阿新 • • 發佈:2018-11-19
STL 和 Boost 都提供了 shared_mutex
來解決「讀者-寫者」問題。shared_mutex
這個名字並不十分貼切,不如 pthread 直呼「讀寫鎖」。
shared_mutex
比一般的 mutex
多了函式 lock_shared() / unlock_shared()
,允許多個(讀者)執行緒同時加鎖、解鎖,而 shared_lock
則相當於共享版的 lock_guard
。
對 shared_mutex
使用 lock_guard
或 unique_lock
class Counter { public: Counter() : value_(0) { } // Multiple threads/readers can read the counter's value at the same time. std::size_t Get() const { std::shared_lock<std::shared_mutex> lock(mutex_); return value_; } // Only one thread/writer can increment/write the counter's value. void Increase() { // You can also use lock_guard here. std::unique_lock<std::shared_mutex> lock(mutex_); value_++; } // Only one thread/writer can reset/write the counter's value. void Reset() { std::unique_lock<std::shared_mutex> lock(mutex_); value_ = 0; } private: mutable std::shared_mutex mutex_; std::size_t value_; };
假如一個執行緒,先作為讀者用 shared_lock
加鎖,讀完後突然又想變成寫者,該怎麼辦?
方法一:先解讀者鎖,再加寫者鎖。這種做法的問題是,一解一加之間,其他寫者說不定已經介入並修改了資料,那麼當前執行緒作為讀者時所持有的狀態(比如指標、迭代器)也就不再有效。
方法二:用 upgrade_lock
(僅限 Boost,STL 未提供),可以當做 shared_lock
用,但是必要時可以直接從讀者「升級」為寫者。
{ // Acquire shared ownership to read. boost::upgrade_lock<boost::shared_mutex> upgrade_lock(shared_mutex_); // Read... // Upgrade to exclusive ownership to write. boost::upgrade_to_unique_lock<boost::shared_mutex> unique_lock(upgrade_lock); // Write... }