From f7c4658c884629e79a9ce98052e19f8a68abdf22 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期五, 14 一月 2022 09:40:13 +0800
Subject: [PATCH] performance

---
 src/bn_api.cpp |  107 +++++++++++++++++++++--------------------------------
 1 files changed, 42 insertions(+), 65 deletions(-)

diff --git a/src/bn_api.cpp b/src/bn_api.cpp
index 292ef4b..90e74ca 100644
--- a/src/bn_api.cpp
+++ b/src/bn_api.cpp
@@ -1,62 +1,16 @@
 #include "bn_api.h"
 
 #include <string.h>
-#include <string>
-#include <unordered_map>
-#include <memory>
-#include <thread>
-using namespace std;
 
 #include "nng_wrap.h"
 using namespace nng_wrap;
 
+#include "common.h"
+using namespace std;
+
 #include "bhome_msg.pb.h"
 #include "bhome_msg_api.pb.h"
 using namespace bhome_msg;
-
-enum{
-    URLReg,
-    URLDeReg,
-    URLRegTopic,
-    URLQueryTopic,
-    URLQueryProcs,
-    URLSubLocal,
-    URLSubNet,
-    URLPubProxy,
-    URLSubQueue,
-    URLHeartBeat,
-};
-
-static char* IPC_REGISTER = (char*)"ipc:///tmp/bhnng-center-reg.ipc"; //杩涚▼娉ㄥ唽
-static char* IPC_UNREGISTER = (char*)"ipc:///tmp/bhnng-center-unregister.ipc"; //娉ㄩ攢
-static char* IPC_REGTOPIC = (char*)"ipc:///tmp/bhnng-center-regtopic.ipc"; //娉ㄥ唽涓婚
-static char* IPC_QUERYTOPIC = (char*)"ipc:///tmp/bhnng-center-querytopic.ipc"; //鏌ヨ鎸囧畾鐨勪富棰�
-static char* IPC_QUERYPROC = (char*)"ipc:///tmp/bhnng-center-queryproc.ipc"; //鏌ヨ鎵�鏈夋敞鍐岀殑杩涚▼
-static char* IPC_SUBLOCALTOPIC = (char*)"ipc:///tmp/bhnng-center-sublocaltopic.ipc"; //璁㈤槄鏈湴涓婚
-static char* IPC_SUBNETTOPIC = (char*)"ipc:///tmp/bhnng-center-subnettopic.ipc"; //璁㈤槄缃戠粶涓婚
-static char* IPC_HEARTBEAT = (char*)"ipc:///tmp/bhnng-center-hb.ipc";
-static char* IPC_PUB_PROXY = (char*)"ipc:///tmp/bhnng-center-pub-proxy.ipc";   //杩欎釜鏄唬鐞嗕腑蹇冿紝鐢ㄤ簬鎺ユ敹寰呭彂甯冪殑娑堟伅
-static char* IPC_SUB_QUEUE = (char*)"ipc:///tmp/bhnng-center-sub-queue.ipc";  //杩欎釜鏄鎴风浠巆enter璁㈤槄鐨勯�氶亾
-
-static const unordered_map<int, string> map_url{
-    {URLReg,                IPC_REGISTER},
-    {URLDeReg,              IPC_UNREGISTER},
-    {URLRegTopic,           IPC_REGTOPIC},
-    {URLQueryTopic,         IPC_QUERYTOPIC},
-    {URLQueryProcs,         IPC_QUERYPROC},
-    {URLSubLocal,           IPC_SUBLOCALTOPIC},
-    {URLSubNet,             IPC_SUBNETTOPIC},
-    {URLPubProxy,           IPC_PUB_PROXY},
-    {URLSubQueue,           IPC_SUB_QUEUE},
-    {URLHeartBeat,          IPC_HEARTBEAT},
-};
-static string get_url(const int type){
-    auto iter = map_url.find(type);
-    if (iter != map_url.end()){
-        return iter->second;
-    }
-    return {};
-}
 
 static void fetch_set_proc_id(std::string& pid){
     static string cur_proc_id{};
@@ -66,8 +20,6 @@
         pid = cur_proc_id;
     }
 }
