Redis内存管理源码阅读

破山中贼易,破心中贼难

C语言中没有自带GC, 所有的内存的分配和释都由开发者自己控制.为了实现内存分配释放的合理性和高效性, Redis封装了C语言中的malloc, calloc和realloc和free函数来实现对内存的管理.这些实现包含在zmalloc.c和zmalloc.h文件中.

Redis封装后的内存管理, 主要具有以下特点:

1. 记录已使用内存大小
2. 内存管理

记录已使用内存大小

通过维护use_memory变量, 表示已使用内存大小, 每当新分配或释放内存时都会对其进行更新.更新操作提供线程安全和非线程安全两种方式.

线程安全

在文件zmalloc.c中我们可以看到这部分的源码

// 使用c11规范的内置函数
#if defined(__ATOMIC_RELAXED)
#define update_zmalloc_stat_add(__n) __atomic_add_fetch(&used_memory, (__n), __ATOMIC_RELAXED)
#define update_zmalloc_stat_sub(__n) __atomic_sub_fetch(&used_memory, (__n), __ATOMIC_RELAXED)
// 使用GCC提供的内置函数
#elif defined(HAVE_ATOMIC)
#define update_zmalloc_stat_add(__n) __sync_add_and_fetch(&used_memory, (__n))
#define update_zmalloc_stat_sub(__n) __sync_sub_and_fetch(&used_memory, (__n))
#else

// 使用互斥量
#define update_zmalloc_stat_add(__n) do { \
    pthread_mutex_lock(&used_memory_mutex); \
    used_memory += (__n); \
    pthread_mutex_unlock(&used_memory_mutex); \
} while(0)

#define update_zmalloc_stat_sub(__n) do { \
    pthread_mutex_lock(&used_memory_mutex); \
    used_memory -= (__n); \
    pthread_mutex_unlock(&used_memory_mutex); \
} while(0)

#endif
// 保存内存使用量
static size_t used_memory = 0;
// 指定对used_memory的操是否是线程安全, 默认为非线程安全
static int zmalloc_thread_safe = 0;
// 互斥量
pthread_mutex_t used_memory_mutex = PTHREAD_MUTEX_INITIALIZER;

Redis中对used_memory变量的操作默认是非线程安全的. 我们可以通过调用zmalloc_enable_thread_safeness函数, 指定为线程安全操作.

• 使用c11规范中的内置函数__atomic_add_fetch和__atomic_sub_fetch

• 使用GCC提供的内置函数__sync_add_and_fetch和__sync_add_and_fetch

• 使用互斥量pthread_mutex_t

目前新版的Redis中使用的是GCC提供的内置函数实现对某一个变量的原子操作.

内存对齐

由于存在内存对齐, 实际申请得到的内存(sizeof(long)的整数倍)可能要大于指定的字节数, 所以在统计内存的时候, 应该考虑实际申请的内存大小.所以如果申请的内存不是sizeof(long)的倍数, 需要向上取整为sizeof(long)的倍数.

在zmalloc.c文件中向上取整的算法