| | |
| | | #define TOPIC_NODE_YVKWA6TF |
| | | |
| | | #include "msg.h" |
| | | #include "socket.h" |
| | | #include "shm_socket.h" |
| | | #include <atomic> |
| | | #include <memory> |
| | | #include <mutex> |
| | |
| | | SharedMemory &shm_; |
| | | ProcInfo info_; |
| | | |
| | | SharedMemory &shm() { return shm_; } |
| | | SharedMemory &shm() const { return shm_; } |
| | | const MQInfo &CenterAddr() const { return BHTopicCenterAddress(shm()); } |
| | | const MQInfo &BusAddr() const { return BHTopicBusAddress(shm()); } |
| | | |
| | | public: |
| | | TopicNode(SharedMemory &shm); |
| | | TopicNode(SharedMemory &shm, MQId ssn_id = 0); |
| | | ~TopicNode(); |
| | | |
| | | // topic node |
| | | bool Register(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms); |
| | | bool Register(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms) { return DoRegister(false, proc, reply_body, timeout_ms); } |
| | | bool DoRegister(const bool internal, ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms); |
| | | bool Unregister(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms); |
| | | bool Heartbeat(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms); |
| | | bool Heartbeat(const int timeout_ms); |
| | | bool QueryTopicAddress(BHAddress &dest, MsgQueryTopic &query, MsgQueryTopicReply &reply_body, const int timeout_ms); |
| | | bool QueryProcs(BHAddress &dest, MsgQueryProc &query, MsgQueryProcReply &reply_body, const int timeout_ms); |
| | | |
| | | // topic rpc server |
| | | typedef std::function<bool(const std::string &client_proc_id, const MsgRequestTopic &request, MsgRequestTopicReply &reply)> ServerSyncCB; |
| | | typedef std::function<void(void *src_info, std::string &client_proc_id, MsgRequestTopic &request)> ServerAsyncCB; |
| | | bool ServerStart(ServerSyncCB const &cb, const int nworker = 2); |
| | | bool ServerStart(ServerAsyncCB const &cb, const int nworker = 2); |
| | | bool ServerRegisterRPC(MsgTopicList &topics, MsgCommonReply &reply, const int timeout_ms); |
| | | bool ServerRegisterRPC(MsgTopicList &topics, MsgCommonReply &reply, const int timeout_ms) { return DoServerRegisterRPC(false, topics, reply, timeout_ms); } |
| | | bool DoServerRegisterRPC(const bool internal, MsgTopicList &topics, MsgCommonReply &reply, const int timeout_ms); |
| | | bool ServerRecvRequest(void *&src_info, std::string &proc_id, MsgRequestTopic &request, const int timeout_ms); |
| | | bool ServerSendReply(void *src_info, const MsgRequestTopicReply &reply); |
| | | |
| | |
| | | bool Init(); |
| | | bool Valid() const { return state() != eStateUninited; } |
| | | std::mutex mutex_; |
| | | MQId ssn_id_ = 0; |
| | | std::atomic<State> state_; |
| | | MQId ssn_id_ = 0; |
| | | int proc_index_ = -1; |
| | | |
| | | TopicQueryCache topic_query_cache_; |
| | | |
| | | class RecvQ |
| | | { |
| | | public: |
| | | void Write(BHMsgHead &&head, std::string &&body) { q_.push_back({std::move(head), std::move(body)}); } |
| | | bool Read(BHMsgHead &head, std::string &body) |
| | | { |
| | | if (q_.empty()) { |
| | | return false; |
| | | } else { |
| | | head = std::move(q_.front().head); |
| | | body = std::move(q_.front().body); |
| | | q_.pop_front(); |
| | | return true; |
| | | } |
| | | } |
| | | |
| | | private: |
| | | struct MsgData { |
| | | BHMsgHead head; |
| | | std::string body; |
| | | }; |
| | | std::deque<MsgData> q_; |
| | | }; |
| | | Synced<RecvQ> server_buffer_; |
| | | Synced<RecvQ> sub_buffer_; |
| | | }; |
| | | |
| | | #endif // end of include guard: TOPIC_NODE_YVKWA6TF |