PostgreSQL C数组扩展开发故障求助：实现数组元素加5遇阻

要开发一个PostgreSQL C扩展，接收int[]数组输入，为每个元素加5后返回新数组（不修改原数组），但测试阶段连返回原数组的代码都出现崩溃或报错。

测试代码与异常情况 #

第一段测试代码 #

#include "postgres.h"
#include "fmgr.h"
#include "utils/array.h"
#include "catalog/pg_type.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

PG_FUNCTION_INFO_V1(arr);

Datum 
arr(PG_FUNCTION_ARGS)
{
    Datum arry = PG_GETARG_DATUM(0);
    ArrayType *inputArray = DatumGetArrayTypeP(arry);
    PG_RETURN_DATUM(arry);
}

执行ArrayType *inputArray = DatumGetArrayTypeP(arry);时，数据库直接断开连接。

第二段测试代码 #

#include "postgres.h"
#include "fmgr.h"
#include "utils/array.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

PG_FUNCTION_INFO_V1(my_return_array_as_is);

Datum
my_return_array_as_is(PG_FUNCTION_ARGS)
{
    // Get the input array
    ArrayType *inputArray = PG_GETARG_ARRAYTYPE_P(0);

    // Normally, you might process the array here

    // Return the array as is
    PG_RETURN_ARRAYTYPE_P(inputArray);
}

执行到ArrayType *inputArray = PG_GETARG_ARRAYTYPE_P(0);行时直接报错。

SQL函数定义 #

CREATE OR REPLACE FUNCTION retarr1(int[])
RETURNS int[]
AS 'MODULE_PATHNAME', 'retarr1'
LANGUAGE C STRICT;

测试查询：SELECT retarr1(ARRAY[10,20])

编译环境与命令 #

• 系统：Windows 11
• 编译器：GCC
• PostgreSQL版本：16.2
  编译命令：

gcc  -I "C:/Program Files/PostgreSQL/16/include/server/port/win32/" -I "C:/Program Files/PostgreSQL/16/include/server" -I "C:/Program Files/PostgreSQL/16/include"  -o arr.o -c arr.c

gcc -g -shared -o arr.dll arr.o -L"C:/Program Files/PostgreSQL/16/lib" -lpostgres

解决方案 #

1. 核心问题：函数名不匹配 #

SQL定义中指定的C函数符号是retarr1，但代码中注册的函数名是arr或my_return_array_as_is，PostgreSQL无法找到正确的函数入口，导致调用崩溃。

2. 修正后的基础测试代码（返回原数组） #

#include "postgres.h"
#include "fmgr.h"
#include "utils/array.h"
#include "catalog/pg_type.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

// 必须与SQL中指定的函数名完全一致
PG_FUNCTION_INFO_V1(retarr1);

Datum 
retarr1(PG_FUNCTION_ARGS)
{
    ArrayType *inputArray = PG_GETARG_ARRAYTYPE_P(0);
    // 直接返回原数组（若要修改元素，需复制数组后操作）
    PG_RETURN_ARRAYTYPE_P(inputArray);
}

3. 完整功能实现（元素加5） #

#include "postgres.h"
#include "fmgr.h"
#include "utils/array.h"
#include "catalog/pg_type.h"
#include "utils/lsyscache.h"

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

PG_FUNCTION_INFO_V1(retarr1);

Datum 
retarr1(PG_FUNCTION_ARGS)
{
    ArrayType *inputArray = PG_GETARG_ARRAYTYPE_P(0);
    Oid elemType = ARR_ELEMTYPE(inputArray);
    int16 elemWidth;
    bool elemByVal;
    char elemAlignment;
    Datum *elems;
    bool *nulls;
    int nelems;
    int i;

    // 验证输入为int数组
    if (elemType != INT4OID)
        ereport(ERROR,
                (errcode(ERRCODE_DATA_TYPE_MISMATCH),
                 errmsg("输入必须是整数数组")));

    // 获取元素类型属性
    get_typlenbyvalalign(elemType, &elemWidth, &elemByVal, &elemAlignment);

    // 解析数组元素
    deconstruct_array(inputArray, elemType, elemWidth, elemByVal, elemAlignment,
                      &elems, &nulls, &nelems);

    // 为每个非空元素加5
    for (i = 0; i < nelems; i++)
    {
        if (!nulls[i])
        {
            int32 val = DatumGetInt32(elems[i]);
            val += 5;
            elems[i] = Int32GetDatum(val);
        }
    }

    // 构建新数组返回
    ArrayType *resultArray = construct_array(elems, nelems, elemType,
                                             elemWidth, elemByVal, elemAlignment);

    // 释放解析数组时分配的内存
    pfree(elems);
    if (nulls)
        pfree(nulls);

    PG_RETURN_ARRAYTYPE_P(resultArray);
}

4. 编译命令修正 #

确保编译架构与PostgreSQL一致（默认64位），且链接库参数顺序正确：

# 编译目标文件
gcc -m64 -I "C:/Program Files/PostgreSQL/16/include/server" -I "C:/Program Files/PostgreSQL/16/include/server/port/win32" -I "C:/Program Files/PostgreSQL/16/include" -o arr.o -c arr.c

# 生成DLL
gcc -m64 -g -shared -o arr.dll arr.o -L"C:/Program Files/PostgreSQL/16/lib" -lpostgres

5. 部署与测试注意事项 #

• 将生成的arr.dll放入PostgreSQL的lib目录，或在CREATE FUNCTION中指定完整路径（如AS 'C:/path/to/arr.dll', 'retarr1'）
• 重启PostgreSQL服务，或执行LOAD 'arr.dll';加载扩展
• 测试时，STRICT函数会直接返回NULL输入，符合预期

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