From d1f89c1eb52a08b97d5be5dd991c5991b6f0bf93 Mon Sep 17 00:00:00 2001
From: liuxiaolong <liuxiaolong@aiotlink.com>
Date: 星期三, 21 四月 2021 16:41:30 +0800
Subject: [PATCH] 修改包名,作为go依赖库

---
 src/topic_node.cpp |   36 +++++++++++++++++++++---------------
 1 files changed, 21 insertions(+), 15 deletions(-)

diff --git a/src/topic_node.cpp b/src/topic_node.cpp
index 16c565b..70247dd 100644
--- a/src/topic_node.cpp
+++ b/src/topic_node.cpp
@@ -32,21 +32,26 @@
 	std::string msg_id;
 };
 
+const int kMqLen = 700;
+
 } // namespace
 
 TopicNode::TopicNode(SharedMemory &shm) :
-    shm_(shm), sock_node_(shm), sock_client_(shm), sock_server_(shm), sock_sub_(shm), registered_(false), registered_ever_(false)
+    shm_(shm), sock_node_(shm), sock_client_(shm, kMqLen), sock_server_(shm, kMqLen), sock_sub_(shm, kMqLen), state_(eStateUnregistered)
 {
 	// recv msgs to avoid memory leak.
 	auto default_ignore_msg = [](ShmSocket &sock, MsgI &imsg, BHMsgHead &head) { return true; };
 	SockNode().Start(default_ignore_msg);
+	SockClient().Start(default_ignore_msg);
+	SockServer().Start(default_ignore_msg);
+	SockSub().Start(default_ignore_msg);
 }
 
 TopicNode::~TopicNode()
 {
 	Stop();
 	SockNode().Stop();
-	if (!registered_ever_) {
+	if (state() == eStateUnregistered) {
 		SockNode().Remove();
 		SockClient().Remove();
 		SockServer().Remove();
@@ -95,9 +100,10 @@
 		          msg.ParseBody(rbody) &&
 		          IsSuccess(rbody.errmsg().errcode());
 		if (ok) {
-			registered_ever_.store(true);
+			state(eStateOnline);
+		} else {
+			state_cas(eStateOnline, eStateOffline);
 		}
-		registered_.store(ok);
 	};
 
 	if (timeout_ms == 0) {
@@ -114,13 +120,13 @@
 		if (r) {
 			CheckResult(reply, reply_head, reply_body);
 		}
-		return IsRegistered();
+		return IsOnline();
 	}
 }
 
 bool TopicNode::Heartbeat(ProcInfo &proc, MsgCommonReply &reply_body, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -153,7 +159,7 @@
 
 bool TopicNode::ServerRegisterRPC(MsgTopicList &topics, MsgCommonReply &reply_body, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -220,7 +226,7 @@
 
 bool TopicNode::ServerRecvRequest(void *&src_info, std::string &proc_id, MsgRequestTopic &request, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -244,7 +250,7 @@
 
 bool TopicNode::ServerSendReply(void *src_info, const MsgRequestTopicReply &body)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -282,7 +288,7 @@
 
 bool TopicNode::ClientAsyncRequest(const MsgRequestTopic &req, std::string &out_msg_id, const RequestResultCB &cb)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -348,7 +354,7 @@
 
 bool TopicNode::ClientSyncRequest(const MsgRequestTopic &request, std::string &out_proc_id, MsgRequestTopicReply &out_reply, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -383,7 +389,7 @@
 
 bool TopicNode::ClientQueryRPCTopic(const Topic &topic, BHAddress &addr, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -425,7 +431,7 @@
 
 bool TopicNode::Publish(const MsgPublish &pub, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -456,7 +462,7 @@
 
 bool TopicNode::Subscribe(MsgTopicList &topics, MsgCommonReply &reply_body, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}
@@ -505,7 +511,7 @@
 
 bool TopicNode::RecvSub(std::string &proc_id, MsgPublish &pub, const int timeout_ms)
 {
-	if (!IsRegistered()) {
+	if (!IsOnline()) {
 		SetLastError(eNotRegistered, "Not Registered.");
 		return false;
 	}

--
Gitblit v1.8.0