Redis源代码分析


首先Zmalloc的接口定义在头文件Zmalloc.h里:

[cpp]
  1. void *zmalloc(size_t size);  
  2. void *zcalloc(size_t size);  
  3. void *zrealloc(void *ptr, size_t size);  
  4. void zfree(void *ptr);  
  5. char *zstrdup(const char *s);  
  6. size_t zmalloc_used_memory(void);  
  7. void zmalloc_enable_thread_safeness(void);  
  8. float zmalloc_get_fragmentation_ratio(void);  
  9. size_t zmalloc_get_rss(void);  
前五个函数对应于C标准库函数。zmalloc_used_memory用来返回当前已用内存。

举例分析zmalloc函数:

[cpp]
  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. }  
和标准库的malloc函数的区别在于,zmalloc还在内存块头部保存了内存块的大小。

PREFIX_SIZE定义为系统中一个标准的size_t的大小:

update_zmalloc_stat_alloc是一个宏,用来更新内存占用量的统计,定义为:

[cpp]
  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)  
如果启用了线程安全选项,将在加锁之后才对临界变量used_memory_mutex进行修改。相关变量定义在:




[cpp] 
  1. static size_t used_memory = 0;  
  2. static int zmalloc_thread_safe = 0;  
  3. pthread_mutex_t used_memory_mutex = PTHREAD_MUTEX_INITIALIZER;  
在函数size_t zmalloc_used_memory(void)中,返回已占用内存量的大小,即静态变量used_memory。



函数zmalloc_get_rss()获取RSS(Resident Set Size) 的方式有三种:如果定义了PROC_FS,那么将从"/proc/[getpid()]/stat"中读取;如果定义了TASK_INFO,将从该进程id对应的task_info_t结构中读取resident_size变量;最后,如果前两者都没有定义,那么简单地返回used_memory。 

zmalloc_get_fragmentation_ratio(void)计算碎片率的公式为:(float)zmalloc_get_rss()/zmalloc_used_memory()。

  • 1
  • 2
  • 下一页
【内容导航】
第1页:内存管理——Zmalloc 第2页:散列表——Dict(上)

相关内容