zmalloc 代码解释（redis代码阅读）

在看redis代码的时候，看到一个常用的函数zmalloc，这个是作者封装的一个malloc函数。

深入看看，对内存空间的申请和线程安全都会有所了解。

void *zmalloc(size_t size) {
    void *ptr = malloc(size+PREFIX_SIZE);                                                   
    
    if (!ptr) zmalloc_oom_handler(size);                                                    
#ifdef HAVE_MALLOC_SIZE
    update_zmalloc_stat_alloc(zmalloc_size(ptr));                                           
    return ptr;                                                                             
#else
    *((size_t*)ptr) = size;
    update_zmalloc_stat_alloc(size+PREFIX_SIZE);                                            
    return (char*)ptr+PREFIX_SIZE;                                                          
#endif                                                                                      
}                                                                                           

下面就分析一下

    if (!ptr) zmalloc_oom_handler(size);                                                    

 这一句里函数的实现是

static void zmalloc_default_oom(size_t size) {
    fprintf(stderr, "zmalloc: Out of memory trying to allocate %zu bytes\n",                
        size);
    fflush(stderr);                                                                         
    abort();                                                                                
}                                                                   

这个函数作用就是，如果申请失败了，打印一个错误消息并输出，同时终止程序。如果内存不够了，redis也就要挂了。

对于

    update_zmalloc_stat_alloc(zmalloc_size(ptr));                                           

对应的实现是

 #define update_zmalloc_stat_alloc(__n) do { \                                               
    size_t _n = (__n); \
    if (_n&(sizeof(long)-1)) _n += sizeof(long)-(_n&(sizeof(long)-1)); \                    
    if (zmalloc_thread_safe) { \
        update_zmalloc_stat_add(_n); \                                                      
    } else { \
        used_memory += _n; \                                                                
    } \
} while(0)          

                                                                        

2. 里面比较核心的有两个，

a。 默认redis的地址以sizeof(log)对齐

b。 如果有线程安全，需要用线程安全原语来操作使用内存的大小