MsgVector.h 2.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152
  1. #pragma once
  2. #include "CcosLock.h"
  3. #include "LinuxEvent.h"
  4. #include <vector>
  5. #include <algorithm>
  6. using namespace std;
  7. template<typename T> class MsgVector
  8. {
  9. CcosLock m_Lock;
  10. std::shared_ptr<LinuxEvent> m_NotifyInVec;
  11. vector<T> m_DataVector;
  12. public:
  13. MsgVector(void)
  14. {
  15. m_NotifyInVec = LinuxEvent::CreateEvent(LinuxEvent::MANUAL_RESET,false);
  16. };
  17. ~MsgVector(void)
  18. {
  19. Clear();
  20. };
  21. //suppose it locked already...
  22. T &operator [](size_t idx)
  23. {
  24. return m_DataVector[idx];
  25. };
  26. //suppose it locked already...
  27. T &operator [](int idx)
  28. {
  29. return m_DataVector[idx];
  30. };
  31. void Lock()
  32. {
  33. m_Lock.Thread_Lock();
  34. };
  35. void UnLock()
  36. {
  37. m_Lock.Thread_UnLock();
  38. };
  39. size_t Size()
  40. {
  41. Lock();
  42. size_t size = m_DataVector.size();
  43. UnLock();
  44. return size;
  45. };
  46. bool Clear()
  47. {
  48. m_NotifyInVec->ResetEvent();
  49. Lock();
  50. m_DataVector.clear();
  51. UnLock();
  52. return true;
  53. };
  54. bool PopBack(T &rMsg)
  55. {
  56. bool ret = false;
  57. Lock();
  58. //DebugPrint("before deQue size:%d\n",m_DataVector.size());
  59. if (m_DataVector.size() > 0)
  60. {
  61. rMsg = m_DataVector.pop_back();
  62. ret = true;
  63. }
  64. //DebugPrint("after deQue size:%d\n",m_DataVector.size());
  65. if (m_DataVector.size() == 0)
  66. {
  67. m_NotifyInVec->ResetEvent();
  68. }
  69. UnLock();
  70. return ret;
  71. };
  72. bool FindAndClear(T &rMsg)
  73. {
  74. bool ret = false;
  75. Lock();
  76. typename vector<T>::iterator result1 = find(m_DataVector.begin(), m_DataVector.end(), rMsg);
  77. if (result1 != m_DataVector.end())
  78. {
  79. m_DataVector.erase(result1);
  80. ret = true;
  81. }
  82. if (m_DataVector.size() == 0)
  83. {
  84. m_NotifyInVec->ResetEvent();
  85. }
  86. UnLock();
  87. return ret;
  88. };
  89. bool PushBack(T &rMsg)
  90. {
  91. Lock();
  92. m_DataVector.push_back(rMsg);
  93. UnLock();
  94. m_NotifyInVec->SetEvent();
  95. return true;
  96. };
  97. DWORD WaitForInVec(DWORD timeout)
  98. {
  99. DWORD ret = WAIT_TIMEOUT;
  100. Lock();
  101. if (m_DataVector.size() > 0)
  102. {
  103. ret = WAIT_OBJECT_0;
  104. }
  105. UnLock();
  106. if (ret == WAIT_TIMEOUT)
  107. {
  108. return m_NotifyInVec->Wait(timeout);
  109. }
  110. return ret;
  111. };
  112. };