multithreadqueue

template <typename TYPE>
class CHMQ
{
public:
	CHMQ()
	{
		m_nMax = -1;
		m_bDrop = FALSE;

		::InitializeCriticalSection(&m_lock);

		m_hPushEvent = ::CreateEvent(NULL, FALSE, FALSE, NULL);
		//ASSERT(m_hPushEvent != NULL);
		m_hPopEvent = ::CreateEvent(NULL, FALSE, FALSE, NULL);
		//ASSERT(m_hPopEvent != NULL);
	}

	~CHMQ()
	{
		::DeleteCriticalSection(&m_lock);
		::CloseHandle(m_hPushEvent);
		::CloseHandle(m_hPopEvent);
		//m_list.RemoveAll();
	}

	// 队列尾增加消息，如果消息队列已满，根据设定或丢弃消息，或阻塞直到队列不为满的时候。
	void Push(TYPE& type)
	{
		::EnterCriticalSection(&m_lock);
		// 如果消息队列已满
		//if ( m_nMax > 0 && m_list.GetCount() >= m_nMax)
		if ( m_nMax > 0 && m_queue.size() >= m_nMax)
		{
			::ResetEvent(m_hPushEvent);
			::LeaveCriticalSection(&m_lock);
			if( m_bDrop )
				return;

			if ( ::WaitForSingleObject(m_hPushEvent, INFINITE) != WAIT_OBJECT_0)
				//ASSERT(FALSE);
				;
			::EnterCriticalSection(&m_lock);
		}

		//m_list.AddTail(type);
		m_queue.push(type);
		::SetEvent(m_hPopEvent);
		::LeaveCriticalSection(&m_lock);
	}

	//// 队列头增加消息，如果消息队列已满，根据设定或丢弃消息，或阻塞直到队列不为满的时候。
	//void Insert(TYPE& type)
	//{
	//	::EnterCriticalSection(&m_lock);
	//	// 如果消息队列已满
	//	//if ( m_nMax > 0 && m_list.GetCount() >= m_nMax)
	//	if ( m_nMax > 0 && m_queue.size() >= m_nMax)
	//	{
	//		::ResetEvent(m_hPushEvent);
	//		::LeaveCriticalSection(&m_lock);
	//		if( m_bDrop )
	//			return;

	//		if ( ::WaitForSingleObject(m_hPushEvent, INFINITE) != WAIT_OBJECT_0)
	//			ASSERT(FALSE);
	//		::EnterCriticalSection(&m_lock);
	//	}

	//	m_list.AddHead(type);
	//	::SetEvent(m_hPopEvent);
	//	::LeaveCriticalSection(&m_lock);
	//}

	// 从队列头中取消息，如果队列为空则阻塞，直到不为空的时候。
	TYPE Pop()
	{
		TYPE type;
		DWORD id = GetCurrentThreadId();

		::EnterCriticalSection(&m_lock);

		// 如果队列为空
		//if (m_list.IsEmpty())
		if (m_queue.size() == 0)
		{
			::ResetEvent(m_hPopEvent);
			::LeaveCriticalSection(&m_lock);
			if ( ::WaitForSingleObject(m_hPopEvent, INFINITE) != WAIT_OBJECT_0)
				//ASSERT(FALSE);
				;
			::EnterCriticalSection(&m_lock);
		}

		//type = m_list.RemoveHead();
		type = m_queue.front();
		m_queue.pop();
		::SetEvent(m_hPushEvent);
		::LeaveCriticalSection(&m_lock);

		return type;
	}

	// 返回目前消息队列中的消息个数。
	int GetQueueCount()
	{
		int nCount = 0;

		::EnterCriticalSection(&m_lock);
		//nCount = m_list.GetCount();
		nCount = m_queue.size();
		::LeaveCriticalSection(&m_lock);

		return nCount;
	}

	// 设置消息队列的最大长度，以及如果消息队列满的时候新消息的处理方式。
	//或丢弃消息，或阻塞直到队列不为满的时候。
	void SetMaxCount(int nMax = -1, BOOL bDrop = FALSE)
	{
		::EnterCriticalSection(&m_lock);
		m_nMax = nMax;
		m_bDrop = bDrop;
		::LeaveCriticalSection(&m_lock);
	}

	// 返回消息队列的最大长度，如果没有设置最大长度则返回-1。
	int GetMaxCount()
	{
		::EnterCriticalSection(&m_lock);
		::LeaveCriticalSection(&m_lock);
		return m_nMax;
	}


private:
	CRITICAL_SECTION					m_lock;
	//CList								m_list;
	queue<TYPE>								m_queue;

	HANDLE								m_hPopEvent;
	HANDLE								m_hPushEvent;

	int									m_nMax;
	BOOL								m_bDrop;
};