From 51be606cae24b3f6de088ec59f69c378ceec0908 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期二, 28 七月 2020 16:03:14 +0800 Subject: [PATCH] udpate --- src/queue/hashtable.c | 61 ++++++++++++++++++++++++------ 1 files changed, 49 insertions(+), 12 deletions(-) diff --git a/src/queue/hashtable.c b/src/queue/hashtable.c index 1fa6266..eaa3ca6 100755 --- a/src/queue/hashtable.c +++ b/src/queue/hashtable.c @@ -3,6 +3,7 @@ #include "mm.h" #include "sem_util.h" #include <set> +#include <functional> typedef struct tailq_entry_t { @@ -32,7 +33,7 @@ } -void *_hashtable_get(hashtable_t *hashtable, int key) +static inline void *_hashtable_get(hashtable_t *hashtable, int key) { size_t code = hashcode(key); tailq_entry_t *item; @@ -54,7 +55,7 @@ } -void * _hashtable_put(hashtable_t *hashtable, int key, void *value) +static inline void * _hashtable_put(hashtable_t *hashtable, int key, void *value) { size_t code = hashcode(key); void *oldvalue; @@ -187,8 +188,11 @@ SemUtil::dec(hashtable->wlock); } SemUtil::inc(hashtable->mutex); + // ================ void * res = _hashtable_get(hashtable, key); + + // ================== SemUtil::dec(hashtable->mutex); hashtable->readcnt--; @@ -204,6 +208,7 @@ void hashtable_put(hashtable_t *hashtable, int key, void *value) { SemUtil::dec(hashtable->mutex); + // 璁剧疆璇讳紭鍏堢骇楂� while (hashtable->readcnt > 0) { SemUtil::set(hashtable->cond, 0); @@ -218,6 +223,7 @@ //鑾峰彇璇诲啓閿� SemUtil::dec(hashtable->wlock); + _hashtable_put(hashtable, key, value); //閲婃斁璇诲啓閿� SemUtil::inc(hashtable->wlock); @@ -225,7 +231,7 @@ -void hashtable_foreach(hashtable_t *hashtable, hashtable_foreach_cb cb) { +static inline void _hashtable_foreach(hashtable_t *hashtable, std::function<void(int, void *)> cb) { tailq_entry_t *item; for (int i = 0; i < MAPSIZE; i++) { tailq_header_t *my_tailq_head = hashtable->array[i] ; @@ -240,18 +246,35 @@ } } -int hashtable_alloc_key(hashtable_t *hashtable) { - int key = START_KEY; - SemUtil::dec(hashtable->wlock); - while(_hashtable_get(hashtable, key) != NULL) { - key++; - } +void hashtable_foreach(hashtable_t *hashtable, hashtable_foreach_cb cb) { + SemUtil::dec(hashtable->mutex); + hashtable->readcnt++; + if (hashtable->readcnt == 1) { + //鑾峰彇璇诲啓閿� + SemUtil::dec(hashtable->wlock); + } + SemUtil::inc(hashtable->mutex); - _hashtable_put(hashtable, key, (void *)1); - SemUtil::inc(hashtable->wlock); - return key; + // ================== + + _hashtable_foreach(hashtable, cb); + + // ================== + + SemUtil::dec(hashtable->mutex); + hashtable->readcnt--; + if(hashtable->readcnt == 0) { + //閲婃斁璇诲啓閿� + SemUtil::inc(hashtable->wlock); + //閫氱煡鍐� + SemUtil::set(hashtable->cond, 1); + } + SemUtil::inc(hashtable->mutex); + } + + std::set<int> * hashtable_keyset(hashtable_t *hashtable) { std::set<int> *keyset = new std::set<int>; @@ -271,3 +294,17 @@ } +int hashtable_alloc_key(hashtable_t *hashtable) { + int key = START_KEY; + SemUtil::dec(hashtable->wlock); + + while(_hashtable_get(hashtable, key) != NULL) { + key++; + } + + _hashtable_put(hashtable, key, (void *)1); + SemUtil::inc(hashtable->wlock); + return key; +} + + -- Gitblit v1.8.0