std::lock_guard/std::unique_lock

时间:2021-11-09 16:34:33

C++多线程编程中通常会对共享的数据进行写保护,以防止多线程在对共享数据成员进行读写时造成资源争抢导致程序出现未定义的行为。通常的做法是在修改共享数据成员的时候进行加锁--mutex。在使用锁的时候通常是在对共享数据进行修改之前进行lock操作,在写完之后再进行unlock操作,进场会出现由于疏忽导致由于lock之后在离开共享成员操作区域时忘记unlock,导致死锁。

针对以上的问题,C++11中引入了std::unique_lock与std::lock_guard两种数据结构。通过对lock和unlock进行一次薄的封装,实现自动unlock的功能。

  1. std::mutex mut;
  2. void insert_data()
  3. {
  4. std::lock_guard<std::mutex> lk(mut);
  5. queue.push_back(data);
  6. }
  7. void process_data()
  8. {
  9. std::unqiue_lock<std::mutex> lk(mut);
  10. queue.pop();
  11. }

std::unique_lock 与std::lock_guard都能实现自动加锁与解锁功能,但是std::unique_lock要比std::lock_guard更灵活,但是更灵活的代价是占用空间相对更大一点且相对更慢一点。

通过实现一个线程安全的队列来说明两者之间的差别。

  template <typename T>

  1. class ThreadSafeQueue{
  2. public:
  3. void Insert(T value);
  4. void Popup(T &value);
  5. bool Empety();
  6. private:
  7. mutable std::mutex mut_;
  8. std::queue<T> que_;
  9. std::condition_variable cond_;
  10. };

  template <typename T>

  1. void ThreadSafeQueue::Insert(T value){
  2. std::lock_guard<std::mutex> lk(mut_);
  3. que_.push_back(value);
  4. cond_.notify_one();
  5. }
  6. template <typename T>
  7. void ThreadSafeQueue::Popup(T &value){
  8. std::unique_lock<std::mutex> lk(mut_);
  9. cond_.wait(lk, [this]{return !que_.empety();});
  10. value = que_.front();
  11. que_.pop();
  12. }
  13. template <typename T>
  14. bool ThreadSafeQueue::Empty() const{
  15. std::lock_guard<std::mutex> lk(mut_);
  16. return que_.empty();
  17. }

上面代码只实现了关键的几个函数,并使用了C++11新引入的condition_variable条件变量。从Popup与Inert两个函数看std::unique_lock相对std::lock_guard更灵活的地方在于在等待中的线程如果在等待期间需要解锁mutex,并在之后重新将其锁定。而std::lock_guard却不具备这样的功能。

上面代码中

可能会比较难以理解,

  [this]{return !Empety();}

是C++11新引入的功能,lambda表达式,是一种匿名函数。方括号内表示捕获变量。当lambda表达式返回true时(即queue不为空),wait函数会锁定mutex。当lambda表达式返回false时,wait函数会解锁mutex同时会将当前线程置于阻塞或等待状态。

还存在另一种读写锁,但是并没有引入C++11,但是boost库提供了对应的实现。读写锁主要适合在于共享数据更新频率较低,但是读取共享数据频率较高的场合。