内存池的实现

概述：本文介绍用户层内存池的实现

Q:为什么需要内存池？

A：在项目中，用户层通过malloc系统调用申请内存的次数可能很多，每一次malloc系统调用，都会引起用户态——内核态的切换，这样的开销对性能的影响是不容忽视的，尤其是当你为了申请一段很小的空间而调用malloc时，性能的损失达到了最大化

Q:什么是内存池？

解决方案：只调用一次malloc

A:只调用一次malloc申请足够大的内存空间，后续需要使用内存时，从已申请的内存中取出一块，这就是内存池的概念

这样就避免了频繁进行malloc系统调用，大大降低了性能损失

内存池的实现

思路：

1.内存池的空间由一次malloc调用分配

2.此时的内存池是一整块连续的内存空间

3.使用时，每次应该分配出一小块（固定大小或任意大小）

4.用户调用malloc申请内存，应该得到一块空间的起始地址

5.用户free归还一块内存时，这块内存需要能被重复使用

为了简要说明原理，本文介绍分配固定大小的内存池版本

后面使用到的二级指针，不懂的朋友可以先移步博主的数据结构专栏——二级指针这篇文章

1.内存池结构体定义：

typedef struct mempool_s {int block_size; // 内存池最小单元大小int free_count; // 空余block数量char *free_ptr; // 当前空余的块的地址 char *mem; // 整个内存池的首地址
} mempool_t;

2.初始化，为内存池申请空间：

int mp_init(mempool_t *m, int size) { // 初始化内存池，指定大小if (!m) return -1;if (size < 16) size = 16; // 最小内存池尺寸为 16字节m->block_size = size;m->mem = (char *)malloc(MEM_PAGE_SIZE); // 为内存池分配空间if (!m->mem) return -1;m->free_ptr = m->mem;   // 当前空闲的区域指针m->free_count = MEM_PAGE_SIZE / size;// 二级指针// 将一整块内存抽象为多块：块大小为size，每一块的首地址的前8个字节保存下一块的首地址int i = 0;char *ptr = m->free_ptr;for (i = 0;i < m->free_count; i++) {*(char **)ptr = ptr + size; // 将当前块的首地址的前8个字节保存下一块的首地址ptr += size;}*(char **)ptr = NULL; // 最后一块return 0;
}

3.用户层接口：malloc

void *mp_alloc(mempool_t *m) { // 向内存池申请一块内存if (!m || m->free_count == 0) return NULL; // 如果内存池不存在或没有剩余空间void *ptr = m->free_ptr;m->free_ptr = *(char **)ptr; // 更新free指针，指向下一个空余的内存块m->free_count --;return ptr;
}

4.用户层接口：free

void mp_free(mempool_t *m, void *ptr) {*(char **)ptr = m->free_ptr; // 将即将被回收的块，指向当前空余的块m->free_ptr = (char *)ptr; // 将当前空余的块修改为即将回收的块// 相当于把回收的块插入空余队列的第一个位置，优先分配回收的块m->free_count ++;
}

5.测试函数main

int  main() { // 测试mempool_t m;mp_init(&m, 32);void *p1 = mp_alloc(&m);printf("1: mp_alloc: %p\n", p1);void *p2 = mp_alloc(&m);printf("2: mp_alloc: %p\n", p2);void *p3 = mp_alloc(&m);printf("3: mp_alloc: %p\n", p3);void *p4 = mp_alloc(&m);printf("4: mp_alloc: %p\n", p4);mp_free(&m, p2);void *p5 = mp_alloc(&m);printf("5: mp_alloc: %p\n", p5);return 0;
}