From 58d904a328c0d849769b483e901a0be9426b8209 Mon Sep 17 00:00:00 2001
From: liuxiaolong <liuxiaolong@aiotlink.com>
Date: 星期二, 20 七月 2021 20:20:44 +0800
Subject: [PATCH] 调整Request C.BHFree的位置

---
 src/topic_node.h |   24 ++++++++++++++++++------
 1 files changed, 18 insertions(+), 6 deletions(-)

diff --git a/src/topic_node.h b/src/topic_node.h
index dcc9518..9e7eed2 100644
--- a/src/topic_node.h
+++ b/src/topic_node.h
@@ -34,14 +34,17 @@
 	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);
@@ -53,7 +56,8 @@
 	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);
 
@@ -69,7 +73,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, MsgCommonReply &reply_body, const int timeout_ms);
+	bool Subscribe(MsgTopicList &topics, MsgCommonReply &reply_body, const int timeout_ms)
+	{
+		return DoSubscribe(topics, false, reply_body, timeout_ms);
+	}
+	bool SubscribeNet(MsgTopicList &topics, MsgCommonReply &reply_body, const int timeout_ms)
+	{
+		return DoSubscribe(topics, true, reply_body, timeout_ms);
+	}
+	bool DoSubscribe(MsgTopicList &topics, const bool net, MsgCommonReply &reply_body, const int timeout_ms);
 	bool RecvSub(std::string &proc_id, MsgPublish &pub, const int timeout_ms);
 
 	void Start(ServerAsyncCB const &server_cb, SubDataCB const &sub_cb, RequestResultCB &client_cb, int nworker = 2);
@@ -159,8 +171,8 @@
 	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_;

--
Gitblit v1.8.0