From 9b29a67af93e4ab9948cc60f743d73c4bb39e738 Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期二, 20 十月 2020 16:46:55 +0800
Subject: [PATCH] update

---
 src/socket/net_mod_socket.c |   21 ++++++++++++---------
 1 files changed, 12 insertions(+), 9 deletions(-)

diff --git a/src/socket/net_mod_socket.c b/src/socket/net_mod_socket.c
index df4e4a2..660c1f3 100644
--- a/src/socket/net_mod_socket.c
+++ b/src/socket/net_mod_socket.c
@@ -447,22 +447,24 @@
   int max_buf_size, buf_size;
  
   net_mod_request_head_t request_head;
-  
   char portstr[32];
+  int nsuc = 0;
 
-  buf = (char *)malloc(MAXBUF);
-  max_buf_size = MAXBUF;
-  if(buf == NULL) {
+  if((buf = (char *)malloc(MAXBUF)) == NULL) {
     LoggerFactory::getLogger()->error(errno, "NetModSocket::sendandrecv malloc");
     exit(1);
+  } else {
+     max_buf_size = MAXBUF;
   }
-  int nsuc = 0;
+  
   for (i = 0; i< arrlen; i++) {
 
     node = &node_arr[i];
     if(node->host == NULL) {
       // 鏈湴鍙戦��
-      shmModSocket.pub(topic, topic_size, content, content_size, node->key);
+      if(shmModSocket.pub(topic, topic_size, content, content_size, node->key) == 0 ) {
+         nsuc++;
+      }
      
     } else {
       sprintf(portstr, "%d", node->port);
@@ -493,12 +495,13 @@
  
       if(rio_writen(clientfd, buf, buf_size) != buf_size ) {
         LoggerFactory::getLogger()->error(errno, "NetModSocket::pub rio_writen conent ");
-        close(clientfd);
-        continue;
+      } else {
+        nsuc++;
       }
+      close(clientfd);
     }
 
-    nsuc++;
+   
   }
 
   free(buf);

--
Gitblit v1.8.0