Optimize the code source.

fujuntang

2021-10-23 bae3a4fd9406635608edf0c0d16c52cf7ca06a66

 src/net/net_mod_socket.cpp

@@ -22,7 +22,7 @@


NetModSocket::~NetModSocket() {
 

}


@@ -46,6 +46,27 @@
  return shmModSocket.force_bind(key);
}

int NetModSocket::reg(void *pData, int len, void **buf, int *size, const int timeout_ms, int flag) {
  
  return shmModSocket.reg(pData, len, buf, size, timeout_ms, flag);
}

void NetModSocket::int_set(int data) {
  int_val = data;
}

void NetModSocket::svr_set(int data) {
  svr_val = data;
}

int NetModSocket::int_get(void) {
  return int_val;
}

int NetModSocket::svr_get(void) {
  return svr_val;
}

// int NetModSocket::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) {
//   return _sendandrecv_(node_arr,  arrlen, send_buf,send_size, recv_arr, recv_arr_size, -1);
@@ -67,7 +88,7 @@

  NetConnPool *mpool = (NetConnPool *)_pool;
  delete mpool;
  logger->debug("destory connPool");

}

 /* One-time key creation function */
@@ -320,6 +341,25 @@
     
}

void NetModSocket::buf_data_set(std::string str, int val) {
  recvbuf.insert({str, val}); 
}

int NetModSocket::buf_data_get(std::string str) {
  
  int i;
  int val = 0;
  std::map<std::string, int>::iterator recvIter;

  recvIter = recvbuf.find(str);
  if(recvIter != recvbuf.end()) {
    
    val = recvIter->second;

  }

  return val;
}

void NetModSocket::free_recv_msg_arr(net_mod_recv_msg_t * arr, size_t size) {

@@ -343,9 +383,6 @@
  return _pub_(node_arr, arrlen, topic, topic_size, content, content_size, msec);
}


// int  pub(char *topic, int topic_size, void *content, int content_size, int port);

int NetModSocket::_pub_(net_node_t *node_arr, int arrlen, const char *topic, int topic_size, const void *content,
 int content_size, int  msec) {
  int i, connfd;
@@ -363,7 +400,7 @@
  net_mod_err_t err_msg;

  // 本地发送
  if(node_arr == NULL || arrlen == 0) {
  if ((node_arr == NULL) || (arrlen == 0)) {
    if(msec == 0) {
      ret = shmModSocket.pub(topic, topic_size, content, content_size, SHM_BUS_KEY,  NULL, BUS_NOWAIT_FLAG);
    } else if(msec > 0) {
@@ -487,20 +524,20 @@
 * @key 发送给谁
 * @return 0 成功， 其他值 失败的错误码
 */
int NetModSocket::sendto(const void *buf, const int size, const int key){
  return shmModSocket.sendto(buf, size, key);
int NetModSocket::sendto(const void *buf, const int size, const int key, int reset, int data_set){
  return shmModSocket.sendto(buf, size, key, 0, 0, reset, data_set);
}

// 发送信息超时返回。 @sec 秒 ， @nsec 纳秒
int NetModSocket::sendto_timeout(const void *buf, const int size, const int key, int sec, int nsec){
int NetModSocket::sendto_timeout(const void *buf, const int size, const int key, int sec, int nsec, int reset, int data_set){
  struct timespec timeout = {sec, nsec};
  return shmModSocket.sendto(buf, size, key, &timeout, BUS_TIMEOUT_FLAG);
  return shmModSocket.sendto(buf, size, key, &timeout, BUS_TIMEOUT_FLAG, reset, data_set);
   
}

// 发送信息立刻返回。
int NetModSocket::sendto_nowait(const void *buf, const int size, const int key){
  return shmModSocket.sendto(buf, size, key, NULL, BUS_NOWAIT_FLAG);
int NetModSocket::sendto_nowait(const void *buf, const int size, const int key, int reset, int data_set){
  return shmModSocket.sendto(buf, size, key, NULL, BUS_NOWAIT_FLAG, reset, data_set);
  
}

@@ -509,23 +546,22 @@
 * @key 从谁哪里收到的信息
 * @return 0 成功， 其他值 失败的错误码
*/
int NetModSocket::recvfrom(void **buf, int *size, int *key) {
int NetModSocket::recvfrom(void **buf, int *size, int *key, int reset, int data_set) {

  return shmModSocket.recvfrom(buf, size, key);
  return shmModSocket.recvfrom(buf, size, key, 0, 0, reset, data_set);
 
}

// 接受信息超时返回。 @sec 秒 ， @nsec 纳秒
int NetModSocket::recvfrom_timeout(void **buf, int *size, int *key, int sec, int nsec){
int NetModSocket::recvfrom_timeout(void **buf, int *size, int *key, int sec, int nsec, int reset, int data_set){
  struct timespec timeout = {sec, nsec};
  return shmModSocket.recvfrom(buf, size, key, &timeout, BUS_TIMEOUT_FLAG);
  return shmModSocket.recvfrom(buf, size, key, &timeout, BUS_TIMEOUT_FLAG, reset, data_set);
  
}

int NetModSocket::recvfrom_nowait(void **buf, int *size, int *key){
  return shmModSocket.recvfrom(buf, size, key, NULL, BUS_NOWAIT_FLAG);
int NetModSocket::recvfrom_nowait(void **buf, int *size, int *key, int reset, int data_set){
  return shmModSocket.recvfrom(buf, size, key, NULL, BUS_NOWAIT_FLAG, reset, data_set);
}


int NetModSocket::recvandsend(recvandsend_callback_fn callback,
                              const struct timespec *timeout , int flag, void * user_data ) {
@@ -764,23 +800,23 @@

  head.mod = ntohl(GET(tmp_ptr));

  tmp_ptr += 4;
  tmp_ptr += sizeof(uint32_t);
  memcpy(head.host, tmp_ptr, sizeof(head.host));

 
  tmp_ptr += sizeof(head.host);
  head.port = ntohl(GET(tmp_ptr));

  tmp_ptr += 4;
  tmp_ptr += sizeof(uint32_t);
  head.key = ntohl(GET(tmp_ptr));

  tmp_ptr += 4;
  tmp_ptr += sizeof(uint32_t);
  head.content_length = ntohl(GET(tmp_ptr));

  tmp_ptr += 4;
  tmp_ptr += sizeof(uint32_t);
  head.topic_length = ntohl(GET(tmp_ptr));

  tmp_ptr += 4;
  tmp_ptr += sizeof(uint32_t);
  head.timeout = ntohl(GET_INT32(tmp_ptr));
 
  return head;