From 68d23225a38a35f1325eb39fa4ed5a005d5de473 Mon Sep 17 00:00:00 2001 From: fujuntang <fujuntang@aiot.com> Date: 星期三, 11 八月 2021 09:50:20 +0800 Subject: [PATCH] fix from 3.1 first commit --- src/shm/shm_mm_wrapper.cpp | 62 +++++++++++++++++++++--------- 1 files changed, 43 insertions(+), 19 deletions(-) diff --git a/src/shm/shm_mm_wrapper.cpp b/src/shm/shm_mm_wrapper.cpp index 6560f83..672fadb 100644 --- a/src/shm/shm_mm_wrapper.cpp +++ b/src/shm/shm_mm_wrapper.cpp @@ -1,20 +1,53 @@ #include "shm_mm_wrapper.h" -#include "mem_pool.h" +#include "shm_mm.h" #include "hashtable.h" #include "lock_free_queue.h" #include "shm_socket.h" -#define BUFFER_TIME 10 +#define BUFFER_TIME 1 + + void shm_mm_wrapper_init(int size) { - mem_pool_init(size); + shm_mm_init(size); + } void shm_mm_wrapper_destroy() { - mem_pool_destroy(); + shm_mm_destroy(); } int shm_mm_wrapper_alloc_key() { return mm_alloc_key(); +} + + +/** + * 鍥炴敹鍋囧垹闄ょ殑key + */ +int shm_mm_wrapper_start_resycle() { + ShmQueueStMap * shmQueueStMap = shm_mm_attach<ShmQueueStMap>(SHM_QUEUE_ST_KEY); + hashtable_t *hashtable = mm_get_hashtable(); + LockFreeQueue<shm_packet_t> *mqueue; + while(true) { + for(auto it = shmQueueStMap->begin(); it != shmQueueStMap->end(); ++it ) { + if(it->second.status == SHM_QUEUE_ST_CLOSED && difftime(time(NULL), it->second.closeTime) > BUFFER_TIME ) { + // mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, keys[i]); + // if(mqueue != NULL) { + // delete mqueue; + // } + + hashtable_remove(hashtable, it->first); + printf("reomved queue %d\n\n", it->first); + it->second.status = SHM_QUEUE_ST_RECYCLED; + // 涓嶈兘 erase ,鍚﹀垯浼氬嚭鐜板杩涚▼涔嬮棿鐨勫悓姝ラ棶棰橈紝 鑰岃繖姝f槸杩欓噷瑕佽В鍐崇殑闂 + // it = shmQueueStMap->erase(it); + // continue; + } + } + + sleep(1); + } + return 0; } //鍒犻櫎鍖呭惈鍦╧eys鍐呯殑queue @@ -24,15 +57,9 @@ int count = 0; for(int i = 0; i< length; i++) { // 閿�姣佸叡浜唴瀛樼殑queue - mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, keys[i]); - if(mqueue == NULL) { - continue; - } - if(difftime(time(NULL), mqueue->getCreateTime()) > BUFFER_TIME ) { - delete mqueue; - hashtable_remove(hashtable, keys[i]); - count++; - } + hashtable_remove(hashtable, keys[i]); + LoggerFactory::getLogger()->debug("remove queue %d", keys[i]); + count++; } return count; @@ -58,12 +85,9 @@ // 100鍐呯殑鏄痓us鍐呴儴鑷繁鐢ㄧ殑 if (!found && *keyItr > 100) { // 閿�姣佸叡浜唴瀛樼殑queue - mqueue = (LockFreeQueue<shm_packet_t> *)hashtable_get(hashtable, *keyItr); - if(difftime(time(NULL), mqueue->getCreateTime()) > BUFFER_TIME ) { - delete mqueue; - hashtable_remove(hashtable, *keyItr); - count++; - } + hashtable_remove(hashtable, *keyItr); + LoggerFactory::getLogger()->debug("remove queue %d", *keyItr); + count++; } } -- Gitblit v1.8.0