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