node remove mq if never registered; refactor.

lichao

2021-04-16 c6964d5af25d4ec7ed9dbe7674dc4e3896b36ead

 utest/utest.cpp

@@ -1,6 +1,5 @@
#include "center.h"
#include "defs.h"
#include "failed_msg.h"
#include "util.h"
#include <atomic>
#include <boost/uuid/uuid_generators.hpp>
@@ -21,8 +20,6 @@
struct IsSameType<A, A> {
   static const bool value = true;
};

typedef FailedMsgQ ServerFailedQ;

BOOST_AUTO_TEST_CASE(Temp)
{
@@ -102,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));
@@ -116,7 +113,8 @@
      for (auto &t : topics) {
         tlist.add_topic_list(t);
      }
      bool r = client.Subscribe(tlist, timeout);
      MsgCommonReply reply_body;
      bool r = client.Subscribe(tlist, reply_body, timeout);
      if (!r) {
         printf("client subscribe failed.\n");
      }
@@ -124,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();
@@ -149,8 +147,10 @@

      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) {
            static std::atomic<int> an(0);
            int n = ++an;
@@ -170,7 +170,7 @@
      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);
   }
@@ -218,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());
         }
@@ -227,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;
         }
@@ -240,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());
   };

@@ -248,9 +252,9 @@
   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);
@@ -272,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());
@@ -282,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");