123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152 |
- #pragma once
- #include "CcosLock.h"
- #include "LinuxEvent.h"
- #include <vector>
- #include <algorithm>
- using namespace std;
- template<typename T> class MsgVector
- {
- CcosLock m_Lock;
- std::shared_ptr<LinuxEvent> m_NotifyInVec;
- vector<T> m_DataVector;
- public:
- MsgVector(void)
- {
- m_NotifyInVec = LinuxEvent::CreateEvent(LinuxEvent::MANUAL_RESET,false);
- };
- ~MsgVector(void)
- {
- Clear();
- };
-
- //suppose it locked already...
- T &operator [](size_t idx)
- {
- return m_DataVector[idx];
- };
-
- //suppose it locked already...
- T &operator [](int idx)
- {
- return m_DataVector[idx];
- };
- void Lock()
- {
- m_Lock.Thread_Lock();
- };
-
- void UnLock()
- {
- m_Lock.Thread_UnLock();
- };
-
- size_t Size()
- {
- Lock();
- size_t size = m_DataVector.size();
- UnLock();
- return size;
- };
- bool Clear()
- {
- m_NotifyInVec->ResetEvent();
- Lock();
- m_DataVector.clear();
- UnLock();
- return true;
- };
- bool PopBack(T &rMsg)
- {
- bool ret = false;
- Lock();
- //DebugPrint("before deQue size:%d\n",m_DataVector.size());
- if (m_DataVector.size() > 0)
- {
- rMsg = m_DataVector.pop_back();
- ret = true;
- }
- //DebugPrint("after deQue size:%d\n",m_DataVector.size());
- if (m_DataVector.size() == 0)
- {
- m_NotifyInVec->ResetEvent();
- }
- UnLock();
- return ret;
- };
- bool FindAndClear(T &rMsg)
- {
- bool ret = false;
- Lock();
- typename vector<T>::iterator result1 = find(m_DataVector.begin(), m_DataVector.end(), rMsg);
- if (result1 != m_DataVector.end())
- {
- m_DataVector.erase(result1);
- ret = true;
- }
- if (m_DataVector.size() == 0)
- {
- m_NotifyInVec->ResetEvent();
- }
- UnLock();
- return ret;
- };
- bool PushBack(T &rMsg)
- {
- Lock();
- m_DataVector.push_back(rMsg);
- UnLock();
- m_NotifyInVec->SetEvent();
- return true;
- };
- DWORD WaitForInVec(DWORD timeout)
- {
- DWORD ret = WAIT_TIMEOUT;
- Lock();
- if (m_DataVector.size() > 0)
- {
- ret = WAIT_OBJECT_0;
- }
- UnLock();
- if (ret == WAIT_TIMEOUT)
- {
- return m_NotifyInVec->Wait(timeout);
- }
- return ret;
- };
- };
|