redis內(nèi)存分配是怎樣的

redis內(nèi)存管理是對(duì)malloc系列函數(shù)做了封裝。額外記錄了申請(qǐng)的內(nèi)存大小信息。

#include <stdlib.h>
#include <string.h>

static size_t used_memory = 0;
/*
    分配sizeof(size_t)+size大小的內(nèi)存，前面sizeof(size_t)個(gè)字節(jié)記錄本次分配的大小，
    記錄分配的總內(nèi)存大小，返回用于存儲(chǔ)數(shù)據(jù)的內(nèi)存首地址，即跨過sizeof(size_t)大小個(gè)字節(jié)
*/
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);
}
// 重新分配內(nèi)存，ptr是舊數(shù)據(jù)的內(nèi)存首地址，size是本次需要分片的內(nèi)存大小
void *zrealloc(void *ptr, size_t size) {
    void *realptr;
    size_t oldsize;
    void *newptr;
    // ptr為空即沒有舊數(shù)據(jù)，新申請(qǐng)一塊內(nèi)存即可，不涉及數(shù)據(jù)遷移
    if (ptr == NULL) return zmalloc(size);
    // 舊數(shù)據(jù)占據(jù)的內(nèi)存大小
    realptr = (char*)ptr-sizeof(size_t);
    // 得到數(shù)據(jù)部分的內(nèi)存大小
    oldsize = *((size_t*)realptr);
    // 以舊數(shù)據(jù)的內(nèi)存地址為基地址，重新分配size+sizeof(size_t)大小的內(nèi)存
    newptr = realloc(realptr,size+sizeof(size_t));
    if (!newptr) return NULL;
    // 記錄數(shù)據(jù)部分的內(nèi)存大小
    *((size_t*)newptr) = size;
    // 重新計(jì)算已分配內(nèi)存的總大小，sizeof(size_t)這塊內(nèi)存仍然在使用，不需要計(jì)算
    used_memory -= oldsize;
    used_memory += size;
    // 返回存儲(chǔ)數(shù)據(jù)的內(nèi)存首地址
    return (char*)newptr+sizeof(size_t);
}

void zfree(void *ptr) {
    void *realptr;
    size_t oldsize;

    if (ptr == NULL) return;
    // 算出真正的內(nèi)存首地址
    realptr = (char*)ptr-sizeof(size_t);
    oldsize = *((size_t*)realptr);
    // 減去釋放的內(nèi)存大小
    used_memory -= oldsize+sizeof(size_t);
    free(realptr);
}
// 復(fù)制字符串
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;
}