Postgresql内存池源码分析

Postgresql内存上下文源码分析

1 数据库内存上下文

postgresql在7.1版本引入了内存上下文机制来解决日益严重的内存泄漏的问题，在引入了这种“内存池”机制后，数据库中的内存分配改为在“内存上下文中”进行，对用户来说，对内存的申请由原来的malloc、free变成了palloc、pfree。对内存上下文的常用操作包括：

• 创建一个内存上下文：MemoryContextCreate
• 在上下文中分配内存片：palloc
• 删除内存上下文：MemoryContextDelete
• 重置内存上下文：MemoryContextReset

这里引入两个概念：内存片和内存块的概念。

内存片(CHUNK)：用户在内存上下文中申请(palloc)到的内存单位。 内存块(BLOCK)：内存上下文在内存中申请(malloc)到的内存单位。

2 数据结构

2.1 AllocSetContext

typedef struct AllocSetContext
{
	MemoryContextData header;	/* Standard memory-context fields */
	/* Info about storage allocated in this context: */
	AllocBlock	blocks;			/* head of list of blocks in this set */
	AllocChunk	freelist[ALLOCSET_NUM_FREELISTS];		/* free chunk lists */
	/* Allocation parameters for this context: */
	Size		initBlockSize;	/* initial block size */
	Size		maxBlockSize;	/* maximum block size */
	Size		nextBlockSize;	/* next block size to allocate */
	Size		allocChunkLimit;	/* effective chunk size limit */
	AllocBlock	keeper;			/* if not NULL, keep this block over resets */
} AllocSetContext;

AllocSetContext是内存上下文的核心的控制结构，我们在代码中经常看到的内存上下文TopMemoryContext的定义为：

MemoryContext TopMemoryContext = NULL;

可以看到这个内存上下文的类型是MemoryContext，即：

typedef struct MemoryContextData
{
	NodeTag		type;			/* identifies exact kind of context */
	MemoryContextMethods *methods;		/* virtual function table */
	MemoryContext parent;		/* NULL if no parent (toplevel context) */
	MemoryContext firstchild;	/* head of linked list of children */
	MemoryContext nextchild;	/* next child of same parent */
	char	   *name;			/* context name (just for debugging) */
	bool		isReset;		/* T = no space alloced since last reset */
} MemoryContextData;

那么MemoryContextData和AllocSetContext是什么样的关系呢？请看下图左半部分。

　　　　　　　　　　　　　　　　 图 2-1 内存上下文数据结构 　　AllocSetContext结构的第一个指针用于指向MemoryContextData，也就是说TopMemoryContext实际上是一个AllocSetContext结构，但是使用时通常将类型转换为MemoryContextData，实际上这也是PG中最常用的技巧之一，在代码中可以看到这样的写法：

AllocSet	set = (AllocSet) context;

由于AllocSetContext结构中的首部存放着MemoryContextData指针，所以这种转换可以成功。这样的使用方法有些类似与类的继承：MemoryContextData代表父类，AllocSetContext在父类（头部的指针）的基础上增加了一些新的功能。实际上PG就是使用了这种机制实现了interface（MemoryContextData作为interface），而后面的实现可以有很多种（AllocSetContext是内存上下文的一种实现）。 　　好说到这里言归正传，继续介绍MemoryContextData数据结构的功能：

• methods：保存着内存上下文操作的函数指针（例如palloc、pfree）
• parent、firstchild、nextchild：形成内存上下文的BTree结构
• name：内存上下文名称（为了调试而存在）
• isReset：记录上次重置后是否有内存申请动作发生

MemoryContextData使内存上下文形成了一个二叉树的结构，这样的数据结构增加了内存上下文的易用性，即在重置或删除内存上下文时，所有当前上下文的子节点也会被递归的删除或重置，避免错删或漏删上下文。methods中保存的全部为函数指针，在内存上下文创建时，这些指针会被赋予具体函数地址。 　　下面继续介绍AllocSetContext数据结构：

• header：前面介绍过了
• blocks：内存块链表，内存上下文向OS申请连续大块内存后，空间由blocks链表维护
• freelist：内存片回收数组，后面具体分析
• initBlockSize：上下文申请的第一个内存块的大小
• maxBlockSize： 上下文申请的最大的内存块的大小
• nextBlockSize： 上下文下一次申请的内存块的大小（MemoryContextCreate函数中介绍这三个参数）
• allocChunkLimit：申请内存片/块的阈值
• keeper：这个指针指向内存上下文重置时不释放的内存块

2.2 AllocChunkData

内存片存在于内存块以内，是内存块分割后形成的一段空间，内存片空间的头部为AllocChunkData结构体，后面跟着该内存片的空间，实际上palloc返回的就这指向这段空间首地址的指针。内存片有两种状态：AllocSetContext中freelist数组中存放的是内存片指针是被回收的内存片；另外一种内存片是用户正在使用的内存片。（注意两种状态的内存片都存在于内存块中，被回收只是改变内存片aset指针，形成链表保存在freelist中；在使用中的内存片aset指针指向所属的AllocSetContext）

typedef struct AllocChunkData
{
	void	   *aset;
	Size		size;
}	AllocChunkData;

在palloc时会发生两种情况：

• allocset会在自己维护的内存块链表（blocks）中寻找空间构造内存片，然后分配给用户。
• 申请新的内存块追加到blocks链表中，在其中分配新的内存片分配给用户。

