From c6964d5af25d4ec7ed9dbe7674dc4e3896b36ead Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期五, 16 四月 2021 16:10:02 +0800
Subject: [PATCH] node remove mq if never registered; refactor.
---
src/topic_node.h | 27 ++++++++++++++-------------
1 files changed, 14 insertions(+), 13 deletions(-)
diff --git a/src/topic_node.h b/src/topic_node.h
index d671026..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_; }
@@ -48,12 +48,12 @@
bool ServerStart(ServerCB const &cb, const int nworker = 2);
bool ServerRegisterRPC(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, const int timeout_ms);
+ 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 int timeout_ms, 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