Fu Juntang
2021-09-03 b386f0c5d35994a54e95d5fe1c4bcbfd1cdea59c
src/net/net_mod_socket.h
@@ -3,6 +3,7 @@
#include "usg_common.h"
#include "shm_mod_socket.h"
#include "socket_io.h"
#include "proc_def.h"
#include <poll.h>
#include "socket_def.h"
#include "net_conn_pool.h"
@@ -17,7 +18,7 @@
   int key;
};
#define NET_MODE_REQUEST_HEAD_LENGTH (NI_MAXHOST + 6 * sizeof(uint32_t))
#define NET_MODE_REQUEST_HEAD_LENGTH sizeof(net_mod_request_head_t)
 
// 请求头
@@ -55,6 +56,14 @@
  
};
struct net_mod_err_t
{
  char host[NI_MAXHOST];
  int port;
  int key;
  int code;
};
class NetModSocket {
 
@@ -83,14 +92,13 @@
  NetConnPool* _get_pool();
  //读取返回信息
  int read_response(int clientfd, net_mod_recv_msg_t *recv_msg);
  int read_response(int clientfd, net_mod_recv_msg_t *recv_msg, net_mod_err_t *err_arr);
  // 发送请求信息
  int write_request(int clientfd, net_mod_request_head_t &request_head, void *send_buf, int send_size, void *topic_buf, int topic_size);
  int write_request(int clientfd, net_mod_request_head_t &request_head, const void *send_buf, int send_size, const void *topic_buf, int topic_size);
  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, int timeout);
  int _pub_(net_node_t *node_arr, int arrlen, char *topic, int topic_size, void *content, int content_size, int timeout) ;
  int _pub_(net_node_t *node_arr, int arrlen, const char *topic, int topic_size, const void *content, int content_size, int timeout) ;
  
public:
@@ -98,7 +106,7 @@
  NetModSocket();
  ~NetModSocket();
  int stop();
  /**
   * 绑定端口到socket, 如果不绑定则系统自动分配一个
   * @return 0 成功, 其他值 失败的错误码
@@ -111,24 +119,9 @@
  */
  int force_bind( int key);
  /**
   * @brief 如果建立连接的节点没有接受到消息等待timeout的时间后返回
   *
   * 向node_arr 中的所有网络节点发送请求消息,节点的返回信息汇总并存储在recv_arr中
   * @node_arr 网络节点组, @node_arr_len该数组长度.如果IP为空则为本地发送。
   * @send_buf 发送的消息,@send_size 该消息体的长度
   * @recv_arr 返回的应答消息组,@recv_arr_size 该数组长度
   * @timeout 等待时间,单位是千分之一秒
   * @return 成功发送的节点的个数
   *
   * 优点:1某个节点的故障不会阻塞其他节点。2 性能好。 3 采用thread local技术即保证了线程安全,又可以使用连接池缓存连接
   */
  int sendandrecv(net_node_t *node_arr, int arrlen, void *send_buf, int send_size,
    net_mod_recv_msg_t ** recv_arr, int *recv_arr_size) ;
  int bind_proc_id(char *buf, int len);
  int reg(void *pData, int len, void **buf, int *size, const int timeout_ms, int flag);
  /**
   * 如果建立连接的节点没有接受到消息等待timeout的时间后返回
   * 向node_arr 中的所有网络节点发送请求消息,节点的返回信息汇总并存储在recv_arr中
@@ -139,15 +132,10 @@
   * @return 成功发送的节点的个数
   * 优点:1某个节点的故障不会阻塞其他节点。2 性能好。 3 采用thread local技术即保证了线程安全,又可以使用连接池缓存连接
   */
  int sendandrecv_timeout(net_node_t *node_arr, int arrlen, void *send_buf, int send_size,
    net_mod_recv_msg_t ** recv_arr, int *recv_arr_size, int  timeout);
  /**
   * 不等待立即返回
   * @timeout 等待时间,单位是千分之一秒
  */
  int sendandrecv_nowait(net_node_t *node_arr, int arrlen, void *send_buf, int send_size,
    net_mod_recv_msg_t ** recv_arr, int *recv_arr_size) ;
  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,
    net_mod_err_t ** _err_arr, int *_err_arr_size, int timeout);
  /**
   * 功能同sendandrecv
@@ -179,7 +167,6 @@
  // 接受信息超时返回。 @sec 秒 , @nsec 纳秒
  int recvfrom_timeout( void **buf, int *size, int *key, int sec, int nsec);
  int recvfrom_nowait( void **buf, int *size, int *key);
  /**
   * 本地发送请求信息并等待接收应答
   * @key 发送给谁
@@ -206,13 +193,13 @@
   * @content 内容,@content_size 内容长度
   * @return 成功发布的节点的个数
   */
  int pub(net_node_t *node_arr, int arrlen, char *topic, int topic_size, void *content, int content_size) ;
  int pub(net_node_t *node_arr, int arrlen, const char *topic, int topic_size, const void *content, int content_size) ;
  int pub_nowait(net_node_t *node_arr, int arrlen, char *topic, int topic_size, void *content, int content_size);
  int pub_nowait(net_node_t *node_arr, int arrlen, const char *topic, int topic_size, const void *content, int content_size);
  /**
   * @msec 毫秒 (千分之一秒)
   */
  int pub_timeout(net_node_t *node_arr, int arrlen, char *topic, int topic_size, void *content, int content_size, int  msec);
  int pub_timeout(net_node_t *node_arr, int arrlen, const char *topic, int topic_size, const void *content, int content_size, int  msec);
  /**
   * 订阅指定主题
   * @topic 主题
@@ -247,10 +234,6 @@
  int  pub_timeout( char *topic, int topic_size, void *content, int content_size, int key, int sec, int nsec);
  int  pub_nowait( char *topic, int topic_size, void *content, int content_size, int key);
  /**
   * 获取soket key