From 08505499d1449f53449e119aefc71324376b348b Mon Sep 17 00:00:00 2001
From: shenxin <shenxin@basic.com>
Date: 星期一, 23 八月 2021 19:40:14 +0800
Subject: [PATCH] 增加BHSubscribeNetTopics、BHQueryTopicAddress接口 修改CMakeLists.txt,增加相应文件到编译中。(需要安装protobuf)

---
 src/shm/shm_mm.h |   21 ++++++++++++++++++---
 1 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/shm/shm_mm.h b/src/shm/shm_mm.h
index db1bea9..5339d8a 100644
--- a/src/shm/shm_mm.h
+++ b/src/shm/shm_mm.h
@@ -1,9 +1,12 @@
 #ifndef __SHM_MM_H__
 #define __SHM_MM_H__
+#include "usg_common.h"
+#include "shm_allocator.h"
+#include "key_def.h"
 
 #define SHM_QUEUE_ST_OPENED 0
-
 #define SHM_QUEUE_ST_CLOSED 1
+#define SHM_QUEUE_ST_RECYCLED 2
 
 struct shm_queue_status_t {
 
@@ -24,8 +27,20 @@
 void shm_mm_free (void *ptr);
 
 
+
 template <typename T>
-T* shm_mm_attach(int key) ;
+T* shm_mm_attach(int key) {
+  void *ptr;
+  // T* tptr;
+  hashtable_t *hashtable = mm_get_hashtable();
+  ptr = hashtable_get(hashtable, key);
+  if(ptr == NULL || ptr == (void *)1 ) {
+    ptr = mm_malloc(sizeof(T));
+    hashtable_put(hashtable, key, ptr);
+    new(ptr) T;
+  }
+  return (T*)ptr; 
+}
 
 void shm_mm_free_by_key(int key) ;
 
@@ -34,4 +49,4 @@
 
 int shm_mm_alloc_key();
 
-#endif 
\ No newline at end of file
+#endif 

--
Gitblit v1.8.0