123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232 |
- #pragma once
- #include <queue>
- using namespace std;
- template<typename T> class MsgLoop
- {
- size_t m_MaxThreshold;
- HANDLE m_Lock;
- HANDLE m_NotifyInQue;
- deque<T> m_DataQueue;
- public:
- MsgLoop(void)
- {
- m_MaxThreshold = 0;
- m_Lock = CreateMutex(0,0,0);
- m_NotifyInQue = CreateEvent(0,1,0,0);
- };
- ~MsgLoop(void)
- {
- Clear();
- CloseHandle(m_Lock);
- CloseHandle(m_NotifyInQue);
- };
-
- void init(size_t LoopThreshold)
- {
- m_MaxThreshold = LoopThreshold;
- };
- bool Clear()
- {
- ResetEvent(m_NotifyInQue);
- WaitForSingleObject(m_Lock,INFINITE);
- m_DataQueue.clear();
- ReleaseMutex(m_Lock);
- return true;
- };
- MsgLoop& operator = (const MsgLoop &tValue)
- {
- if(this != &tValue)
- {
- WaitForSingleObject(m_Lock, INFINITE);
- WaitForSingleObject(tValue.m_Lock, INFINITE);
- m_DataQueue = tValue.m_DataQueue;
-
- ReleaseMutex(m_Lock);
- ReleaseMutex(tValue.m_Lock);
-
- }
- return (*this);
- };
- void Lock()
- {
- WaitForSingleObject(m_Lock, INFINITE);
- };
- void UnLock()
- {
- ReleaseMutex(m_Lock);
- };
- T* PeekLatest()
- {
- size_t Size = m_DataQueue.size();
- if (Size > 0)
- {
- return &(m_DataQueue.front());
- }
- return NULL;
- }
- T* PeekBackware()
- {
- size_t Size = m_DataQueue.size();
- if (Size > 0)
- {
- if (Size <= m_MaxThreshold)
- {
- return &(m_DataQueue.back());
- }
- else
- {
- return &(m_DataQueue[m_MaxThreshold - 1]);
- }
- }
- return NULL;
- }
- bool InQueueBackward(T &rMsg)
- {
- bool ret = false;
- WaitForSingleObject(m_Lock, INFINITE);
- //printf("Thread:%d,Pop REQ one\n", GetCurrentThreadId());
- m_DataQueue.push_back(rMsg);
- //while (m_DataQueue.size() > (m_MaxThreshold*2))
- //{
- // m_DataQueue.pop_front();
- //}
- ReleaseMutex(m_Lock);
- SetEvent(m_NotifyInQue);
- return ret;
- };
- bool InQueue(T &rMsg)
- {
- bool ret = false;
- WaitForSingleObject(m_Lock, INFINITE);
- //printf("Thread:%d,Pop REQ one\n", GetCurrentThreadId());
- m_DataQueue.push_front(rMsg);
- //while (m_DataQueue.size() > (m_MaxThreshold*2))
- //{
- // m_DataQueue.pop_back();
- //}
- ReleaseMutex(m_Lock);
- SetEvent(m_NotifyInQue);
- return ret;
- };
- void DeQueueLatest()
- {
- WaitForSingleObject(m_Lock, INFINITE);
- if (m_DataQueue.size() > 0)
- {
- m_DataQueue.pop_front();
- }
- ReleaseMutex(m_Lock);
- };
- size_t size()
- {
- size_t Size = 0;
- WaitForSingleObject(m_Lock, INFINITE);
- Size = m_DataQueue.size();
- ReleaseMutex(m_Lock);
- if (Size > m_MaxThreshold)
- {
- Size = m_MaxThreshold;
- }
- return Size;
- };
- HANDLE GetNotifyHandle()
- {
- return m_NotifyInQue;
- };
-
- //must use with lock&unlock method
- T& operator [] (size_t idx)
- {
- if (idx >= m_MaxThreshold)
- {
- assert(0);
- }
- return m_DataQueue[idx];
- };
-
- DWORD WaitForInQue(DWORD timeout)
- {
- DWORD ret = WAIT_TIMEOUT;
- WaitForSingleObject(m_Lock, INFINITE);
- if (m_DataQueue.size() > 0)
- {
- ret = WAIT_OBJECT_0;
- }
- ReleaseMutex(m_Lock);
- if (ret == WAIT_TIMEOUT)
- {
- ret = WaitForSingleObject(m_NotifyInQue, timeout);
- if (ret == WAIT_TIMEOUT)
- {
- return ret;
- }
- //check again
- //make sure data is in the que
- WaitForSingleObject(m_Lock, INFINITE);
- if (m_DataQueue.size() > 0)
- {
- ret = WAIT_OBJECT_0;
- }
- ReleaseMutex(m_Lock);
- }
- return ret;
- };
- };
|