From 554529bb69cd610e83db2c9a80b4f36f5225d80f Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期一, 27 七月 2020 17:56:34 +0800
Subject: [PATCH] restart bus

---
 src/queue/include/shm_queue.h |   29 ++++++++++++++++++++---------
 1 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/src/queue/include/shm_queue.h b/src/queue/include/shm_queue.h
index 81913d0..e64f15a 100644
--- a/src/queue/include/shm_queue.h
+++ b/src/queue/include/shm_queue.h
@@ -21,6 +21,8 @@
 
   ~SHMQueue();
 
+  void force_destroy();
+
   inline uint32_t size();
 
   inline bool full();
@@ -36,7 +38,8 @@
 
   inline ELEM_T &operator[](unsigned i);
 
-  static void remove_queues_exclude(int *keys, size_t length);
+  static void remove_queues_exclude(int keys[], size_t length);
+  static void remove_queues_include(int keys[], size_t length);
 
 private:
 protected:
@@ -50,7 +53,7 @@
 };
 
 template <typename ELEM_T>
-void SHMQueue<ELEM_T>::remove_queues_exclude(int *keys, size_t length) {
+void SHMQueue<ELEM_T>::remove_queues_exclude(int keys[], size_t length) {
   hashtable_t *hashtable = mm_get_hashtable();
   std::set<int> *keyset = hashtable_keyset(hashtable);
   std::set<int>::iterator keyItr;
@@ -65,12 +68,21 @@
       }
     }
     if (!found) {
-      mqueue = (LockFreeQueue<ELEM_T, SHM_Allocator> *)hashtable_get(hashtable,
-                                                                     *keyItr);
-      delete mqueue;
+      // mqueue = (LockFreeQueue<ELEM_T, SHM_Allocator> *)hashtable_get(hashtable, *keyItr);
+      // delete mqueue;
+      hashtable_remove(hashtable, *keyItr);
     }
   }
   delete keyset;
+}
+
+
+template <typename ELEM_T>
+void SHMQueue<ELEM_T>::remove_queues_include(int keys[], size_t length) {
+  hashtable_t *hashtable = mm_get_hashtable();
+  for(int i = 0; i< length; i++) {
+    hashtable_remove(hashtable, keys[i]);
+  }
 }
 
 template <typename ELEM_T>
@@ -84,8 +96,7 @@
     hashtable_put(hashtable, key, (void *)queue);
   }
   queue->reference++;
-  LoggerFactory::getLogger().debug("SHMQueue constructor reference===%d",
-                                   queue->reference.load());
+  LoggerFactory::getLogger().debug("SHMQueue constructor reference===%d", queue->reference.load());
 }
 
 template <typename ELEM_T> SHMQueue<ELEM_T>::~SHMQueue() {
@@ -94,7 +105,7 @@
   // LoggerFactory::getLogger().debug("SHMQueue destructor  reference===%d",
   // queue->reference.load());
   if (queue->reference.load() == 0) {
-    delete queue;
+   // delete queue;
     hashtable_t *hashtable = mm_get_hashtable();
     hashtable_remove(hashtable, KEY);
     // LoggerFactory::getLogger().debug("SHMQueue destructor delete queue\n");
@@ -103,7 +114,7 @@
   }
 }
 
-template <typename ELEM_T> inline uint32_t SHMQueue<ELEM_T>::force_destroy() {
+template <typename ELEM_T> void SHMQueue<ELEM_T>::force_destroy() {
   SemUtil::dec(queue->mutex);
   delete queue;
   hashtable_t *hashtable = mm_get_hashtable();

--
Gitblit v1.8.0