From 2c73aac1fe62495d8c863a92a34d191cf8e87fc0 Mon Sep 17 00:00:00 2001
From: lichao <lichao@aiotlink.com>
Date: 星期一, 07 六月 2021 10:11:25 +0800
Subject: [PATCH] add readme.

---
 utest/api_test.cpp |  182 ++++++++++++++++++++++++++++++--------------
 1 files changed, 123 insertions(+), 59 deletions(-)

diff --git a/utest/api_test.cpp b/utest/api_test.cpp
index dc3efb6..5363d6e 100644
--- a/utest/api_test.cpp
+++ b/utest/api_test.cpp
@@ -27,6 +27,15 @@
 {
 typedef std::atomic<uint64_t> Number;
 
+void GetApiError(int &ec, std::string &msg)
+{
+	void *pmsg = 0;
+	int msg_len = 0;
+	ec = BHGetLastError(&pmsg, &msg_len);
+	msg.assign((char *) pmsg, msg_len);
+	BHFree(pmsg, msg_len);
+}
+
 void Assign(Number &a, const Number &b) { a.store(b.load()); }
 struct MsgStatus {
 
@@ -102,25 +111,8 @@
 using namespace std::chrono;
 // using namespace std::chrono_literals;
 
-BOOST_AUTO_TEST_CASE(ApiTest)
+bool Register(const std::string &proc_id)
 {
-	auto max_time = std::chrono::steady_clock::time_point::max();
-	auto dur = max_time.time_since_epoch();
-	auto nsec = std::chrono::duration_cast<std::chrono::seconds>(dur).count();
-	auto nmin = nsec / 60;
-	auto nhour = nmin / 60;
-	auto nday = nhour / 24;
-	auto years = nday / 365;
-	printf("seconds: %ld, hours: %ld , days:%ld, years: %ld\n",
-	       nsec, nhour, nday, years);
-	std::chrono::steady_clock::duration a(123456);
-	printf("nowsec: %ld\n", NowSec());
-
-	printf("maxsec: %ld\n", CountSeconds(max_time));
-
-	// BHCleanup();
-	// return;
-	const std::string proc_id = "demo_client";
 	bool reg = false;
 	for (int i = 0; i < 3 && !reg; ++i) {
 		ProcInfo proc;
@@ -138,20 +130,109 @@
 			// if (!reg) {
 			// 	int ec = 0;
 			// 	std::string msg;
-			// 	GetLastError(ec, msg);
+			// 	GetApiError(ec, msg);
 			// 	printf("reg error: %s\n", msg.c_str());
 			// }
 		} else {
 			int ec = 0;
 			std::string msg;
-			GetLastError(ec, msg);
+			GetApiError(ec, msg);
 			printf("register failed, %d, %s\n", ec, msg.c_str());
 		}
 
 		BHFree(reply, reply_len);
 		// Sleep(1s);
 	}
-	if (!reg) {
+	return reg;
+}
+
+bool SyncRequest(const std::string &topic, const std::string &data)
+{ // SyncRequest
+	MsgRequestTopic req;
+	req.set_topic(topic);
+	req.set_data(data);
+	std::string s(req.SerializeAsString());
+	// Sleep(10ms, false);
+	std::string dest(BHAddress().SerializeAsString());
+	void *proc_id = 0;
+	int proc_id_len = 0;
+	DEFER1(BHFree(proc_id, proc_id_len););
+	void *reply = 0;
+	int reply_len = 0;
+	DEFER1(BHFree(reply, reply_len));
+	bool r = BHRequest(dest.data(), dest.size(), s.data(), s.size(), &proc_id, &proc_id_len, &reply, &reply_len, 100);
+	if (!r) {
+		int ec = 0;
+		std::string msg;
+		GetApiError(ec, msg);
+		printf("request error: %d, %s\n", ec, msg.c_str());
+	} else {
+		MsgRequestTopicReply ret;
+		ret.ParseFromArray(reply, reply_len);
+		printf("request result: %s\n", ret.data().c_str());
+	}
+	return r;
+}
+
+BOOST_AUTO_TEST_CASE(RestartTest)
+{
+	const std::string proc_id = "demo_node";
+	for (int i = 0; i < 5; ++i) {
+		printf("loop: %d --------------------------\n", i);
+		if (!Register(proc_id)) { continue; }
+
+		const std::string topic_ = "topic_";
+
+		{ // Server Register Topics
+			MsgTopicList topics;
+			for (int i = 0; i < 10; ++i) {
+				topics.add_topic_list(topic_ + std::to_string(i));
+			}
+			std::string s = topics.SerializeAsString();
+			void *reply = 0;
+			int reply_len = 0;
+			bool r = BHRegisterTopics(s.data(), s.size(), &reply, &reply_len, 1000);
+			DEFER1(BHFree(reply, reply_len));
+		}
+
+		BHStartWorker(&ServerProc, &SubRecvProc, &ClientProc);
+
+		for (int i = 0; i < 60; ++i) {
+			bool r = SyncRequest(topic_ + std::to_string(0), "request_data_" + std::to_string(i));
+			if (!r) {
+				void *msg = 0;
+				int msg_len = 0;
+				DEFER1(BHFree(msg, msg_len));
+				int ec = BHGetLastError(&msg, &msg_len);
+				if (ec == eNotRegistered) {
+					printf("need re-register\n");
+					break;
+				}
+			}
+			Sleep(1s);
+		}
+	}
+	BHCleanup();
+}
+
+BOOST_AUTO_TEST_CASE(ApiTest)
+{
+	auto max_time = std::chrono::steady_clock::time_point::max();
+	auto dur = max_time.time_since_epoch();
+	auto nsec = std::chrono::duration_cast<std::chrono::seconds>(dur).count();
+	auto nmin = nsec / 60;
+	auto nhour = nmin / 60;
+	auto nday = nhour / 24;
+	auto years = nday / 365;
+	printf("seconds: %ld, hours: %ld , days:%ld, years: %ld\n",
+	       nsec, nhour, nday, years);
+	std::chrono::steady_clock::duration a(123456);
+	printf("nowsec: %ld\n", NowSec());
+
+	printf("maxsec: %ld\n", CountSeconds(max_time));
+
+	const std::string proc_id = "demo_client";
+	if (!Register(proc_id)) {
 		return;
 	}
 
@@ -167,8 +248,6 @@
 		int reply_len = 0;
 		bool r = BHRegisterTopics(s.data(), s.size(), &reply, &reply_len, 1000);
 		DEFER1(BHFree(reply, reply_len));
-		// printf("register topic : %s\n", r ? "ok" : "failed");
-		// Sleep(1s);
 	}
 	auto PrintProcs = [](MsgQueryProcReply const &result) {
 		printf("query proc result: %d\n", result.proc_list().size());
@@ -184,7 +263,7 @@
 		}
 		printf("\n");
 	};
-	{
+	if (0) {
 		// query procs
 		std::string dest(BHAddress().SerializeAsString());
 		MsgQueryProc query;
@@ -202,13 +281,22 @@
 		// printf("register topic : %s\n", r ? "ok" : "failed");
 		// Sleep(1s);
 	}
-	{
+	for (int i = 0; i < 3; ++i) {
 		// query procs with normal topic request
 		MsgRequestTopic req;
-		req.set_topic("@center_query_procs");
+		req.set_topic("#center_query_procs");
+		// req.set_data("{\"proc_id\":\"#center.node\"}");
 		std::string s(req.SerializeAsString());
 		// Sleep(10ms, false);
-		std::string dest(BHAddress().SerializeAsString());
+		BHAddress host;
+		printf("query with ip set\n");
+		host.set_ip("127.0.0.1");
+		host.set_port(kBHCenterPort);
+		// center topic node address.
+		// host.set_mq_id(201);
+		// host.set_abs_addr(10072);
+
+		std::string dest(host.SerializeAsString());
 		void *proc_id = 0;
 		int proc_id_len = 0;
 		DEFER1(BHFree(proc_id, proc_id_len););
@@ -219,12 +307,12 @@
 		if (!r) {
 			int ec = 0;
 			std::string msg;
-			GetLastError(ec, msg);
+			GetApiError(ec, msg);
 			printf("topic query proc error: %s\n", msg.c_str());
 		} else {
 			MsgRequestTopicReply ret;
 			ret.ParseFromArray(reply, reply_len);
-			printf("topic query proc : %s\n", ret.data().c_str());
+			printf("\ntopic query proc : %s\n", ret.data().c_str());
 			// MsgQueryProcReply result;
 			// if (result.ParseFromArray(ret.data().data(), ret.data().size()) && IsSuccess(result.errmsg().errcode())) {
 			// 	PrintProcs(result);
@@ -273,36 +361,11 @@
 		}
 	};
 
-	auto SyncRequest = [&](int idx) { // SyncRequest
-		MsgRequestTopic req;
-		req.set_topic(topic_ + std::to_string(0));
-		req.set_data("request_data_" + std::to_string(idx));
-		std::string s(req.SerializeAsString());
-		// Sleep(10ms, false);
-		std::string dest(BHAddress().SerializeAsString());
-		void *proc_id = 0;
-		int proc_id_len = 0;
-		DEFER1(BHFree(proc_id, proc_id_len););
-		void *reply = 0;
-		int reply_len = 0;
-		DEFER1(BHFree(reply, reply_len));
-		bool r = BHRequest(dest.data(), dest.size(), s.data(), s.size(), &proc_id, &proc_id_len, &reply, &reply_len, 1000);
-		if (!r) {
-			int ec = 0;
-			std::string msg;
-			GetLastError(ec, msg);
-			printf("request error: %s\n", msg.c_str());
-		} else {
-			MsgRequestTopicReply ret;
-			ret.ParseFromArray(reply, reply_len);
-			printf("request result: %s\n", ret.data().c_str());
-		}
-	};
 	{
 		for (int i = 0; i < 1; ++i) {
 			MsgPublish pub;
 			pub.set_topic(topic_ + std::to_string(i));
-			pub.set_data("pub_data_" + std::string(1024 * 1, 'a'));
+			pub.set_data("pub_data_" + std::string(104 * 1, 'a'));
 			std::string s(pub.SerializeAsString());
 			BHPublish(s.data(), s.size(), 0);
 			// Sleep(1s);
@@ -331,7 +394,7 @@
 				if (last.exchange(now) < now) {
 					int ec = 0;
 					std::string msg;
-					GetLastError(ec, msg);
+					GetApiError(ec, msg);
 					printf("request topic error --------- : %s\n", msg.c_str());
 				}
 			}
@@ -373,10 +436,10 @@
 	threads.Launch(hb, &run);
 	threads.Launch(showStatus, &run);
 	int ncli = 10;
-	const int64_t nreq = 10; //00 * 100;
+	const int64_t nreq = 1000 * 100;
 
 	for (int i = 0; i < 10; ++i) {
-		SyncRequest(i);
+		SyncRequest(topic_ + std::to_string(0), "request_data_" + std::to_string(i));
 	}
 
 	for (int i = 0; i < ncli; ++i) {
@@ -397,10 +460,11 @@
 		}
 	}
 
+	Sleep(1s);
+
 	run = false;
 	threads.WaitAll();
 	auto &st = Status();
-	Sleep(1s);
 	printf("nreq: %8ld, nsrv: %8ld, nreply: %8ld\n", st.nrequest_.load(), st.nserved_.load(), st.nreply_.load());
 	BHCleanup();
 	printf("after cleanup\n");

--
Gitblit v1.8.0