From 18705acc13f78ac65458b3dd832545e62fbc9172 Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期一, 08 二月 2021 10:44:29 +0800
Subject: [PATCH] update

---
 src/socket/shm_socket.cpp |  134 ++++++++++++++++++--------------------------
 1 files changed, 54 insertions(+), 80 deletions(-)

diff --git a/src/socket/shm_socket.cpp b/src/socket/shm_socket.cpp
index f55a111..c0ddeee 100644
--- a/src/socket/shm_socket.cpp
+++ b/src/socket/shm_socket.cpp
@@ -75,57 +75,57 @@
 }
 
 //鍒犻櫎鍖呭惈鍦╧eys鍐呯殑queue
-size_t shm_socket_remove_keys(int keys[], size_t length) {
-  hashtable_t *hashtable = mm_get_hashtable();
-  LockFreeQueue<shm_packet_t> *mqueue;
-  size_t count = 0;
-  for(int i = 0; i< length; i++) {
-    // 閿�姣佸叡浜唴瀛樼殑queue
-    mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, keys[i]);
-    delete mqueue;
-    hashtable_remove(hashtable, keys[i]);
-    count++;
-  }
-  return count;
-}
+// size_t shm_socket_remove_keys(int keys[], size_t length) {
+//   hashtable_t *hashtable = mm_get_hashtable();
+//   LockFreeQueue<shm_packet_t> *mqueue;
+//   size_t count = 0;
+//   for(int i = 0; i< length; i++) {
+//     // 閿�姣佸叡浜唴瀛樼殑queue
+//     mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, keys[i]);
+//     delete mqueue;
+//     hashtable_remove(hashtable, keys[i]);
+//     count++;
+//   }
+//   return count;
+// }
 
  
-// 鍒犻櫎涓嶅湪keys鍐呯殑queue 
-size_t shm_socket_remove_keys_exclude(int keys[], size_t length) {
-  hashtable_t *hashtable = mm_get_hashtable();
-  std::set<int> *keyset = hashtable_keyset(hashtable);
-  std::set<int>::iterator keyItr;
-  LockFreeQueue<shm_packet_t> *mqueue;
-  bool found;
-  size_t count = 0;
-  for (keyItr = keyset->begin(); keyItr != keyset->end(); keyItr++) {
-    found = false;
-    for (size_t i = 0; i < length; i++) {
-      if (*keyItr == keys[i]) {
-        found = true;
-        break;
-      }
-    }
-    // 100鍐呯殑鏄痓us鍐呴儴鑷繁鐢ㄧ殑
-    if (!found && *keyItr > 100) {
-      // 閿�姣佸叡浜唴瀛樼殑queue
-      mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, *keyItr);
-      delete mqueue;
-      hashtable_remove(hashtable, *keyItr);
-      count++;
-    }
-  }
-  delete keyset;
-  return count;
-}
+// // 鍒犻櫎涓嶅湪keys鍐呯殑queue 
+// size_t shm_socket_remove_keys_exclude(int keys[], size_t length) {
+//   hashtable_t *hashtable = mm_get_hashtable();
+//   std::set<int> *keyset = hashtable_keyset(hashtable);
+//   std::set<int>::iterator keyItr;
+//   LockFreeQueue<shm_packet_t> *mqueue;
+//   bool found;
+//   size_t count = 0;
+//   for (keyItr = keyset->begin(); keyItr != keyset->end(); keyItr++) {
+//     found = false;
+//     for (size_t i = 0; i < length; i++) {
+//       if (*keyItr == keys[i]) {
+//         found = true;
+//         break;
+//       }
+//     }
+//     // 100鍐呯殑鏄痓us鍐呴儴鑷繁鐢ㄧ殑
+//     if (!found && *keyItr > 100) {
+//       // 閿�姣佸叡浜唴瀛樼殑queue
+//       mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, *keyItr);
+//       delete mqueue;
+//       hashtable_remove(hashtable, *keyItr);
+//       count++;
+//     }
+//   }
+//   delete keyset;
+//   return count;
+// }
 
 
 
