使用数组输入和数组输出为 PostgreSQL 创建简单的 C 函数

Question

谁能分享我如何为 PostgreSQL 创建 C 函数的示例，该函数将两个整数的数组作为输入并返回数组作为输出？

对于简单的整数，我有：

#include "postgres.h"
#include <fmgr.h>

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

int
add_one(int arg) {
    arg++;
    return arg;
}

在 PostgreSQL 中编译后：

load '/usr/lib/postgresql/9.1/lib/add_one';

    create or replace function add_one(integer)
      returns integer as
    '/usr/lib/postgresql/9.1/lib/add_one', 'add_one'
      language c;

    select add_one(1);

我需要类似的东西：

#include "postgres.h"
#include <fmgr.h>

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

int
add_one(int[] arg) {
    arg[1]++;
    arg[2] = arg[2] + 2
    return arg[];
}

在 PostgreSQL 中：

load '/usr/lib/postgresql/9.1/lib/add_one';

create or replace function add_one(integer[])
     returns integer[] as
    '/usr/lib/postgresql/9.1/lib/add_one', 'add_one'
      language c;

    select add_one(ARRAY[1::int,1::int]);

I have tried 修改 numeric.c 中的某些函数，但到目前为止没有任何成功。

Answer 1

您的代码甚至无法远程开始按所编写的那样工作。 PostgreSQL 不会将 Pg 级别的数组作为 int[] 传递，而是通过 PG_FUNCTION_ARGS (fcinfo) 中的函数上下文传递它们，并通过 PG_GETARG_ARRAYTYPE_P 宏访问它们。

基本扩展功能指导见the docs on C language functions。

看看array_cat在src/backend/utils/adt/array_userfuncs.c中的定义，或者array_remove在src/backend/utils/adt/arrayfuncs.c中的定义。或许多其他选项。

你的骨架看起来像：

PG_FUNCTION_INFO_V1(add_arrays);

Datum
add_arrays(PG_FUNCTION_ARGS)
{
    ArrayType  *array1, *array2, *resultarray;

    array1 = PG_GETARG_ARRAYTYPE_P(0);
    array2 = PG_GETARG_ARRAYTYPE_P(1);

    /* Loop over the array bodies and do your mapping to generate resultarray here */

    PG_RETURN_ARRAYTYPE_P(resultarray);
}

PostgreSQL C 数组 API 很糟糕，所以我没有时间填写函数体。关键是你的函数签名是完全错误的——你完全误解了它是如何工作的，你写的甚至不可能被执行。

然后它会被声明为：

create or replace function add_arrays(integer[], integer[])
returns integer[] as
'add_arrays', 'add_arrays'
language c immutable strict;

strict 很重要；我提供的函数骨架不检查空输入，所以你需要告诉执行器不要用它们调用它。

如果src/backend/utils/adt/arrayfuncs.c 中的array_map 有一个map2 或zip 变体来执行两个数组的锁步迭代，那就太好了。不幸的是它没有，所以你需要自己迭代它们。

重新阅读您的问题后，我现在想知道您的意思是 单个 int[] 数组与 两个元素 并且您想要 integer 结果 那是数组的总和。如果是这样，看看intarray 模块是如何工作的；它具有处理简单整数数组的简化函数。