内存片的数据结构相对简单，空指针aset是一个复用的指针，当内存片正在使用时，aset指向它属于的allocset结构，当内存片被释放后，内存片被freelist数组回收，aset作为实现链表的指针，用于形成内存片的链式结构。

2.3 AllocBlockData

typedef struct AllocBlockData
{
	AllocSet	aset;			/* aset that owns this block */
	AllocBlock	next;			/* next block in aset's blocks list */
	char	   *freeptr;		/* start of free space in this block */
	char	   *endptr;			/* end of space in this block */
}	AllocBlockData;

内存块是内存上下文向操作系统申请的连续的一块内存空间，申请后将AllocBlockData结构置于空间的首部，其中freeptr和endptr用与指向当前内存块中空闲空间的首地址和当前内存块的尾地址，见图2-1中的“连续内存段（内存块）”。aset指向控制结构AllocSetContext，next指针形成内存块的链式结构。

2.4 freelist[ALLOCSET_NUM_FREELISTS]

AllocSetContext结构中的一个重要的数组freelist，这是一个定长数组：

#define ALLOCSET_NUM_FREELISTS	11
.
.
AllocChunk	freelist[ALLOCSET_NUM_FREELISTS];		/* free chunk lists */
.
.

这是一个存放内存片指针的数组，数组中每一个元素都是一个内存片指针，就像前面提到的，空闲内存片会形成链表结构，而链表的头结点的指针就存放在这个数组中。从长度来看，这个数组可以保存11个内存片的链表，每一个链表都保存这特定大小的内存片：

　　　　　　　　　　　　　图2-2 freelist

图2-2描述的就是freelist数组的结构，数组下标0位置保存8字节的内存片，下标1位置保存16字节的内存片，以此类推，freelist中可以保存的最大的内存片为8k字节。 　　相同大小的内存片会串在同一个链表中，放在freelist中指定的位置，数组下标的计算按照公式：log(Size)-3。例如大小为512字节的内存片被释放了，套用公式log(512)-3=5，那么这个内存片就会维护到freelist[5]指向的链表中。（具体计算过程见AllocSetFreeIndex函数）。

3 算法

3.1 AllocSetContextCreate：创建内存上下文

MemoryContext
AllocSetContextCreate(MemoryContext parent,
					  const char *name,
					  Size minContextSize,
					  Size initBlockSize,
					  Size maxBlockSize)

内存上下文创建需要传入几个参数：

• parent：当前创建内存上下文的父节点
• name：当前创建内存上下文名称
• minContextSize：创建上下文时申请内存块大小
• initBlockSize：该上下文第一次申请内存块大小
• maxBlockSize：该上下文可以申请的最大内存块大小

让我们看几个数据库中最常见的上下文创建时的参数，结合具体值在说说创建时参数的作用：

• minContextSize：如果这个值设定了并超过了一定大小（一个内存块结构体加上一个内存片结构体的大小），那么在创建上下文时立即申请一个内存块，大小为minContextSize。上图中我们可以看到大部分上下文minContextSize都为0，那么ErrorContext的minContextSize为8k有什么作用呢？在系统出现OOM时，内存空间已经耗尽，但是ereport的错误处理流程仍然需要申请内存空间去打印错误信息，但系统已经没有内存可以申请了。这时ErrorContext中保留的8k空间可以保证最后的错误处理流程可以正确执行。

-initBlockSize、maxBlockSize：内存上下文中的内存块申请的大小是由这两个参数决定的，initBlockSize代表了第一次申请的内存块大小，后面每一次申请都是前一次申请大小的二倍，直到申请内存大小为maxBlockSize为止，当达到maxBlockSize时，以后每一次申请的内存大小都等于maxBlockSize。（事实上如果多次在一个上下文申请内存，那么很快就会到达maxBlockSize，举个例子：TupleSort中申请内存块的大小序列为：8k 16k 32k 64k 128k 256k 512k 1M 2M 4M 8M 8M 8M 8M . . .）

• allocChunkLimit：这里引出一个重要的参数，内存片申请阈值，这个值被开始被设为8k字节，但是后面会适当缩小到maxBlockSize的1/8。这个参数的调整是为了减少内存片空间的浪费（内存块中的最后一段内存不足以放下一个内存片，所以这段空间被舍弃掉了，理论上浪费掉的空间最大为allocChunkLimit）

3.2 AllocSetAlloc

申请内存的流程图：

需要重点关注的有几点：

• 回收当前内存块的剩余空间：将剩余空间切割成freelist能保存的最大值，例如1000字节的内存片回收时首先申请512字节的内存片，然后挂在freelist[6]上，剩余488字节申请256字节的内存片挂在freelist[5]上，剩余232字节继续上面处理流程，直到最后空间小于8字节为止。
• 在多次申请内存块后，内存块的大小总会等于maxBlockSize，这样如果出现内存泄漏导致OOM时，如果某一个内存上下文非常大，可以利用这个特点分析内存问题的根因。例如每100次申请8M的内存块时，打印一次Backtrace。

3.3 AllocSetFree

释放内存流程图：

#### 3.4 AllocSetRealloc

relloc流程图：

3.5 AllocSetStats

这个函数会被MemoryContextStats递归调用，遍历内存上下文树的内个节点，并获取当前节点的信息。

GDB调试时这一个非常好用的函数，可以直接在log中打印内存上下文树，指令：

gdb > p MemoryContextStats(TopMemoryContext)

PG提供了一系列内存上下文debug相关函数，具体可见mcxt.c