這是redis源碼分析的第一篇,選取早期版本,一睹redis的基礎原理。這篇分析內存分配。
redis內存管理是對malloc系列函數做了封裝。額外記錄了申請的內存大小信息。
#include <stdlib.h>
#include <string.h>
static size_t used_memory = 0;
/*
分配sizeof(size_t)+size大小的內存,前面sizeof(size_t)個字節記錄本次分配的大小,
記錄分配的總內存大小,返回用於存儲數據的內存首地址,即跨過sizeof(size_t)大小個字節
*/
void *zmalloc(size_t size) {
void *ptr = malloc(size+sizeof(size_t));
if (!ptr) return NULL;
*((size_t*)ptr) = size;
used_memory += size+sizeof(size_t);
return (char*)ptr+sizeof(size_t);
}
// 重新分配內存,ptr是舊數據的內存首地址,size是本次需要分片的內存大小
void *zrealloc(void *ptr, size_t size) {
void *realptr;
size_t oldsize;
void *newptr;
// ptr爲空即沒有舊數據,新申請一塊內存即可,不涉及數據遷移
if (ptr == NULL) return zmalloc(size);
// 舊數據佔據的內存大小
realptr = (char*)ptr-sizeof(size_t);
// 得到數據部分的內存大小
oldsize = *((size_t*)realptr);
// 以舊數據的內存地址爲基地址,重新分配size+sizeof(size_t)大小的內存
newptr = realloc(realptr,size+sizeof(size_t));
if (!newptr) return NULL;
// 記錄數據部分的內存大小
*((size_t*)newptr) = size;
// 重新計算已分配內存的總大小,sizeof(size_t)這塊內存仍然在使用,不需要計算
used_memory -= oldsize;
used_memory += size;
// 返回存儲數據的內存首地址
return (char*)newptr+sizeof(size_t);
}
void zfree(void *ptr) {
void *realptr;
size_t oldsize;
if (ptr == NULL) return;
// 算出真正的內存首地址
realptr = (char*)ptr-sizeof(size_t);
oldsize = *((size_t*)realptr);
// 減去釋放的內存大小
used_memory -= oldsize+sizeof(size_t);
free(realptr);
}
// 複製字符串
char *zstrdup(const char *s) {
size_t l = strlen(s)+1;
char *p = zmalloc(l);
memcpy(p,s,l);
return p;
}
size_t zmalloc_used_memory(void) {
return used_memory;
}