From c64c54d8e75b9354dc49a7b6b2d326e7dd59eb37 Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期四, 15 四月 2021 19:32:16 +0800 Subject: [PATCH] add api; fix send, socknode mem leak. --- src/topic_node.h | 14 ++++++++------ 1 files changed, 8 insertions(+), 6 deletions(-) diff --git a/src/topic_node.h b/src/topic_node.h index 0627930..8c3c48e 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 @@ -118,7 +118,9 @@ auto &SockClient() { return SockRequest(); } auto &SockReply() { return sock_reply_; } auto &SockServer() { return SockReply(); } + bool IsRegistered() const { return registered_.load(); } + std::atomic<bool> registered_; ShmSocket sock_node_; ShmSocket sock_request_; ShmSocket sock_reply_; -- Gitblit v1.8.0