From f0732bbcbdca67ef8f9444f2aafaf44da81c9e5a Mon Sep 17 00:00:00 2001
From: wangzhengquan <wangzhengquan85@126.com>
Date: 星期六, 01 八月 2020 16:44:00 +0800
Subject: [PATCH] update
---
src/queue/include/shm_queue.h | 28 +++++++++++++++++++---------
1 files changed, 19 insertions(+), 9 deletions(-)
diff --git a/src/queue/include/shm_queue.h b/src/queue/include/shm_queue.h
index 99f4802..512a913 100644
--- a/src/queue/include/shm_queue.h
+++ b/src/queue/include/shm_queue.h
@@ -38,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:
@@ -52,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;
@@ -67,12 +68,24 @@
}
}
if (!found) {
- mqueue = (LockFreeQueue<ELEM_T, SHM_Allocator> *)hashtable_get(hashtable,
- *keyItr);
+ 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();
+ LockFreeQueue<ELEM_T, SHM_Allocator> *mqueue;
+ for(int i = 0; i< length; i++) {
+ mqueue = (LockFreeQueue<ELEM_T, SHM_Allocator> *)mm_get_by_key(keys[i]);
+ delete mqueue;
+ hashtable_remove(hashtable, keys[i]);
+ }
}
template <typename ELEM_T>
@@ -80,26 +93,23 @@
hashtable_t *hashtable = mm_get_hashtable();
queue = (LockFreeQueue<ELEM_T, SHM_Allocator> *)hashtable_get(hashtable, key);
- // LockFreeQueue<int, 10000> q;
if (queue == NULL || (void *)queue == (void *)1) {
queue = new LockFreeQueue<ELEM_T, SHM_Allocator>(qsize);
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() {
SemUtil::dec(queue->mutex);
queue->reference--;
// LoggerFactory::getLogger().debug("SHMQueue destructor reference===%d",
- // queue->reference.load());
if (queue->reference.load() == 0) {
delete queue;
hashtable_t *hashtable = mm_get_hashtable();
hashtable_remove(hashtable, KEY);
- // LoggerFactory::getLogger().debug("SHMQueue destructor delete queue\n");
+ // printf("SHMQueue destructor delete queue\n");
} else {
SemUtil::inc(queue->mutex);
}
--
Gitblit v1.8.0