-shm_socket_t *shm_open_socket(shm_socket_type_t socket_type) {
+shm_socket_t *shm_socket_open(shm_socket_type_t socket_type) {
   int s, type;
   pthread_mutexattr_t mtxAttr;
 
-  logger->debug("shm_open_socket\n");
+  logger->debug("shm_socket_open\n");
   // shm_socket_t *socket = (shm_socket_t *)calloc(1, sizeof(shm_socket_t));
   shm_socket_t *sockt = new shm_socket_t;
   sockt->socket_type = socket_type;
@@ -151,10 +151,14 @@
   return sockt;
 }
 
-int shm_close_socket(shm_socket_t *sockt) {
+int shm_socket_close(shm_socket_t *sockt) {
+
+}
+
+int _shm_socket_close_(shm_socket_t *sockt) {
   
   int rv;
-  logger->debug("shm_close_socket\n");
+  logger->debug("shm_socket_close\n");
   if(sockt->queue != NULL) {
     delete sockt->queue;
     sockt->queue = NULL;
@@ -162,7 +166,7 @@
 
   rv =  pthread_mutex_destroy(&(sockt->mutex) );
   if(rv != 0) {
-    err_exit(rv, "shm_close_socket");
+    err_exit(rv, "shm_socket_close");
   }
 
   free(sockt);
@@ -320,7 +324,7 @@
     return;
 
   logger->debug("%d destroy tmp socket\n", pthread_self()); 
-  shm_close_socket((shm_socket_t *)tmp_socket);
+  shm_socket_close((shm_socket_t *)tmp_socket);
   rv =  pthread_setspecific(_perthread_socket_key_, NULL);
   if ( rv != 0) {
     logger->error(rv, "shm_sendandrecv : pthread_setspecific");
@@ -453,7 +457,7 @@
   {
     /* If first call from this thread, allocate buffer for thread, and save its location */
     logger->debug("%ld create tmp socket\n", (long)pthread_self() );
-    tmp_socket = shm_open_socket(SHM_SOCKET_DGRAM);
+    tmp_socket = shm_socket_open(SHM_SOCKET_DGRAM);
 
     rv =  pthread_setspecific(_perthread_socket_key_, tmp_socket);
     if ( rv != 0) {
@@ -461,36 +465,6 @@
       exit(1);
     }
   }
-   // int rv;
-  // int tryn = 0;
-  // int recv_key;
-  // rv = shm_sendto(tmp_socket, send_buf, send_size, send_key, timeout, flags);
-
-  // if (() == 0) {
-
-  //   while(tryn < 3) {
-  //     tryn++;
-  //     rv = shm_recvfrom(tmp_socket, recv_buf, recv_size, &recv_key, timeout, flags);
-  //     if(rv != 0) {
-  //       logger->error("_shm_sendandrecv_thread_local : %s\n", bus_strerror(rv));
-  //       return rv;
-  //     }
-     
-  //      // 瓒呮椂瀵艰嚧鎺ュ彂閫佸璞★紝涓庤繑鍥炲璞′笉瀵瑰簲鐨勬儏鍐�
-  //     if(send_key != recv_key) {
-  //       logger->debug("======%d use tmp_socket %d, send to  %d, receive from  %d\n", shm_socket_get_key(sockt), shm_socket_get_key(tmp_socket), send_key, recv_key);
-  //       // logger->error( "_shm_sendandrecv_alloc_new: send key expect to equal to recv key! send key =%d , recv key=%d", send_key, recv_key);
-  //       // exit(1);
-  //       continue;
-  //       // return EBUS_RECVFROM_WRONG_END;
-  //     }
-
-  //     return 0;
-  //   }
-    
-  //   return EBUS_RECVFROM_WRONG_END;
-  // }  
-
  
 
   sendpak.key = tmp_socket->key;
@@ -569,7 +543,7 @@
   shm_socket_t *tmp_socket;
  
  
-  tmp_socket = shm_open_socket(SHM_SOCKET_DGRAM);
+  tmp_socket = shm_socket_open(SHM_SOCKET_DGRAM);
 
   if ((rv = shm_sendto(tmp_socket, send_buf, send_size, send_key, timeout, flags)) == 0) {
     rv = shm_recvfrom(tmp_socket, recv_buf, recv_size, &recv_key, timeout, flags);
@@ -595,7 +569,7 @@
     return EBUS_RECVFROM_WRONG_END;
   } 
    
-  shm_close_socket(tmp_socket);  
+  shm_socket_close(tmp_socket);  
   return rv;
  
 }

--
Gitblit v1.8.0