Locate_communicator.cpp 7.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
  1. //
  2. // Created by zx on 2020/6/18.
  3. //
  4. #include "Locate_communicator.h"
  5. //#include "locate_message.pb.h"
  6. namespace message {
  7. bool operator<(const message::Measure_request_msg msg1, const message::Measure_request_msg msg2) {
  8. return (msg1.command_id() < msg2.command_id());
  9. }
  10. bool operator==(const message::Measure_request_msg msg1, const message::Measure_request_msg msg2) {
  11. if (msg1.base_info().msg_type() == msg2.base_info().msg_type()
  12. && msg1.base_info().sender() == msg2.base_info().sender()
  13. && msg1.base_info().receiver() == msg2.base_info().receiver()
  14. && msg1.command_id() == msg2.command_id()
  15. && msg1.terminal_id() == msg2.terminal_id()) {
  16. return true;
  17. }
  18. return false;
  19. }
  20. }
  21. Locate_communicator::Locate_communicator()
  22. {
  23. }
  24. Locate_communicator::~Locate_communicator()
  25. {
  26. }
  27. Error_manager Locate_communicator::encapsulate_msg(Communication_message* message)
  28. {
  29. //记录请求
  30. if(message->get_message_type()==Communication_message::eLocate_request_msg)
  31. {
  32. message::Measure_request_msg request;
  33. request.ParseFromString(message->get_message_buf());
  34. m_response_table[request]=message::Measure_response_msg();
  35. //发送请求
  36. return Communication_socket_base::encapsulate_msg(message);
  37. }
  38. else
  39. {
  40. return Error_manager(LOCATER_MSG_TABLE_NOT_EXIST,NEGLIGIBLE_ERROR,"message table is not exist");
  41. }
  42. }
  43. Error_manager Locate_communicator::locate_request(message::Measure_request_msg& request,message::Measure_response_msg& result)
  44. {
  45. /*
  46. * 检查request合法性,以及模块状态
  47. */
  48. if(request.base_info().sender()!=message::eMain||request.base_info().receiver()!=message::eMeasurer)
  49. return Error_manager(LOCATER_MSG_REQUEST_INVALID,MINOR_ERROR,"measure request invalid");
  50. if(m_response_table.find(request)!=m_response_table.end())
  51. return Error_manager(LOCATER_MSG_REQUEST_REPEATED,MAJOR_ERROR," measure reques repeated");
  52. //设置超时,若没有设置,默认3000
  53. int timeout=request.base_info().has_timeout_ms()?request.base_info().timeout_ms():3000;
  54. //向测量节点发送测量请求,并记录请求
  55. Error_manager code;
  56. Communication_message message;
  57. message::Base_info base_msg;
  58. base_msg.set_msg_type(message::eLocate_request_msg);
  59. base_msg.set_sender(message::eMain);
  60. base_msg.set_receiver(message::eMeasurer);
  61. base_msg.set_timeout_ms(timeout);
  62. message.reset(base_msg,request.SerializeAsString());
  63. code=encapsulate_msg(&message);
  64. if(code!=SUCCESS)
  65. return code;
  66. //循环查询请求是否被处理
  67. auto start_time=std::chrono::system_clock::now();
  68. double time=0;
  69. do{
  70. //在请求表中查询结果
  71. std::map<message::Measure_request_msg,message::Measure_response_msg>::iterator it=m_response_table.find(request);
  72. //查询到记录
  73. if(it!=m_response_table.end())
  74. {
  75. message::Measure_response_msg response=it->second;
  76. //判断是否接收到回应,若回应信息被赋值则证明有回应
  77. if (response.has_base_info() && response.has_command_id())
  78. {
  79. message::Base_info response_base = response.base_info();
  80. //检查类型是否匹配
  81. if (response_base.msg_type() != message::eLocate_response_msg) {
  82. return Error_manager(LOCATER_MSG_RESPONSE_TYPE_ERROR, MAJOR_ERROR,
  83. "measure response msg type error");
  84. }
  85. //检查基本信息是否匹配
  86. if (response_base.sender() != message::eMeasurer ||
  87. response_base.receiver() != message::eMain ||
  88. response.command_id() != request.command_id()) {
  89. return Error_manager(LOCATER_MSG_RESPONSE_INFO_ERROR, MAJOR_ERROR,
  90. "measure response msg info error");
  91. }
  92. result = response;
  93. m_response_table.erase(request);
  94. return SUCCESS;
  95. }
  96. }
  97. else
  98. {
  99. //未查询到记录,任务已经被提前取消,记录被删除
  100. return Error_manager(LOCATER_MSG_REQUEST_CANCELED,MINOR_ERROR,"measure request canceled");
  101. }
  102. auto end_time=std::chrono::system_clock::now();
  103. auto duration = std::chrono::duration_cast<std::chrono::microseconds>(end_time - start_time);
  104. time=1000.0*double(duration.count()) * std::chrono::microseconds::period::num / std::chrono::microseconds::period::den;
  105. std::this_thread::yield();
  106. usleep(1000);
  107. }while(time<double(timeout));
  108. return Error_manager(RESPONSE_TIMEOUT,MINOR_ERROR,"measure request timeout");
  109. }
  110. //检查消息是否可以被解析, 需要子类重载
  111. Error_manager Locate_communicator::check_executer(Communication_message* p_msg)
  112. {
  113. return SUCCESS;
  114. }
  115. Error_manager Locate_communicator::check_statu()
  116. {
  117. return SUCCESS;
  118. }
  119. //检查消息是否有效, 主要检查消息类型和接受者, 判断这条消息是不是给我的.
  120. Error_manager Locate_communicator::check_msg(Communication_message* p_msg)
  121. {
  122. //通过 p_msg->get_message_type() 和 p_msg->get_receiver() 判断这条消息是不是给我的.
  123. //子类重载时, 增加自己模块的判断逻辑, 以后再写.
  124. if ( p_msg->get_message_type() == Communication_message::Message_type::eLocate_response_msg
  125. && p_msg->get_receiver() == Communication_message::Communicator::eMain )
  126. {
  127. return Error_code::SUCCESS;
  128. }
  129. else
  130. {
  131. //认为接受人
  132. return Error_code::INVALID_MESSAGE;
  133. }
  134. }
  135. Error_manager Locate_communicator::encapsulate_send_data(){
  136. return SUCCESS;
  137. }
  138. Error_manager Locate_communicator::execute_msg(Communication_message* p_msg)
  139. {
  140. if(p_msg== nullptr)
  141. return Error_manager(POINTER_IS_NULL,CRITICAL_ERROR,"measure response msg pointer is null");
  142. //测量response消息
  143. switch (p_msg->get_message_type())
  144. {
  145. ///测量结果反馈消息
  146. case Communication_message::eLocate_response_msg:
  147. {
  148. message::Measure_response_msg response;
  149. response.ParseFromString(p_msg->get_message_buf());
  150. message::Measure_request_msg request=create_request_by_response(response);
  151. ///查询请求表是否存在
  152. if(m_response_table.find(request)==m_response_table.end())
  153. {
  154. return Error_manager(LOCATER_MSG_RESPONSE_HAS_NO_REQUEST,NEGLIGIBLE_ERROR,"measure response without request");
  155. }
  156. ////更新表
  157. m_response_table[request]=response;
  158. break;
  159. }
  160. ///测量系统状态
  161. case Communication_message::eLocate_status_msg:
  162. {
  163. m_measure_statu_msg=*p_msg;
  164. break;
  165. }
  166. }
  167. return SUCCESS;
  168. }
  169. Error_manager Locate_communicator::cancel_request(message::Measure_request_msg& request)
  170. {
  171. if(m_response_table.find(request)!=m_response_table.end())
  172. m_response_table.erase(request);
  173. return SUCCESS;
  174. }
  175. message::Measure_request_msg Locate_communicator::create_request_by_response(message::Measure_response_msg& response)
  176. {
  177. message::Measure_request_msg request;
  178. message::Base_info baseInfo;
  179. baseInfo.set_msg_type(message::eLocate_request_msg);
  180. baseInfo.set_sender(response.base_info().sender());
  181. baseInfo.set_receiver(response.base_info().receiver());
  182. request.mutable_base_info()->CopyFrom(baseInfo);
  183. request.set_command_id(response.command_id());
  184. request.set_terminal_id(response.command_id());
  185. return request;
  186. }