1. 程式人生 > >C++讀寫鎖(stl,boost)

C++讀寫鎖(stl,boost)

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...
}