STL实现读写锁

在这里不考虑std::shared_mutex, 只用条件变量和std::mutex来实现。

分析读写锁的逻辑：

读锁之间是不冲突的，无论多少个线程申请读锁，不需要等待。但是要想获得写锁，就必须等所有的读锁全部释放完毕。那么怎么知道所有的读全部结束了呢？只能用一个int变量去记录读锁的总数。这个变量变为0意味着所有的读都结束了。所以，写锁的条件变量的写法类似于：

		while (readerCount != 0)
		{
			cv.wait(lock);
		}

但这是不完善的，如果其他线程也在写呢？写锁也必须等待写操作结束。那么写锁是否也需要int型来统计总数呢？不需要。bool型足矣。因此可以改为：

		while (readerCount != 0 || writerUsed == true)
		{
			cv.wait(lock);
		}

读锁只需要检测是否有写操作，递增写的个数：

		while (writerUsed == true)
		{
			cv.wait(lock);
		}
		readerCount++;

读锁通知和写锁通知的时机也不同。读锁要等所有的读全部结束的时候进行通知，写锁则是每个写结束都要通知。

	void unlockReader()
	{
		std::unique_lock lock(m);
		readerCount--;
		if (readerCount == 0)
		{
			cv.notify_all();
		}
	}
	void unlockWriter()
	{
		std::unique_lock lock(m);
		writerUsed = false;
		cv.notify_all();
	}

所有的代码：

class rwLock
{
public:
	void getReadLock()
	{
		std::unique_lock lock(m);
		while (writerUsed == true)
		{
			cv.wait(lock);
		}
		readerCount++;
	}
	void getWriteLock()
	{
		std::unique_lock lock(m);
		while (readerCount != 0 || writerUsed == true)
		{
			cv.wait(lock);
		}
		writerUsed = true;
	}
	void unlockReader()
	{
		std::unique_lock lock(m);
		readerCount--;
		if (readerCount == 0)
		{
			cv.notify_all();
		}
	}
	void unlockWriter()
	{
		std::unique_lock lock(m);
		writerUsed = false;
		cv.notify_all();
	}

private:
	int readerCount = 0;
	bool writerUsed = false;
	std::mutex m;
	std::condition_variable cv;
};