Apache AGE中RTE替换为PNSI的内存优化方案问询

以下是几个适配PostgreSQL版本统一需求、同时解决内存浪费问题的实用方案：

1. 定义轻量包装结构替代全量PNSI #

既然在PG12场景下仅需使用PNSI的p_rte字段，没必要每次分配完整的ParseNamespaceItem结构体。可以在AGE的PG12适配层中定义一个极简包装体：

typedef struct AGE_LightPNSI {
    RangeTblEntry *p_rte;
    // 仅保留必要字段，避免冗余内存占用
} AGE_LightPNSI;

然后在AGE的公共接口层做版本适配：当编译环境为PG12时，用AGE_LightPNSI替代原生ParseNamespaceItem；高版本则直接使用原生结构。同时封装统一的访问宏，确保上层逻辑无需关心版本差异：

#ifdef PG_VERSION_12
#define AGE_GET_RTE_FROM_NSI(nsi) ((nsi)->p_rte)
typedef AGE_LightPNSI AGE_NamespaceItem;
#else
#define AGE_GET_RTE_FROM_NSI(nsi) ((nsi)->p_rte)
typedef ParseNamespaceItem AGE_NamespaceItem;
#endif

这种方式彻底避免了冗余内存分配，同时保持接口一致性。

2. 用联合体实现版本无关的抽象结构 #

在AGE的公共代码层定义一个包含版本标识和联合体的抽象结构，直接存储RTE或PNSI指针，无需强制转换：

typedef struct AGE_NamespaceItem {
    int pg_major_version;
    union {
        RangeTblEntry *rte;
        ParseNamespaceItem *pnsi;
    } ptr;
} AGE_NamespaceItem;

在函数内部根据版本直接访问对应指针：

Expr *make_vertex_expr(AGE_NamespaceItem *ns_item, ...) {
    RangeTblEntry *target_rte;
    if (ns_item->pg_major_version >= 13) {
        target_rte = ns_item->ptr.pnsi->p_rte;
    } else {
        target_rte = ns_item->ptr.rte;
    }
    // 原有核心逻辑复用target_rte即可
}

该方案既满足版本统一的接口需求，又避免了低版本下的内存浪费，后续版本升级仅需调整版本判断阈值。

3. 复用内存上下文实现PNSI对象池 #

如果存在批量创建PNSI的场景，可以利用PostgreSQL的内存上下文（MemoryContext）实现对象复用：

// 在查询解析初始化时创建PNSI对象池
static List *age_pnsi_pool = NIL;

ParseNamespaceItem *age_get_reusable_pnsi(RangeTblEntry *rte) {
    if (age_pnsi_pool != NIL) {
        ParseNamespaceItem *pnsi = linitial(age_pnsi_pool);
        age_pnsi_pool = list_delete_first(age_pnsi_pool);
        pnsi->p_rte = rte;
        // 重置其他必要字段
        return pnsi;
    }
    // 无可用对象时再分配新的
    ParseNamespaceItem *pnsi = palloc0(sizeof(ParseNamespaceItem));
    pnsi->p_rte = rte;
    return pnsi;
}

// 查询结束后将PNSI放回池
void age_return_pnsi_to_pool(ParseNamespaceItem *pnsi) {
    pnsi->p_rte = NULL;
    age_pnsi_pool = lcons(pnsi, age_pnsi_pool);
}

通过对象池减少频繁的内存分配/释放开销，尤其适合批量处理场景，同时避免重复分配冗余内存。

4. 栈上临时PNSI适配（PG12专属） #

在PG12环境下，直接使用栈上的临时ParseNamespaceItem结构体适配接口，无需堆内存分配：

#ifdef PG_VERSION_12
#define AGE_ADAPT_RTE_TO_PNSI(rte) \
    ((ParseNamespaceItem){.p_rte = rte, .p_namespace = NULL, /* 填充必要的空字段 */})
#endif

调用函数时直接传递栈上临时对象：

// PG12场景下
RangeTblEntry *rte = ...;
make_vertex_expr(&AGE_ADAPT_RTE_TO_PNSI(rte), ...);

这种方式完全避免堆内存浪费，但需确保函数不会持有该临时对象的指针到栈帧销毁之后。

内容的提问来源于stack exchange，提问作者rrrokhtar