lichao
2021-06-02 94f2a94f38261528d98a8ece4fcdb386cbca6566
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 {
@@ -138,13 +147,13 @@
         // 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());
      }
@@ -167,8 +176,21 @@
      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);
   }
   { // Server Register Topics
      MsgTopicList topics;
      topics.add_topic_list("@should_fail");
      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));
      if (!r) {
         int ec = 0;
         std::string msg;
         GetApiError(ec, msg);
         printf("register rpc failed, %d, %s\n", ec, msg.c_str());
      }
   }
   auto PrintProcs = [](MsgQueryProcReply const &result) {
      printf("query proc result: %d\n", result.proc_list().size());
@@ -184,7 +206,7 @@
      }
      printf("\n");
   };
   {
   if (0) {
      // query procs
      std::string dest(BHAddress().SerializeAsString());
      MsgQueryProc query;
@@ -202,13 +224,21 @@
      // 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);
      host.set_mq_id(1000011);
      host.set_abs_addr(10296);
      std::string dest(host.SerializeAsString());
      void *proc_id = 0;
      int proc_id_len = 0;
      DEFER1(BHFree(proc_id, proc_id_len););
@@ -219,12 +249,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);
@@ -290,7 +320,7 @@
      if (!r) {
         int ec = 0;
         std::string msg;
         GetLastError(ec, msg);
         GetApiError(ec, msg);
         printf("request error: %s\n", msg.c_str());
      } else {
         MsgRequestTopicReply ret;
@@ -302,7 +332,7 @@
      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 +361,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,7 +403,7 @@
   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);
@@ -397,10 +427,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");