CMessagePush.h 1.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990
  1. #pragma once
  2. #include "../pch/pch.h"
  3. #include "../order/CWaimaiOrder.h"
  4. using boost::asio::ip::tcp;
  5. class CClientMessage
  6. {
  7. public:
  8. int m_type; //消息类型 0:新的外卖订单
  9. std::string m_username;
  10. std::string m_order_id;
  11. std::string m_order_no;
  12. };
  13. class WaimaiPinterInfo
  14. {
  15. public:
  16. std::string order_id;
  17. std::string order_no;
  18. };
  19. class CMessagePush
  20. {
  21. public:
  22. CMessagePush()
  23. : socket_(m_io_context)
  24. {
  25. }
  26. ~CMessagePush();
  27. //开始工作
  28. void Start();
  29. //停止工作
  30. void Stop();
  31. //发送心跳包
  32. void KeepAlive();
  33. //专门处理推送消息
  34. void ReceiveMessage();
  35. //队列处理
  36. void HandleVoice();
  37. void HandleConfirm();
  38. void HandlePrinter();
  39. void HandleShouyinPrinter();
  40. void HandleChufangPrinter();
  41. void AddVoice(std::string order_id);
  42. void AddConfirm(std::string order_id);
  43. void AddPinter(std::string order_id, std::string order_no);
  44. void AddShouyinPrinter(CWaimaiOrder order);
  45. void AddChufangPrinter(CWaimaiOrder order);
  46. private:
  47. void AddStopNum();
  48. private:
  49. bool m_is_work;
  50. int m_nStopNum = 0;
  51. std::mutex m_nStopNumMutex;
  52. boost::asio::io_context m_io_context;
  53. tcp::socket socket_;
  54. enum { max_length = 1024 };
  55. char data_[max_length];
  56. std::queue<std::string> m_voice_queue;
  57. std::queue<std::string> m_confirm_queue;
  58. std::queue<WaimaiPinterInfo> m_printer_queue;
  59. std::queue<CWaimaiOrder> m_shouyin_printer_queue;
  60. std::queue<CWaimaiOrder> m_chufang_printer_queue;
  61. std::mutex m_voice_mutex;
  62. std::mutex m_confirm_mutex;
  63. std::mutex m_printer_mutex;
  64. std::mutex m_shouyin_printer_mutex;
  65. std::mutex m_chufang_printer_mutex;
  66. };