PostgreSQL C脚本开发中如何将text类型变量转换为cstring

首先修正你现有代码的不规范写法：PG_GETARG_DATUM返回的是通用Datum类型，直接赋值给text*虽然在多数架构下能运行，但没有做类型适配和压缩字段解压，存在兼容风险。如果你要保留原有Datum获取逻辑，需要先做显式转换：

#include "postgres.h"
#include "fmgr.h"
#include "utils/builtins.h" // 必须引入该头文件，text_to_cstring等接口声明在这里

// 原有写法的修正版本
text *str_01 = DatumGetTextPP(PG_GETARG_DATUM(0));
text *txt_02 = DatumGetTextPP(PG_GETARG_DATUM(1));

// 更推荐直接用官方封装的入参宏，写法更简洁，效果和上面完全一致
// text *str_01 = PG_GETARG_TEXT_PP(0);
// text *txt_02 = PG_GETARG_TEXT_PP(1);

转换方法 #

PostgreSQL的text类型结构是变长长度头 + 连续字符数据，默认不会在字符末尾补\0结束符，绝对不能直接强转成char*使用，否则会读到脏数据甚至触发内存越界。根据你的使用场景选对应方案即可：

方案1：只读访问字符串（最常用） #

如果只需要读取字符串内容做计算，不需要修改原数据，直接调用内置的text_to_cstring函数即可，接口会自动分配内存、补字符串结束符：

char *cstr_01 = text_to_cstring(str_01);
char *cstr_02 = text_to_cstring(txt_02);

// 后续直接对cstr_01、cstr_02做字符串计算、处理逻辑即可
// 这两个指针绑定当前函数的内存上下文，函数返回后会自动回收，不需要手动free

方案2：需要修改字符串/独立持有数据 #

如果需要修改字符串内容，或者要保留字符串副本不影响原text数据，可以手动拷贝实现：

// 自动识别压缩/非压缩存储，计算text中实际字符数据的长度
int len_01 = VARSIZE_ANY_EXHDR(str_01);
int len_02 = VARSIZE_ANY_EXHDR(txt_02);

// 用palloc分配内存，多分配1字节存'\0'结束符
char *cstr_01 = palloc(len_01 + 1);
char *cstr_02 = palloc(len_02 + 1);

// 拷贝实际字符数据
memcpy(cstr_01, VARDATA_ANY(str_01), len_01);
memcpy(cstr_02, VARDATA_ANY(txt_02), len_02);

// 手动补字符串结束符
cstr_01[len_01] = '\0';
cstr_02[len_02] = '\0';

// 处理完成后如果不需要继续持有这两个字符串，调用pfree释放即可
// 如果字符串要作为返回结果的一部分，则不需要手动释放
// pfree(cstr_01);
// pfree(cstr_02);

注意事项 #

• 内存分配优先使用PG内置的palloc/pfree，不要直接用C标准库的malloc/free，palloc会绑定当前内存上下文，能避免内存泄漏
• 上述提到的*_PP/VARSIZE_ANY*/VARDATA_ANY宏会自动处理TOAST压缩大字段的解压逻辑，不需要你手动判断字段是否压缩
• 不要直接访问text结构的vl_len_或者直接用VARDATA宏取数据，遇到压缩存储的字段会拿到错误内容

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