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

1899阅读　0评论2012-06-03　Aquester
分类：C/C++

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

点击(此处)折叠或打开

  1. #ifndef MOOON_AGENT_SEND_MACHINE_H
  2. #define MOOON_AGENT_SEND_MACHINE_H
  3. #include <agent/config.h>
  4. AGENT_NAMESPACE_BEGIN

  6. class CAgentConnector;
  7. class CSendMachine
  8. {
  9. public:
  10.     CSendMachine(CAgentConnector* connector);
  11.     bool is_finish() const;
  12.     util::handle_result_t continue_send();
  13.     util::handle_result_t send(const char* msg, size_t msg_size);
  14.     void reset();
  15.     
  16. private:
  17.     CAgentConnector* _connector;
  18.     
  19. private:
  20.     const char* _cursor;
  21.     size_t _remain_size;
  22. };

  24. AGENT_NAMESPACE_END
  25. #endif // MOOON_AGENT_SEND_MACHINE_H

点击(此处)折叠或打开

  1. #include "send_machine.h"
  2. #include "agent_connector.h"
  3. AGENT_NAMESPACE_BEGIN

  5. CSendMachine::CSendMachine(CAgentConnector* connector)
  6.  :_connector(connector)
  7. {
  8.     reset();
  9. }

  11. // 当前消息是否已经发送完
  12. bool CSendMachine::is_finish() const
  13. {
  14.     return 0 == _remain_size;
  15. }

  17. // 发送消息，可能是一个消息的第一次发送，也可能是一个消息的非第一次发送
  18. util::handle_result_t CSendMachine::continue_send()
  19. {
  20.     ssize_t bytes_sent = _connector->send(_cursor, _remain_size);
  21.     if (bytes_sent > -1)
  22.     {
  23.         _cursor += bytes_sent;
  24.         _remain_size -= bytes_sent;
  25.     }
  26.     
  27.     return is_finish()
  28.          ? util::handle_finish
  29.          : util::handle_continue;
  30. }

  32. // 发送消息，总是一个消息的第一次发送
  33. // 参数说明：
  34. // msg - 需要发送的消息
  35. // msg_size - 需要发送的消息字节数
  36. util::handle_result_t CSendMachine::send(const char* msg, size_t msg_size)
  37. {
  38.     _cursor = msg;
  39.     _remain_size = msg_size;
  40.     
  41.     return continue_send();
  42. }

  44. void CSendMachine::reset()
  45. {
  46.     _cursor = NULL;
  47.     _remain_size = 0;
  48. }

  50. AGENT_NAMESPACE_END

上一篇：mooon-agent接收状态机代码摘要
下一篇：socket的五大误区