condition_variable条件变量的简单使用

来源：微信公众号「编程学习基地」

condition_variable条件变量

简单使用

#include <iostream>           // std::cout
#include <string>
#include <thread>             // std::thread
#include <chrono>             // std::chrono::seconds
#include <mutex>              // std::mutex, std::unique_lock
#include <condition_variable> // std::condition_variable, std::cv_status
using namespace std;
class WaitMessage
{
public:
	WaitMessage(int messageType = 0, int llSize = 1024)
		: kSessionAvailable(0)
		, kSessionResults(nullptr)
	{

	}

	~WaitMessage()
	{
	}

public:
	inline void InitResult(void* lpRequest = NULL, int available = 0)
	{
		std::unique_lock <std::mutex> _lock(kSessionLock);
		kSessionResults = nullptr;
		kSessionAvailable = available;
	}
	inline void PostResult(void* lpResults, int available = 1)
	{
		std::unique_lock <std::mutex> _lock(kSessionLock);
		kSessionResults = lpResults;
		kSessionAvailable = available;
		cout << "post info" << endl;
		kSessionSignal.notify_one();
	}

	inline void* WaitResult()
	{
		std::unique_lock <std::mutex> _lock(kSessionLock);
		while (kSessionAvailable == 0) {
			cout << "start write" << endl;
			kSessionSignal.wait(_lock);
			cout << "run continue" << endl;
		}
		return kSessionResults;
	}

	inline void* WaitResult(int ms)
	{
		std::unique_lock <std::mutex> _lock(kSessionLock);
		while (kSessionAvailable == 0) {
			cout << ms << endl;
			//执行一个时间段,在收到唤醒通知或者超时之前，阻塞等待，收到唤醒通知或者超时后 wait_for 返回
			std::cv_status staue = kSessionSignal.wait_for(_lock, std::chrono::milliseconds(ms));
			if (staue == std::cv_status::timeout) {
				cout << "timeout" << endl;
				break;
			}
			else if(staue == std::cv_status::no_timeout)
			{
				cout << "no_timeout" << endl;
			}
		}
		cout << "continue run" << endl;
		return kSessionResults;
	}
	std::mutex				kSessionLock;
	std::condition_variable kSessionSignal;
	void* kSessionResults;
	int 					kSessionAvailable;
};

WaitMessage m_WaitMessage;

void do_wait()
{
	cout << "run do_wite thread" << endl;
#if 0
	m_WaitMessage.WaitResult();
#else
	void* result = m_WaitMessage.WaitResult(3 * 1000);

	if (result == NULL) {
		cout << "error result is null、" << endl;
		return;
	}
	string str = (char*)result;
	cout << "recv return:" << str << endl;
#endif
}

void do_post()
{
	string str = "do_post string";
	cout << "run do_post thread" << endl;
	while (true)
	{
		//_sleep(4000);
		_sleep(2000);
		m_WaitMessage.PostResult((void*)str.c_str());
	}
}

int main()
{
	m_WaitMessage.InitResult(NULL);

	std::thread wait_thread(do_wait);
	std::thread post_thread(do_post);
	post_thread.join();
	wait_thread.join();
}