add thread safe & fix auto extend

Author: userpro

Makefile

@@ -1,17 +1,18 @@
 CC = gcc
 CPP = g++
+GCCFLAG = -Wall -lpthread
 SOURCES = memorypool.c
 MAIN_SOURCES = test.cpp
 MAIN_OUTPUT = test
 EXAMPLE_SOURCES = example.c
 EXAMPLE_OUTPUT = example
 
 run_test:
-	$(CPP) -Wall $(MAIN_SOURCES) $(SOURCES) -o $(MAIN_OUTPUT).out
+	$(CPP) $(GCCFLAG) $(MAIN_SOURCES) $(SOURCES) -o $(MAIN_OUTPUT).out
 	./$(MAIN_OUTPUT).out
 
 run_example:
-	$(CC)  -Wall $(EXAMPLE_SOURCES) $(SOURCES) -o $(EXAMPLE_OUTPUT).out
+	$(CC)  $(GCCFLAG) $(EXAMPLE_SOURCES) $(SOURCES) -o $(EXAMPLE_OUTPUT).out
 	./$(EXAMPLE_OUTPUT).out
 
 .PHONY: clean

README.md

@@ -8,20 +8,18 @@
 ## Example
 
 ~~~c
-#include <stdio.h>
 #include "memorypool.h"
+#include <stdio.h>
 
-struct TAT
-{
+struct TAT {
     int T_T;
 };
-
-int max_mem = 2*GB + 1000*MB + 1000*KB;
-int mem_pool_size = 1*GB + 500*MB + 500*KB;
+mem_size_t max_mem = 2*GB + 500*MB + 500*KB;
+mem_size_t mem_pool_size = 1*GB + 500*MB + 500*KB;
 
 int main()
 {
-    MemoryPool *mp = MemoryPool_Init(max_mem, mem_pool_size, 1);
+    MemoryPool *mp = MemoryPool_Init(max_mem, mem_pool_size, 0);
     struct TAT *tat = (struct TAT *)MemoryPool_Alloc(mp, sizeof(struct TAT));
     tat->T_T = 2333;
     printf("%d\n", tat->T_T);
@@ -42,26 +40,24 @@ int main()
 
 `mem_size_t` => `unsigned long long`
 
-`MemoryPool_Init` 参数(`mem_size_t max_mem_pool_size`, `mem_size_t mempoolsize`, `int auto_extend`)
+`MemoryPool_Init` 参数(`mem_size_t maxmempoolsize`, `mem_size_t mempoolsize`, `int thread_safe`)
 
-> `mempoolsize`: 内存池字节数
+> `maxmempoolsize`: 内存池字节数上限
+>
+> `mempoolsize`: 内存池字节数 (maxmempoolsize与mempoolsize不相等时会自动扩展, 直到上限)
+>
+> `thread_safe`: 是否线程安全 (1->线程安全)
 
-> `auto_extend`: 是否自动扩展 (1->开启自动扩展 0->不开启)
+`MemoryPool_Alloc` 行为与系统malloc一致(参数多了一个)
 
-`MemoryPool_Alloc` 行为与系统malloc一致(唔 参数多了一个)
-
-`MemoryPool_Free` 行为与系统free一致(唔 多了个返回值)
+`MemoryPool_Free` 行为与系统free一致(返回值0为正常)
 
 ~~~c
-MemoryPool* MemoryPool_Init(mem_size_t max_mem_pool_size, mem_size_t mempoolsize, int auto_extend);
-
-void* MemoryPool_Alloc(MemoryPool *mp, mem_size_t wantsize);
-
-int   MemoryPool_Free(MemoryPool *mp, void *p);
-
-MemoryPool* MemoryPool_Clear(MemoryPool *mp);
-
-int   MemoryPool_Destroy(MemoryPool *mp);
+MemoryPool* MemoryPool_Init   (mem_size_t maxmempoolsize, mem_size_t mempoolsize, int thread_safe);
+void*       MemoryPool_Alloc  (MemoryPool *mp, mem_size_t wantsize);
+int         MemoryPool_Free   (MemoryPool *mp, void *p);
+MemoryPool* MemoryPool_Clear  (MemoryPool *mp);
+int         MemoryPool_Destroy(MemoryPool *mp);
 ~~~
 
 - 获取内存池信息
@@ -71,27 +67,26 @@ int   MemoryPool_Destroy(MemoryPool *mp);
 `get_mempool_prog_usage` 获取内存池中真实分配内存比例(除去了内存池管理结构占用的内存)
 
 ~~~c
-double get_mempool_usage(MemoryPool *mp);
-
-double get_mempool_prog_usage(MemoryPool *mp);
+// 实际分配空间
+mem_size_t get_used_memory  (MemoryPool *mp);
+float      get_mempool_usage(MemoryPool *mp);
+// 数据占用空间
+mem_size_t get_prog_memory  (MemoryPool *mp);
+float      get_mempool_prog_usage(MemoryPool *mp);
 ~~~
 
 ## Update
 
 - 18-1-7 12.53 增加了自动扩展 (内存池耗尽时自动新扩展一个mempoolsize大小的内存)
 - 18-5-27 1.10 改进输出信息 增强测试程序(详见main.cpp)
 - 19-3-18 11.05 改进格式, 修复潜在bug
+- 19-4-1 20:46 增加线程安全选项, 修改了自动扩展逻辑.
 
 ## Tips
 
 - 可通过注释`test.c`里的`#include "memorypool.h"`来切换对比系统`malloc` `free`和内存池
-
-- 暂不支持多线程
-
-- 多食用`MemoryPool_Clear`
-
+- 线程安全(如无必要, 关闭线程安全以改进性能)
+- 多食用`MemoryPool_Clear` (多线程情况下慎用)
 - 在 **2GB** 数据量 **顺序分配释放** 的情况下比系统`malloc` `free`平均快 **30%-50%** (食用`MemoryPool_Clear`效果更明显)
-
-- `mem_size_t`使用`unsigned long long`以支持4GB以上内存管理(头文件中`MAX_MEM_SIZE`宏定义了最大可管理内存)
-
-- 大量小块内存分配会有 **20%-30%** 内存空间损失(用于存储管理结构体 emmmmm)
+- `mem_size_t`使用`unsigned long long`以支持4GB以上内存管理
+- 大量小块内存分配会有 **20%-30%** 内存空间损失(用于存储管理结构体)

example.c

@@ -1,5 +1,5 @@
-#include <stdio.h>
 #include "memorypool.h"
+#include <stdio.h>
 
 struct TAT
 {
@@ -11,7 +11,7 @@ mem_size_t mem_pool_size = 1*GB + 500*MB + 500*KB;
 
 int main()
 {
-    MemoryPool *mp = MemoryPool_Init(max_mem, mem_pool_size, 1);
+    MemoryPool *mp = MemoryPool_Init(max_mem, mem_pool_size, 0);
     struct TAT *tat = (struct TAT *)MemoryPool_Alloc(mp, sizeof(struct TAT));
     tat->T_T = 2333;
     printf("%d\n", tat->T_T);