From f52f2c2828047c2f30d30fc1fe2b54d8db146d49 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期四, 25 二月 2021 15:56:35 +0800 Subject: [PATCH] update --- src/socket/shm_mod_socket.h | 65 +++++++++++++------------------- 1 files changed, 26 insertions(+), 39 deletions(-) diff --git a/src/socket/shm_mod_socket.h b/src/socket/shm_mod_socket.h index 4c82497..0c65f52 100644 --- a/src/socket/shm_mod_socket.h +++ b/src/socket/shm_mod_socket.h @@ -3,7 +3,7 @@ #include "usg_common.h" #include "shm_socket.h" #include "shm_allocator.h" -#include "mem_pool.h" +#include "shm_mm.h" #include "hashtable.h" #include "sem_util.h" #include "logger_factory.h" @@ -26,23 +26,18 @@ friend class BusServerSocket; private: shm_socket_t *shm_socket; - socket_mod_t mod; std::set<int> *bus_set; private: - inline int _recvfrom_(void **buf, int *size, int *key, struct timespec *timeout, int flags); - 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); + - int _desub_( char *topic, int size, int key, struct timespec *timeout, int flags); - - - static int get_bus_sendbuf(bus_head_t &request_head, void *topic_buf, int topic_size, void *content_buf, int content_size, void **retbuf); + 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(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); @@ -53,6 +48,7 @@ ShmModSocket(); ~ShmModSocket(); + int stop(); /** * 缁戝畾绔彛鍒皊ocket, 濡傛灉涓嶇粦瀹氬垯绯荤粺鑷姩鍒嗛厤涓�涓� * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 @@ -67,50 +63,45 @@ /** * 鍙戦�佷俊鎭� * @key 鍙戦�佺粰璋� + * @flag BUS_TIMEOUT_FLAG BUS_NOWAIT_FLAG * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 */ - int sendto(const void *buf, const int size, const int key); - // 鍙戦�佷俊鎭秴鏃惰繑鍥炪�� @sec 绉� 锛� @nsec 绾崇 - int sendto_timeout(const void *buf, const int size, const int key, const struct timespec *timeout); - // 鍙戦�佷俊鎭珛鍒昏繑鍥炪�� - int sendto_nowait(const void *buf, const int size, const int key); + + 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); - // 鎺ュ彈淇℃伅瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇 - int recvfrom_timeout(void **buf, int *size, int *key, struct timespec *timeout); - int recvfrom_nowait(void **buf, int *size, int *key); + + int recvfrom(void **buf, int *size, int *key, const struct timespec *timeout = NULL, int flag = 0); /** * 鍙戦�佽姹備俊鎭苟绛夊緟鎺ユ敹搴旂瓟 * @key 鍙戦�佺粰璋� + * @flag BUS_TIMEOUT_FLAG BUS_NOWAIT_FLAG * @return 0 鎴愬姛锛� 鍏朵粬鍊� 澶辫触鐨勯敊璇爜 */ - int sendandrecv(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size) ; - // 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇 - int sendandrecv_timeout(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size, struct timespec *timeout) ; - int sendandrecv_nowait(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size) ; + + int sendandrecv(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size, + const struct timespec *timeout = NULL, int flag = 0); - int sendandrecv_unsafe(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size) ; - // 瓒呮椂杩斿洖銆� @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) ; + /** + * + */ + int recvandsend( recvandsend_callback_fn callback, const struct timespec *timeout = NULL , int flag = 0, void * user_data = NULL); /** * 璁㈤槄鎸囧畾涓婚 * @topic 涓婚 * @size 涓婚闀垮害 * @key 鎬荤嚎绔彛 + * @flag BUS_TIMEOUT_FLAG BUS_NOWAIT_FLAG */ - int sub(char *topic, int size, int key); - // 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇 - int sub_timeout(char *topic, int size, int key, struct timespec *timeout); - int sub_nowait(char *topic, int size, int key); + int sub(const char *topic, int size, int key, const struct timespec *timeout = NULL, int flag = 0); /** @@ -118,22 +109,18 @@ * @topic 涓婚,涓婚涓虹┖鏃跺彇娑堝叏閮ㄨ闃� * @size 涓婚闀垮害 * @key 鎬荤嚎绔彛 + * @flag BUS_TIMEOUT_FLAG BUS_NOWAIT_FLAG */ - int desub( char *topic, int size, int key); - // 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇 - int desub_timeout(char *topic, int size, int key, struct timespec *timeout); - int desub_nowait(char *topic, int size, int key) ; + int desub(const char *topic, int size, int key, const struct timespec *timeout = NULL, int flag = 0); /** * 鍙戝竷涓婚 * @topic 涓婚 * @content 涓婚鍐呭 * @key 鎬荤嚎绔彛 + * @flag BUS_TIMEOUT_FLAG BUS_NOWAIT_FLAG */ - int pub(char *topic, int topic_size, void *content, int content_size, int key); - // 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇 - int pub_timeout(char *topic, int topic_size, void *content, int content_size, int key, struct timespec *timeout); - int pub_nowait(char *topic, int topic_size, void *content, int content_size, int key); + int pub(const char *topic, int topic_size, const void *content, int content_size, int key, const struct timespec *timeout = NULL, int flag = 0); /** -- Gitblit v1.8.0