且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

mooon-agent发送状态机代码摘要

更新时间:2022-09-21 20:41:53

发送状态机比接收状态机相对要简单

  • send_machine.h

  1. #ifndef MOOON_AGENT_SEND_MACHINE_H  
  2.  
  3. #define MOOON_AGENT_SEND_MACHINE_H  
  4.  
  5. #include <agent/config.h>  
  6.  
  7. AGENT_NAMESPACE_BEGIN  
  8.  
  9.  
  10.  
  11. class CAgentConnector;  
  12.  
  13. class CSendMachine  
  14.  
  15. {  
  16.  
  17. public:  
  18.  
  19.     CSendMachine(CAgentConnector* connector);  
  20.  
  21.     bool is_finish() const;  
  22.  
  23.     util::handle_result_t continue_send();  
  24.  
  25.     util::handle_result_t send(const char* msg, size_t msg_size);  
  26.  
  27.     void reset();  
  28.  
  29.       
  30.  
  31. private:  
  32.  
  33.     CAgentConnector* _connector;  
  34.  
  35.       
  36.  
  37. private:  
  38.  
  39.     const char* _cursor;  
  40.  
  41.     size_t _remain_size;   
  42.  
  43. };  
  44.  
  45.  
  46.  
  47. AGENT_NAMESPACE_END  
  48.  
  49. #endif // MOOON_AGENT_SEND_MACHINE_H  
  • send_machine.cpp

  1. #include "send_machine.h" 
  2.  
  3. #include "agent_connector.h" 
  4.  
  5. AGENT_NAMESPACE_BEGIN  
  6.  
  7.  
  8.  
  9. CSendMachine::CSendMachine(CAgentConnector* connector)  
  10.  
  11.  :_connector(connector)   
  12.  
  13. {  
  14.  
  15.     reset();  
  16.  
  17. }  
  18.  
  19.  
  20.  
  21. // 当前消息是否已经发送完  
  22.  
  23. bool CSendMachine::is_finish() const  
  24.  
  25. {  
  26.  
  27.     return 0 == _remain_size;  
  28.  
  29. }  
  30.  
  31.  
  32.  
  33. // 发送消息,可能是一个消息的第一次发送,也可能是一个消息的非第一次发送  
  34.  
  35. util::handle_result_t CSendMachine::continue_send()  
  36.  
  37. {  
  38.  
  39.     ssize_t bytes_sent = _connector->send(_cursor, _remain_size);  
  40.  
  41.     if (bytes_sent > -1)  
  42.  
  43.     {  
  44.  
  45.         _cursor += bytes_sent;  
  46.  
  47.         _remain_size -= bytes_sent;  
  48.  
  49.     }  
  50.  
  51.       
  52.  
  53.     return is_finish()   
  54.  
  55.          ? util::handle_finish   
  56.  
  57.          : util::handle_continue;  
  58.  
  59. }  
  60.  
  61.  
  62.  
  63. // 发送消息,总是一个消息的第一次发送  
  64.  
  65. // 参数说明:  
  66.  
  67. // msg - 需要发送的消息  
  68.  
  69. // msg_size - 需要发送的消息字节数  
  70.  
  71. util::handle_result_t CSendMachine::send(const char* msg, size_t msg_size)  
  72.  
  73. {  
  74.  
  75.     _cursor = msg;  
  76.  
  77.     _remain_size = msg_size;  
  78.  
  79.       
  80.  
  81.     return continue_send();  
  82.  
  83. }  
  84.  
  85.  
  86.  
  87. void CSendMachine::reset()  
  88.  
  89. {  
  90.  
  91.     _cursor = NULL;  
  92.  
  93.     _remain_size = 0;  
  94.  
  95. }  
  96.  
  97.  
  98.  
  99. AGENT_NAMESPACE_END  


本文转自eyjian 51CTO博客,原文链接:http://blog.51cto.com/mooon/910295,如需转载请自行联系原作者