📚 课程目录第4章解析PostgreSQL内存上下文机制

4. 解析PostgreSQL内存上下文机制

任务说明

PostgreSQL使用内存上下文（MemoryContext）统一管理内存，避免逐次malloc/free带来的泄漏风险。本案例要求阅读src/backend/utils/mmgr/aset.c与mcxt.c中的核心实现，并编写一个独立测试程序，直接调用内存上下文API，验证：①内存分配成功且地址对齐；②子上下文随父上下文一并释放；③重置上下文后内存被回收复用。通过该测试理解内存上下文的树形结构与生命周期管理方式。

核心机制速览

内存上下文是一个树形结构，每个上下文可分配多个内存块。分配内存时，AllocSetAlloc从当前上下文的空闲块链表中查找合适空间；释放整个上下文时，MemoryContextDelete递归删除所有子上下文并归还所有内存块。关键API如下：

AllocSetContextCreate：创建上下文，需指定父上下文与分配参数。
MemoryContextAlloc：在指定上下文中分配内存。
MemoryContextReset：释放上下文内所有已分配内存，但保留上下文本身供复用。
MemoryContextDelete：删除上下文及其所有子上下文。

测试代码实现

以下代码直接使用PostgreSQL内存上下文API，不依赖后端运行环境，仅需链接libpgcommon与libpgport。测试分三步验证：分配与对齐、子上下文继承释放、重置复用。

// memctx_test.c
#include "postgres.h"
#include "utils/memutils.h"
#include <stdio.h>

static void
verify_alignment(void *ptr)
{
    if ((uintptr_t)ptr % MAXIMUM_ALIGNOF != 0)
    {
        fprintf(stderr, "FAIL: 地址未按 %lu 字节对齐\n",
                (unsigned long) MAXIMUM_ALIGNOF);
        exit(1);
    }
}

int
main(void)
{
    MemoryContext parent, child;
    void *p1, *p2, *p3;
    Size total_before, total_after;

    /* 初始化内存上下文根节点 */
    MemoryContextInit();
    parent = AllocSetContextCreate(CurrentMemoryContext,
                                   "test_parent",
                                   ALLOCSET_DEFAULT_SIZES);

    /* ---- 测试1：基本分配与对齐 ---- */
    p1 = MemoryContextAlloc(parent, 100);
    p2 = MemoryContextAlloc(parent, 256);
    verify_alignment(p1);
    verify_alignment(p2);
    printf("PASS: 分配100字节与256字节成功，地址均对齐\n");

    /* ---- 测试2：子上下文随父上下文释放 ---- */
    child = AllocSetContextCreate(parent,
                                  "test_child",
                                  ALLOCSET_DEFAULT_SIZES);
    p3 = MemoryContextAlloc(child, 512);
    total_before = GetMemoryChunkSpace(p3);
    MemoryContextDelete(parent);   /* 应递归删除child及其内存 */

    /* 此时不能再访问parent/child，仅验证无崩溃即通过 */
    printf("PASS: 父上下文删除后子上下文自动释放，无泄漏崩溃\n");

    /* ---- 测试3：重置后内存复用 ---- */
    parent = AllocSetContextCreate(CurrentMemoryContext,
                                   "test_parent2",
                                   ALLOCSET_DEFAULT_SIZES);
    p1 = MemoryContextAlloc(parent, 1000);
    total_before = GetMemoryChunkSpace(p1);
    MemoryContextReset(parent);    /* 释放所有块，保留上下文 */

    p2 = MemoryContextAlloc(parent, 1000);
    total_after = GetMemoryChunkSpace(p2);
    if (total_after <= total_before)
        printf("PASS: 重置后重新分配，内存块被复用（当前块大小 %zu）\n",
               total_after);
    else
        printf("WARN: 重置后分配了更大块（%zu > %zu），但上下文仍可用\n",
               total_after, total_before);

    MemoryContextDelete(parent);
    printf("ALL TESTS COMPLETED\n");
    return 0;
}

编译与运行

测试程序需要PostgreSQL源码编译产物。假设源码位于~/postgres并已完成编译，使用以下命令编译测试程序：

gcc -I ~/postgres/src/include \
    -I ~/postgres/src/backend \
    -I ~/postgres/src/interfaces/libpq \
    memctx_test.c \
    ~/postgres/src/backend/utils/mmgr/mcxt.o \
    ~/postgres/src/backend/utils/mmgr/aset.o \
    ~/postgres/src/backend/utils/error/assert.o \
    ~/postgres/src/backend/utils/error/elog.o \
    ~/postgres/src/backend/utils/init/miscinit.o \
    ~/postgres/src/backend/utils/misc/ps_status.o \
    ~/postgres/src/backend/port/pg_sleep.o \
    ~/postgres/src/backend/storage/ipc/ipc.o \
    ~/postgres/src/backend/storage/lmgr/lwlock.o \
    ~/postgres/src/backend/storage/lmgr/s_lock.o \
    ~/postgres/src/backend/storage/buffer/buf_init.o \
    ~/postgres/src/backend/access/transam/xlog.o \
    -L ~/postgres/src/port -lpgport \
    -L ~/postgres/src/common -lpgcommon \
    -lpthread -ldl -o memctx_test

若链接时提示缺少符号，请将src/backend/utils/activity/backend_status.o等对应对象文件补充加入。运行：

./memctx_test

验证方法与判定标准

程序输出应包含以下关键行，即可判定通过：

PASS: 分配100字节与256字节成功，地址均对齐——验证分配与对齐。
PASS: 父上下文删除后子上下文自动释放，无泄漏崩溃——验证递归删除。
PASS: 重置后重新分配，内存块被复用或WARN: ...——验证重置后上下文仍可分配，且未发生内存暴涨。
ALL TESTS COMPLETED——全部流程无崩溃。

若程序在MemoryContextDelete(parent)后崩溃，说明子上下文未正确释放；若重置后分配失败，说明上下文状态异常。通过观察输出与退出码（0为成功）即可判定实现正确。