From ad4f3dcedab29a690c5eedbb08ba1b393917db0b Mon Sep 17 00:00:00 2001 From: lichao <lichao@aiotlink.com> Date: 星期三, 21 四月 2021 17:39:34 +0800 Subject: [PATCH] update go api. --- src/topic_node.h | 24 +++++++++++++++--------- 1 files changed, 15 insertions(+), 9 deletions(-) diff --git a/src/topic_node.h b/src/topic_node.h index 8287b4a..b2fae5b 100644 --- a/src/topic_node.h +++ b/src/topic_node.h @@ -22,6 +22,7 @@ #include "socket.h" #include <atomic> #include <memory> +#include <vector> using namespace bhome_shm; using namespace bhome_msg; @@ -107,17 +108,22 @@ }; // some sockets may be the same one, using functions make it easy to change. + enum { eSockStart, + eSockNode = eSockStart, + eSockPub = eSockNode, + eSockServer, + eSockClient, + eSockSub, + eSockEnd, + }; + std::vector<std::unique_ptr<ShmSocket>> sockets_; - ShmSocket &SockNode() { return sock_node_; } - ShmSocket &SockPub() { return SockNode(); } - ShmSocket &SockSub() { return sock_sub_; } - ShmSocket &SockClient() { return sock_client_; } - ShmSocket &SockServer() { return sock_server_; } + ShmSocket &SockNode() { return *sockets_[eSockNode]; } + ShmSocket &SockPub() { return *sockets_[eSockPub]; } + ShmSocket &SockSub() { return *sockets_[eSockSub]; } + ShmSocket &SockClient() { return *sockets_[eSockClient]; } + ShmSocket &SockServer() { return *sockets_[eSockServer]; } - ShmSocket sock_node_; - ShmSocket sock_client_; - ShmSocket sock_server_; - ShmSocket sock_sub_; enum State { eStateUnregistered, eStateOnline, -- Gitblit v1.8.0