// 一个互斥量、一个条件变量、一个计数变量实现读写锁
class CRWMutex2
{
public:
CRWMutex2()
: m_count_(0)
{
}
virtual ~CRWMutex2() = default;
void LockRead()
{
std::unique_lock<std::mutex> locker(m_mutex_);
while (m_count_ < 0)
m_cond_.wait(locker, [=] {return 0 == m_count_; }); // 有正在写时需要等待
++m_count_;
}
void UnLockRead()
{
m_mutex_.lock();
if (--m_count_ == 0)
m_cond_.notify_one(); // 通知一个写操作
m_mutex_.unlock();
}
void LockWrite()
{
std::unique_lock<std::mutex> locker(m_mutex_);
while (m_count_ != 0)
m_cond_.wait(locker, [=] {return 0 == m_count_; }); // 有正在读或写时需要等待
m_count_ = -1;
}
void UnLockWrite()
{
std::unique_lock<std::mutex> locker(m_mutex_);
m_count_ = 0;
m_cond_.notify_all(); // 通知全部的读写线程
}
private:
std::mutex m_mutex_; // 保护计数资源
volatile int m_count_; // 计数,==0无读写,>0读数量,<0在写
std::condition_variable m_cond_; // 条件变量
};
C++读写锁- 一个互斥量、一个条件变量、一个计数变量实现读写锁
最新推荐文章于 2023-06-27 23:41:35 发布