From 4c73fd7179e92bee9cccb65e46823b00f568acb3 Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期五, 22 一月 2021 16:57:34 +0800
Subject: [PATCH] tmp

---
 src/socket/shm_socket.cpp |   16 +++++++---------
 1 files changed, 7 insertions(+), 9 deletions(-)

diff --git a/src/socket/shm_socket.cpp b/src/socket/shm_socket.cpp
index 76e906f..e370c72 100644
--- a/src/socket/shm_socket.cpp
+++ b/src/socket/shm_socket.cpp
@@ -373,9 +373,9 @@
   memcpy(dest.buf, buf, size);
 
  
-  if(flags & SHM_MSG_NOWAIT != 0) {
+  if( (flags & BUS_NOWAIT_FLAG) == BUS_NOWAIT_FLAG) {
     rv = remoteQueue->push_nowait(dest);
-  } else if(timeout != NULL) {
+  } else if((flags & BUS_TIMEOUT_FLAG) == BUS_TIMEOUT_FLAG && timeout != NULL) {
   	rv = remoteQueue->push_timeout(dest, timeout);
   } else {
   	rv = remoteQueue->push(dest);
@@ -393,8 +393,6 @@
       logger->error(rv, "sendto key %d failed", key);
     }
     return rv;
-   
-   
   }
 }
 
@@ -433,9 +431,9 @@
 
   shm_msg_t src;
  
-   if(flags & SHM_MSG_NOWAIT != 0) {
+   if((flags & BUS_NOWAIT_FLAG) == BUS_NOWAIT_FLAG) {
     rv = socket->queue->pop_nowait(src);
-  } else if(timeout != NULL) {
+  } else if((flags & BUS_TIMEOUT_FLAG) == BUS_TIMEOUT_FLAG && timeout != NULL) {
     rv = socket->queue->pop_timeout(src, timeout);
 // printf("0 shm_recvfrom====%d\n", rv);
   } else {
@@ -649,7 +647,7 @@
 
     switch (src.type) {
     case SHM_SOCKET_OPEN:
-      socket->acceptQueue->push_timeout(src, &timeout);
+      socket->acceptQueue->push(src, &timeout, BUS_TIMEOUT_FLAG);
       break;
     case SHM_SOCKET_CLOSE:
       _server_close_conn_to_client(socket, src.key);
@@ -660,7 +658,7 @@
       if (iter != socket->clientSocketMap->end()) {
         client_socket = iter->second;
         // print_msg("_server_run_msg_rev push before", src);
-        client_socket->messageQueue->push_timeout(src, &timeout);
+        client_socket->messageQueue->push(src, &timeout, BUS_TIMEOUT_FLAG);
         // print_msg("_server_run_msg_rev push after", src);
       }
 
@@ -695,7 +693,7 @@
       _client_close_conn_to_server(socket);
       break;
     case SHM_COMMON_MSG:
-      socket->messageQueue->push_timeout(src, &timeout);
+      socket->messageQueue->push(src, &timeout, BUS_TIMEOUT_FLAG);
       break;
     default:
        logger->error( "shm_socket._client_run_msg_rev: undefined message type.");

--
Gitblit v1.8.0