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_SECTIONm_lock;//CListm_list;queue<TYPE>m_queue;HANDLEm_hPopEvent;HANDLEm_hPushEvent;intm_nMax;BOOLm_bDrop;};