From 213c00f5146d9e25b0fd7ab09860185dc4eaabbf Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期三, 14 十月 2020 11:59:11 +0800
Subject: [PATCH] update
---
src/socket/net_mod_socket.h | 94 +++++++++++++++++++++++++++++++++++++++++++----
1 files changed, 86 insertions(+), 8 deletions(-)
diff --git a/src/socket/net_mod_socket.h b/src/socket/net_mod_socket.h
index 7ce98dc..acc6a5d 100644
--- a/src/socket/net_mod_socket.h
+++ b/src/socket/net_mod_socket.h
@@ -3,17 +3,95 @@
#include "usg_common.h"
#include "shm_mod_socket.h"
#include "socket_io.h"
-class NetModSocket {
-private:
- int clientfd;
- rio_t rio;
-public:
- NetModSocket(const char *host, int port);
- ssize_t send(void *buf, size_t size);
- ~NetModSocket();
+#define GET(p) (*(uint32_t *)(p))
+#define PUT(p, val) (*(uint32_t *)(p) = (val))
+
+#define NET_MODE_REQUEST_HEAD_LENGTH 16
+#define NET_MODE_RESPONSE_HEAD_LENGTH 4
+
+class NetModServerSocket;
+
+struct net_node_t
+{
+ const char *host;
+ int port;
+ int key;
};
+struct net_mod_request_head_t {
+ uint32_t mod;
+ uint32_t key;
+ uint32_t content_length;
+ uint32_t topic_length;
+};
+
+struct net_mod_response_head_t {
+ // socket_mod_t mod;
+ // int key;
+ uint32_t content_length;
+};
+
+
+struct net_mod_recv_msg_t
+{
+ char host[128];
+ int port;
+ int key;
+ void *content;
+ int content_length;
+
+};
+
+class NetModSocket {
+ friend class NetModServerSocket;
+private:
+ static std::map<std::string, int> connectionMap;
+ ShmModSocket shmModSocket;
+
+
+ static void * encode_request_head(net_mod_request_head_t & request);
+ static net_mod_request_head_t decode_request_head(void *headbs);
+
+ static void * encode_response_head(net_mod_response_head_t & response);
+ static net_mod_response_head_t decode_response_head(void *_headbs);
+
+ int connect(const char *host, int port);
+ void remove_connect(const char *host, int port);
+
+public:
+
+ NetModSocket();
+ ~NetModSocket();
+
+ /**
+ * 鍚憂ode_arr 涓殑鎵�鏈夌綉缁滆妭鐐瑰彂閫佽姹傛秷鎭紝鑺傜偣鐨勮繑鍥炰俊鎭眹鎬诲苟瀛樺偍鍦╮ecv_arr涓�
+ * @node_arr 缃戠粶鑺傜偣缁�, @node_arr_len璇ユ暟缁勯暱搴�
+ * @send_buf 鍙戦�佺殑娑堟伅锛孈send_size 璇ユ秷鎭綋鐨勯暱搴�
+ * @recv_arr 杩斿洖鐨勫簲绛旀秷鎭粍锛孈recv_arr_size 璇ユ暟缁勯暱搴�
+ * @return 鎴愬姛鍙戦�佺殑鑺傜偣鐨勪釜鏁�
+ */
+ int sendandrecv(net_node_t *node_arr, int node_arr_len, void *send_buf, int send_size,
+ net_mod_recv_msg_t ** recv_arr, int *recv_arr_size);
+
+ /**
+ * 閿�姣乻endandrecv鏂规硶杩斿洖鐨勬秷鎭粍
+ * @arr 娑堟伅缁�
+ * @size 娑堟伅缁勭殑闀垮害
+ */
+ static void free_recv_msg_arr(net_mod_recv_msg_t * arr, size_t size);
+
+ /**
+ * 鍚憂ode_arr 涓殑鎵�鏈夌綉缁滆妭鐐瑰彂甯冩秷鎭�
+ * @node_arr 缃戠粶鑺傜偣缁�, @node_arr_len璇ユ暟缁勯暱搴�
+ * @topic 涓婚锛孈topic_size 璇ヤ富棰樼殑闀垮害
+ * @content 鍐呭锛孈content_size 鍐呭闀垮害
+ * @return 鎴愬姛鍙戝竷鐨勮妭鐐圭殑涓暟
+ */
+ int pub(net_node_t *node_arr, int node_arr_len, char *topic, int topic_size, void *content, int content_size);
+
+
+};
#endif
\ No newline at end of file
--
Gitblit v1.8.0