message_queue.cpp 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. #include <atomic>
  2. #include <string>
  3. #include "../common/comm.h"
  4. #include "api.h"
  5. #include "message_queue.h"
  6. #include "../common/iobuffer.h"
  7. #include "../common/peer_connection.h"
  8. #include <json/json.h>
  9. #include "protocol.pb.h"
  10. #include <fstream>
  11. #include <cstring>
  12. CMessageQueue::CMessageQueue():_head(nullptr),_tail(nullptr)
  13. {
  14. _ping = false;
  15. }
  16. CMessageQueue::~CMessageQueue()
  17. {
  18. }
  19. void CMessageQueue::Start()
  20. {
  21. _thread = std::thread(&CMessageQueue::Run, this);
  22. }
  23. void CMessageQueue::Stop()
  24. {
  25. _run = false;
  26. _thread.join();
  27. }
  28. void CMessageQueue::EnQueue(CIOBuffer* pBuffer)
  29. {
  30. bool bNullBuffer=false;
  31. std::unique_lock <std::mutex> lck(_lock);
  32. if(_head ==nullptr)
  33. {
  34. _head =_tail=pBuffer;
  35. bNullBuffer=true;
  36. }
  37. else{
  38. _tail->NextBuf=pBuffer;
  39. _tail = _tail->NextBuf;
  40. }
  41. pBuffer->NextBuf=nullptr;
  42. if(bNullBuffer)
  43. {
  44. _cv.notify_one();
  45. }
  46. }
  47. void CMessageQueue::Run()
  48. {
  49. _run = true;
  50. CIOBuffer* ptr = nullptr;
  51. while(_run)
  52. {
  53. std::unique_lock <std::mutex> lck(_lock);
  54. /*
  55. while(Head==nullptr)
  56. {
  57. _cv.wait(lck);
  58. }
  59. */
  60. while(_head==nullptr&&_cv.wait_for(lck,std::chrono::seconds(1))==std::cv_status::timeout)
  61. {
  62. OnIdle();
  63. // CheckSignal();
  64. // std::cout<<".";
  65. // std::cout.flush();
  66. }
  67. while (_head != nullptr)
  68. {
  69. ptr = _head;
  70. _head = _head->NextBuf;
  71. if (ptr != nullptr)
  72. {
  73. Process(ptr);
  74. ptr->Release();
  75. }
  76. }
  77. }
  78. }