From 51be606cae24b3f6de088ec59f69c378ceec0908 Mon Sep 17 00:00:00 2001 From: wangzhengquan <wangzhengquan85@126.com> Date: 星期二, 28 七月 2020 16:03:14 +0800 Subject: [PATCH] udpate --- /dev/null | 0 test/lambda.c | 19 +++++++++ test/Makefile | 2 src/queue/include/hashtable.h | 11 ++++- Makefile | 3 + src/queue/hashtable.c | 61 ++++++++++++++++++++++++------ test/test.c | 7 ++- demo/dgram_mod_bus | 0 8 files changed, 85 insertions(+), 18 deletions(-) diff --git a/Makefile b/Makefile index 18a1a54..b4fa973 100755 --- a/Makefile +++ b/Makefile @@ -1,5 +1,6 @@ -DIRS = src test_socket demo +DIRS = src test_socket demo test TAR_NAME = shm_queue.tar.gz + all: for i in $(DIRS); do \ (cd $$i && echo "making $$i" && $(MAKE) ) || exit 1; \ diff --git a/demo/dgram_mod_bus b/demo/dgram_mod_bus new file mode 100755 index 0000000..c2e1159 --- /dev/null +++ b/demo/dgram_mod_bus Binary files differ diff --git a/demo/dgram_mod_req_rep b/demo/dgram_mod_req_rep deleted file mode 100755 index ed8fe41..0000000 --- a/demo/dgram_mod_req_rep +++ /dev/null Binary files differ diff --git a/demo/dgram_mod_survey b/demo/dgram_mod_survey deleted file mode 100755 index 00bf41d..0000000 --- a/demo/dgram_mod_survey +++ /dev/null Binary files differ diff --git a/src/libshm_queue.a b/src/libshm_queue.a deleted file mode 100644 index 1e2bd99..0000000 --- a/src/libshm_queue.a +++ /dev/null Binary files differ 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; +} + + diff --git a/src/queue/include/hashtable.h b/src/queue/include/hashtable.h index 726a5bc..c1fba98 100755 --- a/src/queue/include/hashtable.h +++ b/src/queue/include/hashtable.h @@ -23,10 +23,17 @@ void *hashtable_remove(hashtable_t *hashtable, int key); void hashtable_removeall(hashtable_t *hashtable); - +/** + * 閬嶅巻hash_table + * @demo + * hashtable_foreach(&hashtable, [&](int key, void * value){ + * printf("%d, %p\n", key, value); + * }); + * +*/ void hashtable_foreach(hashtable_t *hashtable, hashtable_foreach_cb cb); -void hashtable_printall(hashtable_t *hashtable); +// void hashtable_printall(hashtable_t *hashtable); int hashtable_alloc_key(hashtable_t *hashtable); diff --git a/test/Makefile b/test/Makefile index 4bcb663..ffc7b16 100755 --- a/test/Makefile +++ b/test/Makefile @@ -14,7 +14,7 @@ include $(ROOT)/Make.defines.$(PLATFORM) -PROGS = protocle_parse strtok test_set test_vector +PROGS = protocle_parse strtok test_set test_vector lambda test build: $(PROGS) diff --git a/test/lambda.c b/test/lambda.c new file mode 100644 index 0000000..b8c2fd4 --- /dev/null +++ b/test/lambda.c @@ -0,0 +1,19 @@ +#include "usg_common.h" +#include "usg_typedef.h" +#include "hashtable.h" +#include "shm_mm.h" + +int main() { + shm_init(512); + hashtable_t hashtable; + + hashtable_init(&hashtable); + for(int i = 0; i < 10; i++) { + hashtable_put(&hashtable, i, (void *)(i)); + } + + hashtable_foreach(&hashtable, [&](int key, void * value){ + printf("%d, %p\n", key, value); + }); + shm_destroy(); +} \ No newline at end of file diff --git a/test/protocle_parse b/test/protocle_parse deleted file mode 100755 index e18682e..0000000 --- a/test/protocle_parse +++ /dev/null Binary files differ diff --git a/test/strtok b/test/strtok deleted file mode 100755 index df7f33e..0000000 --- a/test/strtok +++ /dev/null Binary files differ diff --git a/test/test b/test/test deleted file mode 100755 index 1c5c801..0000000 --- a/test/test +++ /dev/null Binary files differ diff --git a/test/test.c b/test/test.c index 744ac72..0aaa68e 100644 --- a/test/test.c +++ b/test/test.c @@ -1,12 +1,15 @@ #include "usg_common.h" #include "usg_typedef.h" -void test(char *src, int size) { +int test(char *src, int size) { + int i = strlen(src); char dest[size]; strncpy(dest, src, size); puts(dest); + return i; } int main() { char *str = "hello"; - test(str, strlen(str)); + int r = test(str, strlen(str)); + printf("%d\n", r); } \ No newline at end of file diff --git a/test/test_set b/test/test_set deleted file mode 100755 index c8b6db0..0000000 --- a/test/test_set +++ /dev/null Binary files differ diff --git a/test/test_vector b/test/test_vector deleted file mode 100755 index c308410..0000000 --- a/test/test_vector +++ /dev/null Binary files differ diff --git a/test_socket/dgram_mod_bus b/test_socket/dgram_mod_bus deleted file mode 100755 index b5eccd6..0000000 --- a/test_socket/dgram_mod_bus +++ /dev/null Binary files differ diff --git a/test_socket/dgram_mod_req_rep b/test_socket/dgram_mod_req_rep deleted file mode 100755 index 29f0599..0000000 --- a/test_socket/dgram_mod_req_rep +++ /dev/null Binary files differ diff --git a/test_socket/dgram_mod_survey b/test_socket/dgram_mod_survey deleted file mode 100755 index 48db086..0000000 --- a/test_socket/dgram_mod_survey +++ /dev/null Binary files differ -- Gitblit v1.8.0