update

wangzhengquan

2020-10-13 d6a27f15acd08e99841595cece2b3e3e8045491a

 src/socket/net_mod_socket.h

@@ -4,6 +4,14 @@
#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;
@@ -13,9 +21,10 @@


struct net_mod_request_head_t {
   socket_mod_t mod;
   int key;
   uint32_t mod;
   uint32_t key;
   uint32_t content_length;
   uint32_t topic_length;
};

struct net_mod_response_head_t {
@@ -31,26 +40,55 @@
  int port;
  int key;
  void *content;
  uint32_t content_length;
  int content_length;
  
};

class NetModSocket {

  friend class NetModServerSocket;
private:
   static std::map<std::string, rio_t *> 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();
  int sendandrecv(net_node_t *node_arr, int arrlen, void *send_buf, int send_size, 
     net_mod_recv_msg_t ** resp_arr, int *resp_arr_size);
 
  ~NetModSocket();

   static void  free_recv_msg_arr(net_mod_recv_msg_t * arr, size_t size);
  /**
   * 向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