add log

zhangmeng

2022-04-11 f552a0ea788225d8fe80942d6e8bd6500f788393

 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";  //这个是客户端从center订阅的通道

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;
}
@@ -144,6 +97,10 @@
    if (!topics || topics_len <= 0) return false;

    const auto& url = get_url(URLRegTopic);
    if (url.empty()) {
        set_last_error("BHRegisterTopics url empty");
        return false;
    }

    bhome_msg::MsgTopicList mtl;
    if (!mtl.ParseFromArray(topics, topics_len)){
@@ -177,16 +134,19 @@
    if (!topic || topic_len <= 0) return false;

    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());
            }
        }
    if (url.empty()) {
        set_last_error("BHQueryTopicAddress url empty");
        return false;
    }
    // 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);
}
@@ -200,19 +160,23 @@
                 int *reply_len,
                 const int timeout_ms)
{
    if (!query || query_len <= 0) return false;
    // if (!query || query_len <= 0) return false;

    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());
            }
        }
    if (url.empty()) {
        set_last_error("BHQueryProcs url empty");
        return false;
    }

    // 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);
}
@@ -240,14 +204,16 @@
    newPub.set_data(string{(const char*)msgpub, (const size_t)msgpub_len});

    string msg(newPub.SerializeAsString());
    if (pub.topic().empty()) {
        set_last_error("BHPublish topic empty");
        return false;
    }
    auto ret = publish(pub.topic(), msg.data(), msg.size());
    if (ret == 0) return false;
    return true;
}

// 订阅
int BHSubscribeTopics(const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms)
{
static int sub(const string& url, const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms){
    if (!topics || topics_len <= 0) return false;

    bhome_msg::MsgTopicList mtl;
@@ -265,17 +231,33 @@
        // printf("BHSubscribeTopics %s\n", t.c_str());
        subscribe_topic(t);
    }
    return true;
    auto ret = simple_request(get_url(URLSubLocal), topics, topics_len, reply, reply_len, timeout_ms);

    auto ret = simple_request(url, topics, topics_len, reply, reply_len, timeout_ms);
    if (!ret){
        printf("BHSubscribeTopics simple_request failed\n");
    }
    return ret;
}
// 订阅
int BHSubscribeTopics(const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms)
{
    const auto& url = get_url(URLSubLocal);
    if (url.empty()) {
        set_last_error("BHSubscribeTopics url empty");
        return false;
    }
    return sub(url, topics, topics_len, reply, reply_len, timeout_ms);
}
// 订阅网络,不实现
int BHSubscribeNetTopics(const void *topics, const int topics_len, void **reply, int *reply_len, const int timeout_ms)
{
    return BHSubscribeTopics(topics, topics_len, reply, reply_len, timeout_ms);
    const auto& url = get_url(URLSubNet);
    if (url.empty()) {
        set_last_error("BHSubscribeNetTopics url empty");
        return false;
    }

    return sub(url, topics, topics_len, reply, reply_len, timeout_ms);
}

// 读取订阅消息,proc_id暂时没用，返回fake msg
@@ -313,9 +295,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;
@@ -366,26 +348,27 @@
{
    if (!request || request_len <= 0) return false;

    string url{};
    // BHQueryTopicAddress获取proc_id
    string procid{};
    if (!get_proc_id_from_MsgRequestTopic(request, request_len, timeout_ms, &procid)){
        return false;
    if (get_proc_id_from_MsgRequestTopic(request, request_len, timeout_ms, &procid)){
        // PRNTVITAG("get_proc_id_from_MsgRequestTopic failed");
        // return false;
        url = "ipc:///tmp/" + procid;
    }

    auto url("ipc:///tmp/" + procid);

    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());
                // printf("======>> BHRequest use remote address %s\n", url.c_str());
            }
        }
    }
    // 使用procid作为ipc通信
    // printf("BHRequest procid %s\n", procid.c_str());

    if (url.empty()) {
        set_last_error("BHRequest url empty");
        return false;
    }

    int rc = request2(url, request, request_len, reply, reply_len, timeout_ms);
    if (rc < 0) return false;
@@ -407,11 +390,10 @@
    if (ret != 0) return false;

    string procid{};
    if (!get_proc_id_from_MsgRequestTopic(msg.data(), msg.size(), timeout_ms, &procid)){
        return false;
    if (get_proc_id_from_MsgRequestTopic(msg.data(), msg.size(), timeout_ms, &procid)){
        copy_memory(proc_id, proc_id_len, procid.data(), procid.size());
    }

    copy_memory(proc_id, proc_id_len, procid.data(), procid.size());
    copy_memory(request, request_len, msg.data(), msg.size());

    return true;