node remove mq if never registered; refactor.

lichao

2021-04-16 c6964d5af25d4ec7ed9dbe7674dc4e3896b36ead

 utest/utest.cpp

@@ -9,6 +9,7 @@
#include <string>
#include <thread>
#include <vector>

using namespace bhome_msg;

template <class A, class B>
@@ -22,6 +23,24 @@

BOOST_AUTO_TEST_CASE(Temp)
{
   const std::string shm_name("ShmTemp");
   ShmRemover auto_remove(shm_name); //remove twice? in case of killed?
   SharedMemory shm(shm_name, 1024 * 1024 * 10);

   typedef std::chrono::steady_clock clock;
   int n = 1000 * 1000;
   std::vector<clock::time_point> tps(n);
   {
      printf("thread switch %d times, ", n);
      boost::timer::auto_cpu_timer timer;
      for (auto &tp : tps) {
         tp = clock::now();
         std::this_thread::yield();
      }
   }
   printf("time: %ld ns\n", (tps.back() - tps.front()).count());
   return;
   // sub topic partial match.
   Topic topics[] = {
       "",
       ".",
@@ -80,7 +99,7 @@
   BHCenter center(shm);
   center.Start();

   std::this_thread::sleep_for(100ms);
   Sleep(100ms);

   std::atomic<uint64_t> total_count(0);
   std::atomic<ptime> last_time(Now() - seconds(1));
@@ -90,8 +109,12 @@
   const int timeout = 1000;
   auto Sub = [&](int id, const std::vector<std::string> &topics) {
      DemoNode client("client_" + std::to_string(id), shm);

      bool r = client.Subscribe(topics, timeout);
      MsgTopicList tlist;
      for (auto &t : topics) {
         tlist.add_topic_list(t);
      }
      MsgCommonReply reply_body;
      bool r = client.Subscribe(tlist, reply_body, timeout);
      if (!r) {
         printf("client subscribe failed.\n");
      }
@@ -99,7 +122,7 @@
      std::condition_variable cv;

      std::atomic<uint64_t> n(0);
      auto OnTopicData = [&](const std::string &proc_id, const std::string &topic, const std::string &data) {
      auto OnTopicData = [&](const std::string &proc_id, const MsgPublish &pub) {
         ++total_count;

         auto cur = Now();
@@ -124,10 +147,14 @@

      for (unsigned i = 0; i < nmsg; ++i) {
         std::string data = topic + std::to_string(i) + std::string(1000, '-');

         bool r = provider.Publish(topic, data.data(), data.size(), timeout);
         MsgPublish pub;
         pub.set_topic(topic);
         pub.set_data(data);
         bool r = provider.Publish(pub, 0);
         if (!r) {
            printf("pub ret: %s\n", r ? "ok" : "fail");
            static std::atomic<int> an(0);
            int n = ++an;
            printf("pub %d ret: %s\n", n, r ? "ok" : "fail");
         }
      }
   };
@@ -138,11 +165,12 @@
      topics.push_back("t" + std::to_string(i));
   }
   Topics part;
   boost::timer::auto_cpu_timer pubsub_timer;
   for (size_t i = 0; i < topics.size(); ++i) {
      part.push_back(topics[i]);
      threads.Launch(Sub, i, topics);
   }
   std::this_thread::sleep_for(100ms);
   Sleep(100ms);
   for (auto &topic : topics) {
      threads.Launch(Pub, topic);
   }
@@ -190,8 +218,8 @@

      std::atomic<int> count(0);
      std::string reply;
      auto onRecv = [&](const std::string &rep) {
         reply = rep;
      auto onRecv = [&](const BHMsgHead &head, const MsgRequestTopicReply &msg) {
         reply = msg.data();
         if (++count >= nreq) {
            printf("count: %d\n", count.load());
         }
@@ -199,7 +227,11 @@
      client.ClientStartWorker(onRecv, 2);
      boost::timer::auto_cpu_timer timer;
      for (int i = 0; i < nreq; ++i) {
         if (!client.ClientAsyncRequest(topic, "data " + std::to_string(i), 1000)) {
         MsgRequestTopic req;
         req.set_topic(topic);
         req.set_data("data " + std::to_string(i));
         std::string msg_id;
         if (!client.ClientAsyncRequest(req, msg_id)) {
            printf("client request failed\n");
            ++count;
         }
@@ -212,7 +244,7 @@
      do {
         std::this_thread::yield();
      } while (count.load() < nreq);
      client.StopAll();
      client.Stop();
      printf("request %s %d done ", topic.c_str(), count.load());
   };

@@ -220,19 +252,19 @@
   auto Server = [&](const std::string &name, const std::vector<std::string> &topics) {
      DemoNode server(name, shm);

      auto onData = [&](const std::string &topic, const std::string &data, std::string &reply) {
      auto onData = [&](const std::string &proc_id, const MsgRequestTopic &request, MsgRequestTopicReply &reply) {
         ++server_msg_count;
         reply = topic + ':' + data;
         reply.set_data(request.topic() + ':' + request.data());
         return true;
      };
      server.ServerStart(onData);

      MsgRegisterRPC rpc;
      MsgTopicList rpc;
      for (auto &topic : topics) {
         rpc.add_topics(topic);
         rpc.add_topic_list(topic);
      }
      MsgCommonReply reply_body;
      if (!server.RegisterRPC(rpc, reply_body, 100)) {
      if (!server.ServerRegisterRPC(rpc, reply_body, 100)) {
         printf("server register topic failed\n");
         return;
      }
@@ -244,9 +276,9 @@
   ThreadManager clients, servers;
   std::vector<Topic> topics = {"topic1", "topic2"};
   servers.Launch(Server, "server", topics);
   std::this_thread::sleep_for(100ms);
   Sleep(100ms);
   for (auto &t : topics) {
      clients.Launch(Client, t, 1000 * 1);
      clients.Launch(Client, t, 1000 * 100);
   }
   clients.WaitAll();
   printf("clients done, server replyed: %ld\n", server_msg_count.load());
@@ -254,6 +286,35 @@
   servers.WaitAll();
}

BOOST_AUTO_TEST_CASE(HeartbeatTest)
{
   const std::string shm_name("ShmHeartbeat");
   ShmRemover auto_remove(shm_name);
   SharedMemory shm(shm_name, 1024 * 1024 * 50);

   BHCenter center(shm);
   center.Start();

   {

      DemoNode node("demo_node", shm);
      auto Check = [&]() {
         bool r = node.Heartbeat(100);
         printf("hearbeat ret : %s\n", r ? "ok" : "failed");
      };
      Check();
      for (int i = 0; i < 3; ++i) {
         Sleep(1s);
         Check();
      }
      Sleep(4s);
      for (int i = 0; i < 2; ++i) {
         Sleep(1s);
         Check();
      }
   }
   Sleep(8s);
}
inline int MyMin(int a, int b)
{
   printf("MyMin\n");