update

wangzhengquan

2020-10-19 af85260254bacac40a68d4f5f61950523beb3a27

 src/socket/net_mod_socket.h

@@ -7,8 +7,11 @@
#define GET(p)       (*(uint32_t *)(p))
#define PUT(p, val)  (*(uint32_t *)(p) = (val))

#define NET_MODE_REQUEST_HEAD_LENGTH 12
#define NET_MODE_RESPONSE_HEAD_LENGTH 4




class NetModServerSocket;

struct net_node_t
{
@@ -17,16 +20,21 @@
   int key;
};

#define NET_MODE_REQUEST_HEAD_LENGTH 16

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

#define NET_MODE_RESPONSE_HEAD_LENGTH 8

struct net_mod_response_head_t {
   // socket_mod_t mod;
   // int key;
  uint32_t code;
   uint32_t content_length;
};

@@ -42,28 +50,90 @@
};

class NetModSocket {
  struct pool{ /* Represents a pool of connected descriptors */ //line:conc:echoservers:beginpool
    int maxfd;        /* Largest descriptor in read_set */
    fd_set read_set;  /* Set of all active descriptors */
    fd_set ready_read_set; /* Subset of descriptors ready for reading  */

    fd_set write_set;
    fd_set ready_write_set;

    fd_set except_set;
    fd_set ready_except_set;

    int nready;       /* Number of ready descriptors from select */
    int maxi;         /* Highwater index into client array */
    int connfd[FD_SETSIZE];    /* Set of active descriptors */
    // net_node_t *nodes[FD_SETSIZE];
   // rio_t clientrio[FD_SETSIZE]; /* Set of active read buffers */
    std::map<int, net_node_t*> connfdNodeMap;
    std::map<std::string, int> connectionMap;
  } ; 

  friend class NetModServerSocket;
private:
   static std::map<std::string, rio_t *> connectionMap;
   ShmModSocket shmModSocket;
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);
   
  ShmModSocket shmModSocket;
  pool req_resp_pool;
 
  ~NetModSocket();

  static void  free_recv_msg_arr(net_mod_recv_msg_t * arr, size_t size);

  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);
  static void * encode_response_head(net_mod_response_head_t & response);
  static net_mod_response_head_t  decode_response_head(void *_headbs);

  void init_req_rep_req_resp_pool();
  int connect( net_node_t*);
  void close_connect(int connfd);
  int read_response(int clientfd, net_mod_recv_msg_t *recv_msg);
  int write_request(int clientfd, int key, void *send_buf, int send_size);


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
   * 优点：线程安全
   * 缺点：阻塞的，性能不如sendandrecv
   * 
   */
  int sendandrecv_safe(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