From 708ff9e8af731e2799767ed8bfca7df3b74fc26a Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期五, 16 四月 2021 19:20:21 +0800 Subject: [PATCH] sendq use less shm, copy data. --- src/topic_node.h | 25 +++++++++++++------------ 1 files changed, 13 insertions(+), 12 deletions(-) diff --git a/src/topic_node.h b/src/topic_node.h index 0627930..5a3b86e 100644 --- a/src/topic_node.h +++ b/src/topic_node.h @@ -29,7 +29,7 @@ class TopicNode { SharedMemory &shm_; - MsgRegister info_; + ProcInfo info_; SharedMemory &shm() { return shm_; } @@ -51,9 +51,9 @@ bool ServerSendReply(void *src_info, const MsgRequestTopicReply &reply); // topic client - typedef std::function<void(const std::string &proc_id, const MsgRequestTopicReply &reply)> RequestResultCB; + typedef std::function<void(const BHMsgHead &head, const MsgRequestTopicReply &reply)> RequestResultCB; bool ClientStartWorker(RequestResultCB const &cb, const int nworker = 2); - bool ClientAsyncRequest(const MsgRequestTopic &request, const RequestResultCB &rrcb = RequestResultCB()); + bool ClientAsyncRequest(const MsgRequestTopic &request, std::string &msg_id, const RequestResultCB &rrcb = RequestResultCB()); bool ClientSyncRequest(const MsgRequestTopic &request, std::string &proc_id, MsgRequestTopicReply &reply, const int timeout_ms); // publish @@ -62,15 +62,15 @@ // subscribe typedef std::function<void(const std::string &proc_id, const MsgPublish &data)> SubDataCB; bool SubscribeStartWorker(const SubDataCB &tdcb, int nworker = 2); - bool Subscribe(MsgTopicList &topics, const int timeout_ms); + bool Subscribe(MsgTopicList &topics, MsgCommonReply &reply_body, const int timeout_ms); bool RecvSub(std::string &proc_id, MsgPublish &pub, const int timeout_ms); - void Start(ServerCB const &server_cb, SubDataCB const &sub_cb); + void Start(ServerCB const &server_cb, SubDataCB const &sub_cb, RequestResultCB &client_cb, int nworker = 2); void Stop(); private: bool ClientQueryRPCTopic(const Topic &topic, bhome::msg::BHAddress &addr, const int timeout_ms); - const std::string &proc_id() { return info_.proc().proc_id(); } + const std::string &proc_id() { return info_.proc_id(); } typedef bhome_msg::BHAddress Address; class TopicQueryCache @@ -114,15 +114,16 @@ auto &SockNode() { return sock_node_; } auto &SockPub() { return SockNode(); } auto &SockSub() { return sock_sub_; } - auto &SockRequest() { return sock_request_; } - auto &SockClient() { return SockRequest(); } - auto &SockReply() { return sock_reply_; } - auto &SockServer() { return SockReply(); } + auto &SockClient() { return sock_client_; } + auto &SockServer() { return sock_server_; } + bool IsRegistered() const { return registered_.load(); } ShmSocket sock_node_; - ShmSocket sock_request_; - ShmSocket sock_reply_; + ShmSocket sock_client_; + ShmSocket sock_server_; ShmSocket sock_sub_; + std::atomic<bool> registered_; + std::atomic<bool> registered_ever_; TopicQueryCache topic_query_cache_; }; -- Gitblit v1.8.0