From c46be6db32872bfd7c4010b43526b5e6bc0fa6a5 Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期一, 25 一月 2021 14:04:59 +0800
Subject: [PATCH] update

---
 src/socket/shm_mod_socket.h |   41 ++++++++++++++++++++++++++++++-----------
 1 files changed, 30 insertions(+), 11 deletions(-)

diff --git a/src/socket/shm_mod_socket.h b/src/socket/shm_mod_socket.h
index 5c66fbb..79b3247 100644
--- a/src/socket/shm_mod_socket.h
+++ b/src/socket/shm_mod_socket.h
@@ -11,23 +11,42 @@
 #include <set>
 #include "socket_def.h"
 
+#define BUS_HEAD_SIZE (64 + 2 * sizeof(uint32_t))
+class BusServerSocket;
+
+struct bus_head_t
+{
+	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;
   
 	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 _sub_( char *topic, int size, int key, const struct timespec *timeouts,  int flags);
+	int _pub_( char *topic, int topic_size, void *content, int content_size, int key, const struct timespec *timeouts,  int flags);
 
-	int  _desub_( char *topic, int size, int key, struct timespec *timeout, int flags);
+	int  _desub_( char *topic, int size, int key, const struct timespec *timeouts, 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);
 
 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();
@@ -61,7 +80,7 @@
 	*/
 	int recvfrom(void **buf, int *size, int *key);
 	// 鎺ュ彈淇℃伅瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇
-	int recvfrom_timeout(void **buf, int *size, int *key,  struct timespec *timeout);
+	int recvfrom_timeout(void **buf, int *size, int *key,  const struct timespec *timeout);
 	int recvfrom_nowait(void **buf, int *size, int *key);
 
 	/**
@@ -71,13 +90,13 @@
 	*/
 	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_timeout(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size, const 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_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_timeout(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size, const  struct timespec *timeout) ;
 	int sendandrecv_unsafe_nowait(const void *send_buf, const int send_size, const int key, void **recv_buf, int *recv_size) ;
 
 	/**
@@ -88,7 +107,7 @@
 	 */
 	int  sub(char *topic, int size, int key);
 	// 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇
-	int  sub_timeout(char *topic, int size, int key,  struct timespec *timeout);
+	int  sub_timeout(char *topic, int size, int key,  const struct timespec *timeout);
 	int  sub_nowait(char *topic, int size, int key);
 
 
@@ -100,7 +119,7 @@
 	 */
 	int desub( char *topic, int size, int key);
 	// 瓒呮椂杩斿洖銆� @sec 绉� 锛� @nsec 绾崇
-	int desub_timeout(char *topic, int size, int key, struct timespec *timeout);
+	int desub_timeout(char *topic, int size, int key, const struct timespec *timeout);
 	int desub_nowait(char *topic, int size, int key) ;
 
 	/**
@@ -111,7 +130,7 @@
 	 */
 	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_timeout(char *topic, int topic_size, void *content, int content_size, int key, const  struct timespec *timeout);
 	int  pub_nowait(char *topic, int topic_size, void *content, int content_size, int key);
 
 

--
Gitblit v1.8.0