内存管理

源码版本：redis 2.4.4

redis内存相关函数都放在zmalloc.h zmalloc.c中

redis中可以使用tcmalloc、jemalloc
Makefile：

[cpp] view plaincopy

1. ifeq ($(USE_TCMALLOC),yes)  
2.   ALLOC_DEP=  
3.   ALLOC_LINK=-ltcmalloc  
4.   ALLOC_FLAGS=-DUSE_TCMALLOC  
5. endif  
6.   
7. ifeq ($(USE_JEMALLOC),yes)  
8.   ALLOC_DEP=../deps/jemalloc/lib/libjemalloc.a  
9.   ALLOC_LINK=$(ALLOC_DEP) -ldl  
10.   ALLOC_FLAGS=-DUSE_JEMALLOC -I../deps/jemalloc/include  
11. endif  

zmalloc.c中

[cpp] view plaincopy

1. #if defined(USE_TCMALLOC)  
2. #define malloc(size) tc_malloc(size)  
3. #define calloc(count,size) tc_calloc(count,size)  
4. #define realloc(ptr,size) tc_realloc(ptr,size)  
5. #define free(ptr) tc_free(ptr)  
6.   
7. #elif defined(USE_JEMALLOC)  
8. #define malloc(size) je_malloc(size)  
9. #define calloc(count,size) je_calloc(count,size)  
10. #define realloc(ptr,size) je_realloc(ptr,size)  
11. #define free(ptr) je_free(ptr)  
12. #endif  

简单封装
void *zmalloc(size_t size);
void *zcalloc(size_t size);
void *zrealloc(void *ptr, size_t size);
void zfree(void *ptr);
分别对malloc、calloc、realloc、free做了简单封装
对zmalloc和zfree做分析

[cpp] view plaincopy

1. void *zmalloc(size_t size) {  
2.     void *ptr = malloc(size+PREFIX_SIZE);  
3.   
4.     if (!ptr) zmalloc_oom(size);  
5. #ifdef HAVE_MALLOC_SIZE  
6.     update_zmalloc_stat_alloc(zmalloc_size(ptr),size);  
7.     return ptr;  
8. #else  
9.     *((size_t*)ptr) = size;  
10.     update_zmalloc_stat_alloc(size+PREFIX_SIZE,size);  
11.     return (char*)ptr+PREFIX_SIZE;  
12. #endif  
13. }  
14. void zfree(void *ptr) {  
15. #ifndef HAVE_MALLOC_SIZE  
16.     void *realptr;  
17.     size_t oldsize;  
18. #endif  
19.   
20.     if (ptr == NULL) return;  
21. #ifdef HAVE_MALLOC_SIZE  
22.     update_zmalloc_stat_free(zmalloc_size(ptr));  
23.     free(ptr);  
24. #else  
25.     realptr = (char*)ptr-PREFIX_SIZE;  
26.     oldsize = *((size_t*)realptr);  
27.     update_zmalloc_stat_free(oldsize+PREFIX_SIZE);  
28.     free(realptr);  
29. #endif  
30. }  

除了分配给指定大小的内存之外，还分配了PREFIX_SIZE
内存格局如下：

申请内存返回Real_ptr

[cpp] view plaincopy

1. #define update_zmalloc_stat_alloc(__n,__size) do { \  
2.     size_t _n = (__n); \  
3.     if (_n&(sizeof(long)-1)) _n += sizeof(long)-(_n&(sizeof(long)-1)); \  
4.     if (zmalloc_thread_safe) { \  
5.         pthread_mutex_lock(&used_memory_mutex);  \  
6.         used_memory += _n; \  
7.         pthread_mutex_unlock(&used_memory_mutex); \  
8.     } else { \  
9.         used_memory += _n; \  
10.     } \  
11. } while(0)  
12.   
13. #define update_zmalloc_stat_free(__n) do { \  
14.     size_t _n = (__n); \  
15.     if (_n&(sizeof(long)-1)) _n += sizeof(long)-(_n&(sizeof(long)-1)); \  
16.     if (zmalloc_thread_safe) { \  
17.         pthread_mutex_lock(&used_memory_mutex);  \  
18.         used_memory -= _n; \  
19.         pthread_mutex_unlock(&used_memory_mutex); \  
20.     } else { \  
21.         used_memory -= _n; \  
22.     } \  
23. } while(0)  

used_memory记录当前分配的总内存
在定义了 int vm_max_threads; /* Max number of I/O threads running at the same time */
    if (server.vm_max_threads != 0)
        zmalloc_enable_thread_safeness(); /* we need thread safe zmalloc() */
在使用Threaded Virtual Memory I/O的时候，需要安全的zmalloc

[cpp] view plaincopy

1. size_t zmalloc_used_memory(void) {  
2.     size_t um;  
3.   
4.     if (zmalloc_thread_safe) pthread_mutex_lock(&used_memory_mutex);  
5.     um = used_memory;  
6.     if (zmalloc_thread_safe) pthread_mutex_unlock(&used_memory_mutex);  
7.     return um;  
8. }  

zmalloc_used_memory返回进程当前使用的内存，用以做相应的清内存操作，eg：

[cpp] view plaincopy

1. zmalloc_used_memory() > server.maxmemory //和配置的最大内存比较  
2. ....  

zmalloc_get_rss()用来计算进程实际使用物理内存