From 87b5445e2a08a18bbbe1e0ce63d5f65e81bfb79f Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期一, 18 一月 2021 15:09:30 +0800 Subject: [PATCH] update --- src/socket/shm_mod_socket.h | 59 ++++++++++++++++++++--------------------------------------- 1 files changed, 20 insertions(+), 39 deletions(-) diff --git a/src/socket/shm_mod_socket.h b/src/socket/shm_mod_socket.h index bca34c7..73c91c3 100644 --- a/src/socket/shm_mod_socket.h +++ b/src/socket/shm_mod_socket.h @@ -9,55 +9,44 @@ #include "logger_factory.h" #include "key_def.h" #include <set> +#include "socket_def.h" -#define ACTION_LIDENTIFIER "<**" -#define ACTION_RIDENTIFIER "**>" -#define TOPIC_LIDENTIFIER "{" -#define TOPIC_RIDENTIFIER "}" +#define BUS_HEAD_SIZE (64 + 2 * sizeof(uint32_t)) +class BusServerSocket; - -//typedef std::basic_string<char, std::char_traits<char>, SHM_STL_Allocator<char> > SHMString; -typedef std::set<int, std::less<int>, SHM_STL_Allocator<int> > SHMKeySet; -typedef std::map<SHMString, SHMKeySet *, std::less<SHMString>, SHM_STL_Allocator<std::pair<const SHMString, SHMKeySet *> > > SHMTopicSubMap; - -enum socket_mod_t +struct bus_head_t { - PULL_PUSH = 1, - REQ_REP = 2, - PAIR = 3, - PUB_SUB = 4, - SURVEY = 5, - BUS = 6 - + char action[64]; + uint32_t topic_size; + uint32_t content_size; }; + class ShmModSocket { +friend class BusServerSocket; private: shm_socket_t *shm_socket; - socket_mod_t mod; - // pthread_t recv_thread; - // <涓婚锛� 璁㈤槄鑰�> - SHMTopicSubMap *topic_sub_map; + std::set<int> *bus_set; private: - inline int _recvfrom_(void **buf, int *size, int *key, struct timespec *timeout, int flags); - void _proxy_sub( char *topic, int key); - void _proxy_pub( char *topic, size_t head_len, void *buf, size_t size, int key); - void *run_pubsub_proxy(); - int parse_pubsub_topic(char *str, size_t size, char **_action, char **_topic, size_t *head_len ); + int _sub_( char *topic, int size, int key, struct timespec *timeout, int flags); int _pub_( char *topic, int topic_size, void *content, int content_size, int key, struct timespec *timeout, int flags); - void _proxy_desub( char *topic, int key); - void _proxy_desub_all(int key); int _desub_( char *topic, int size, int key, struct timespec *timeout, int flags); - static void foreach_subscripters(std::function<void(SHMKeySet *, int)> cb); - static bool include_in_keys(int key, int keys[], size_t length); - static size_t remove_subscripters(int keys[], size_t length) ; + + static int get_bus_sendbuf(bus_head_t &request_head, void *topic_buf, int topic_size, void *content_buf, int content_size, void **retbuf); + public: static size_t remove_keys(int keys[], size_t length); + + // bus header 缂栫爜涓虹綉缁滀紶杈撶殑瀛楄妭 + static void * encode_bus_head(bus_head_t & bushead); + // 瑙g爜 bus header + static bus_head_t decode_bus_head(void *headbs); + public: ShmModSocket(); ~ShmModSocket(); @@ -109,14 +98,6 @@ // 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇 int sendandrecv_unsafe_timeout(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size, struct timespec *timeout) ; int sendandrecv_unsafe_nowait(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size) ; - - - /** - * 鍚姩bus - * - * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 - */ - int start_bus(); /** * 璁㈤槄鎸囧畾涓婚 -- Gitblit v1.8.0