From dc01e4cbb01e96d19b470a366bbe648d426ed171 Mon Sep 17 00:00:00 2001 From: fujuntang <fujuntang@smartai.com> Date: 星期六, 11 九月 2021 10:06:15 +0800 Subject: [PATCH] Add topics sub and request support. --- src/socket/shm_mod_socket.h | 38 ++++++++++++++++++-------------------- 1 files changed, 18 insertions(+), 20 deletions(-) diff --git a/src/socket/shm_mod_socket.h b/src/socket/shm_mod_socket.h index 3795cfb..5e234bf 100644 --- a/src/socket/shm_mod_socket.h +++ b/src/socket/shm_mod_socket.h @@ -3,15 +3,16 @@ #include "usg_common.h" #include "shm_socket.h" #include "shm_allocator.h" -#include "mem_pool.h" +#include "shm_mm.h" #include "hashtable.h" +#include "proc_def.h" #include "sem_util.h" #include "logger_factory.h" #include "key_def.h" #include <set> #include "socket_def.h" -#define BUS_HEAD_SIZE (64 + 2 * sizeof(uint32_t)) +#define BUS_HEAD_SIZE sizeof(bus_head_t) class BusServerSocket; struct bus_head_t @@ -36,8 +37,8 @@ static int get_bus_sendbuf(bus_head_t &request_head, const void *topic_buf, int topic_size, const void *content_buf, int content_size, void **retbuf); public: - static size_t remove_keys(int keys[], size_t length); - static size_t remove_keys_exclude(int keys[], size_t length); + // static size_t remove_keys(int keys[], size_t length); + // static size_t remove_keys_exclude(int keys[], size_t length); // bus header 缂栫爜涓虹綉缁滀紶杈撶殑瀛楄妭 static void * encode_bus_head(bus_head_t & bushead); @@ -60,23 +61,13 @@ * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 */ int force_bind(int key); - /** - * 鍙戦�佷俊鎭� - * @key 鍙戦�佺粰璋� - * @flag BUS_TIMEOUT_FLAG BUS_NOWAIT_FLAG - * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 - */ - - int sendto(const void *buf, const int size, const int key, const struct timespec *timeout = NULL, int flag = 0); - - /** - * 鎺ユ敹淇℃伅 - * @key 浠庤皝鍝噷鏀跺埌鐨勪俊鎭� - * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 - */ - - int recvfrom(void **buf, int *size, int *key, const struct timespec *timeout = NULL, int flag = 0); + int bind_proc_id(char *buf, int len); + int reg(void *pData, int len, void **buf, int *size, const int timeout_ms, int flag); + + int sendto(const void *buf, const int size, const int key, const struct timespec *timeout = NULL, int flag = 0, int reset = 0, int data_set = 0); + + int recvfrom(void **buf, int *size, int *key, const struct timespec *timeout = NULL, int flag = 0, int reset = 0, int data_set = 0); /** * 鍙戦�佽姹備俊鎭苟绛夊緟鎺ユ敹搴旂瓟 @@ -128,7 +119,14 @@ */ int get_key() ; + int get_procid(char *buf, int len); + }; + +typedef std::map<int, ProcInfo, std::less<int>, SHM_STL_Allocator<std::pair<int, ProcInfo> > > ProcZone; +typedef std::set<SHMString, std::less<SHMString>, SHM_STL_Allocator<SHMString> > TcsZone; +typedef std::map<int, TcsZone *, std::less<int>, SHM_STL_Allocator<std::pair<const int, TcsZone *> > > ProcTcsMap; + #endif -- Gitblit v1.8.0