#ifndef __NET_MODE_SOCKET_H__ #define __NET_MODE_SOCKET_H__ #include "usg_common.h" #include "shm_mod_socket.h" #include "socket_io.h" #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 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); public: NetModSocket(); ~NetModSocket(); /** * 向node_arr 中的所有网络节点发送请求消息,节点的返回信息汇总并存储在recv_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); /** * 销毁sendandrecv方法返回的消息组 * @arr 消息组 * @size 消息组的长度 */ static void free_recv_msg_arr(net_mod_recv_msg_t * arr, size_t size); /** * 向node_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