-
-/////////////////////////////////////////////
 
 int BHRegister(const void *proc_info, const int proc_info_len, void **reply, int *reply_len, const int timeout_ms)
 {
@@ -90,6 +42,7 @@
         unsigned short port = *(unsigned short*)(out);
         copy_memory(reply, reply_len, (char*)out + port_size, out_len - port_size);
         BHFree(out, out_len);
+        printf("======>> recv port %d\n", port);
 /////////////////////////////////////////////////////////////////////////
 
         const auto& url_pub_proxy = get_url(URLPubProxy);
@@ -100,7 +53,7 @@
         const auto& url_hb = get_url(URLHeartBeat);
         respond_survey(url_hb, string{(const char*)proc_info, (const size_t)proc_info_len});
 
-        start_reply(pi.proc_id(), 0);
+        start_reply(pi.proc_id(), port);
     }
     return ret;
 }
@@ -176,7 +129,19 @@
 {
     if (!topic || topic_len <= 0) return false;
 
-    return simple_request(get_url(URLQueryTopic), topic, topic_len, reply, reply_len, timeout_ms);
+    auto url(get_url(URLQueryTopic));
+
+    // if (remote && remote_len > 0){
+    //     BHAddress addr;
+    //     if (addr.ParseFromArray(remote, remote_len)){
+    //         if (!addr.ip().empty() && addr.port() > 0){
+    //             // url = "tcp://" + addr.ip() + ":" + to_string(addr.port());
+    //             printf("======>> BHQueryTopicAddress use remote address %s\n", url.c_str());
+    //         }
+    //     }
+    // }
+
+    return simple_request(url, topic, topic_len, reply, reply_len, timeout_ms);
 }
 
 // 璇锋眰鍦ㄧ嚎杩涚▼ request
@@ -188,11 +153,21 @@
                  int *reply_len,
                  const int timeout_ms)
 {
-    if (!query || query_len <= 0) return false;
+    // if (!query || query_len <= 0) return false;
 
-    auto ret = simple_request(get_url(URLQueryProcs), query, query_len, reply, reply_len, timeout_ms);
-    // printf("======>> BHQueryProcs *reply %p reply_len %d\n", *reply, *reply_len);
-    return ret;
+    auto url(get_url(URLQueryProcs));
+
+    // if (remote && remote_len > 0){
+    //     BHAddress addr;
+    //     if (addr.ParseFromArray(remote, remote_len)){
+    //         if (!addr.ip().empty() && addr.port() > 0){
+    //             // url = "tcp://" + addr.ip() + ":" + to_string(addr.port());
+    //             printf("======>> BHQueryProcs use remote address %s\n", url.c_str());
+    //         }
+    //     }
+    // }
+
+    return simple_request(url, query, query_len, reply, reply_len, timeout_ms);
 }
 
 // above communicate with center
@@ -291,9 +266,9 @@
     // return true;
 }
 
-static int get_proc_id_from_MsgRequestTopic(
-    const void* request, const int request_len, const int timeout_ms,
-    string* proc_id){
+int get_proc_id_from_MsgRequestTopic(
+  const void* request, const int request_len, const int timeout_ms, string* proc_id)
+{
 
     // BHQueryTopicAddress鑾峰彇proc_id
     bhome_msg::MsgRequestTopic req;
@@ -352,11 +327,13 @@
 
     auto url("ipc:///tmp/" + procid);
 
-    BHAddress addr;
-    if (addr.ParseFromArray(remote, remote_len)){
-        if (!addr.ip().empty() && addr.port() > 0){
-            url = "tcp://" + addr.ip() + ":" + to_string(addr.port());
-            printf("======>>use remote address %s\n", url.c_str());
+    if (remote && remote_len > 0){
+        BHAddress addr;
+        if (addr.ParseFromArray(remote, remote_len)){
+            if (!addr.ip().empty() && addr.port() > 0){
+                // url = "tcp://" + addr.ip() + ":" + to_string(addr.port());
+                printf("======>> BHRequest use remote address %s\n", url.c_str());
+            }
         }
     }
     // 浣跨敤procid浣滀负ipc閫氫俊

--
Gitblit v1.8.0