RpcCommunicator.cc 2.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. #include <functional>
  2. #include <boost/asio.hpp>
  3. #include <boost/lexical_cast.hpp>
  4. #include "Rpc/RpcCommunicator.h"
  5. namespace Egametang {
  6. RpcCommunicator::RpcCommunicator(boost::asio::io_service& service):
  7. ioService(service), socket(service)
  8. {
  9. }
  10. RpcCommunicator::~RpcCommunicator()
  11. {
  12. socket.close();
  13. }
  14. boost::asio::ip::tcp::socket& RpcCommunicator::Socket()
  15. {
  16. return socket;
  17. }
  18. void RpcCommunicator::Stop()
  19. {
  20. }
  21. void RpcCommunicator::RecvMeta(RpcMetaPtr meta, StringPtr message)
  22. {
  23. boost::asio::async_read(socket,
  24. boost::asio::buffer(reinterpret_cast<char*>(meta.get()), sizeof(*meta)),
  25. std::bind(&RpcCommunicator::RecvMessage, this,
  26. meta, message, std::placeholders::_1));
  27. }
  28. void RpcCommunicator::RecvMessage(RpcMetaPtr meta, StringPtr message,
  29. const boost::system::error_code& err)
  30. {
  31. if (err)
  32. {
  33. Stop();
  34. return;
  35. }
  36. message->resize(meta->size, 0);
  37. boost::asio::async_read(socket,
  38. boost::asio::buffer(reinterpret_cast<char*>(&message->at(0)), meta->size),
  39. std::bind(&RpcCommunicator::RecvDone, this,
  40. meta, message, std::placeholders::_1));
  41. }
  42. void RpcCommunicator::RecvDone(RpcMetaPtr meta, StringPtr message,
  43. const boost::system::error_code& err)
  44. {
  45. if (err)
  46. {
  47. Stop();
  48. return;
  49. }
  50. OnRecvMessage(meta, message);
  51. }
  52. void RpcCommunicator::OnRecvMessage(RpcMetaPtr meta, StringPtr message)
  53. {
  54. }
  55. void RpcCommunicator::SendMeta(RpcMetaPtr meta, StringPtr message)
  56. {
  57. boost::asio::async_write(socket,
  58. boost::asio::buffer(reinterpret_cast<char*>(meta.get()), sizeof(*meta)),
  59. std::bind(&RpcCommunicator::SendMessage, this,
  60. meta, message, std::placeholders::_1));
  61. }
  62. void RpcCommunicator::SendMessage(RpcMetaPtr meta, StringPtr message,
  63. const boost::system::error_code& err)
  64. {
  65. if (err)
  66. {
  67. Stop();
  68. return;
  69. }
  70. boost::asio::async_write(socket, boost::asio::buffer(*message),
  71. std::bind(&RpcCommunicator::SendDone, this,
  72. meta, message, std::placeholders::_1));
  73. }
  74. void RpcCommunicator::SendDone(RpcMetaPtr meta, StringPtr message,
  75. const boost::system::error_code& err)
  76. {
  77. if (err)
  78. {
  79. Stop();
  80. return;
  81. }
  82. OnSendMessage(meta, message);
  83. }
  84. void RpcCommunicator::OnSendMessage(RpcMetaPtr meta, StringPtr message)
  85. {
  86. }
  87. } // namespace Egametang