postgresql - 创建 PostgreSQL 内部 C 函数的副本并将其加载为用户定义的函数

Question

我想创建位于 src/backend/utils/adt/numeric.c 中的 C 函数 int2_avg_accum 的修改版本。我想我可以(作为开始)只编译 numeric.h 并将 int2_avg_accum 作为用户定义的函数加载。

我做了什么:

1. 将 PG_MODULE_MAGIC 添加到 numeric.h(如 here 所述)
2. 在 numeric.h 中将 int2_avg_accum 重命名为 int2_avg_accum2
3. 按照文档中的描述编译了 numeric.h(没有错误，没有警告)(cc -fpic -Ipg_config --includedir-server-c numeric.c 然后 cc -shared -o numeric.so numeric.o)
4. 在 PostgreSQL 中创建了一个函数:

.

create or replace function int2_avg_accum2(bigint[], smallint)
  returns bigint[] as
'/usr/lib/postgresql/9.1/lib/numeric', 'int2_avg_accum2'
  language c
  cost 1;
alter function int2_avg_accum2(bigint[], smallint)
  owner to postgres;

当我尝试运行 select int2_avg_accum2(array[1::bigint,1],1::smallint); 我只收到消息(在 pgAdmin 中):“你想尝试重新连接到数据库？”。没有其他消息或错误。

当我调用该函数时，我在 /var/log/postgresql/postgresql-9.1-main.log 中看到以下内容:

2013-12-03 09:52:02 CET LOG:  server process (PID 3366) was terminated by signal 11: Segmentation fault
2013-12-03 09:52:02 CET LOG:  terminating any other active server processes
2013-12-03 09:52:02 CET WARNING:  terminating connection because of crash of another server process
2013-12-03 09:52:02 CET DETAIL:  The postmaster has commanded this server process to roll back the current transaction and exit, because another server process exited abnormally and possibly corrupted shared memory.
2013-12-03 09:52:02 CET HINT:  In a moment you should be able to reconnect to the database and repeat your command.
2013-12-03 09:52:02 CET LOG:  all server processes terminated; reinitializing
2013-12-03 09:52:02 CET LOG:  database system was interrupted; last known up at 2013-12-03 09:50:53 CET
2013-12-03 09:52:02 CET LOG:  database system was not properly shut down; automatic recovery in progress
2013-12-03 09:52:02 CET LOG:  record with zero length at 0/B483EA0
2013-12-03 09:52:02 CET LOG:  redo is not required
2013-12-03 09:52:03 CET LOG:  autovacuum launcher started
2013-12-03 09:52:03 CET LOG:  database system is ready to accept connections

为了获得 int2_avg_accum 的工作副本，我必须做哪些不同的事情？

Answer 1

根据评论，psql 客户端询问您是否希望重新连接的原因是后端出现段错误。

可以从这样的崩溃中收集核心转储并使用调试器(例如 gdb)检查它以找出崩溃的确切位置。然而，我最好的猜测是它正在崩溃，因为你已经将一个大文件编写为 postgresql 的核心组件，单独编译它，并试图将它作为扩展模块加载。

文件 numeric.c 包含大量函数、静态变量和数据结构，而您只想复制其中的一个。所有这些函数、变量等都已经存在于正在运行的 postgresql 系统中。当你编译你的 numeric.c 版本并加载它时，你添加的新函数将引用你库中的函数和变量，而不是使用主 postgresql 程序中的函数和变量。它可能引用了未正确初始化的数据结构，导致它崩溃。

我建议您从一个空白文件开始，然后仅复制 numeric.c 中的 int2_avg_accum 函数(按您所做的重命名)。如果该函数正在调用 postgresql 中的其他函数，或者引用变量，它将使用主 postgresql 二进制文件中的函数和变量，这就是您想要的。您可以 #include 原始 numeric.h 以获取所有外部函数的声明。

将函数定义为内部函数的方式与将其加载为动态加载模块时需要定义的方式之间还有一些其他差异:

• 您需要通过添加宏来指定您使用的是 V1 调用约定:

  PG_FUNCTION_INFO_V1(int2_avg_accum2);

  如果缺少这也会导致段错误，因为 postgresql 将采用版本 0 调用约定，这与函数定义不匹配!

• 正如您所指出的，您必须包含 PG_MODOULE_MAGIC。

对我有用的完整文件是:

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

#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

typedef struct Int8TransTypeData
{
    int64       count;
    int64       sum;
} Int8TransTypeData;

PG_FUNCTION_INFO_V1(int2_avg_accum2);

Datum
int2_avg_accum2(PG_FUNCTION_ARGS)
{
    ArrayType  *transarray;
    int16       newval = PG_GETARG_INT16(1);
    Int8TransTypeData *transdata;

    /*
     * If we're invoked as an aggregate, we can cheat and modify our first
     * parameter in-place to reduce palloc overhead. Otherwise we need to make
     * a copy of it before scribbling on it.
     */
    if (AggCheckCallContext(fcinfo, NULL))
        transarray = PG_GETARG_ARRAYTYPE_P(0);
    else
        transarray = PG_GETARG_ARRAYTYPE_P_COPY(0);

    if (ARR_HASNULL(transarray) ||
        ARR_SIZE(transarray) != ARR_OVERHEAD_NONULLS(1) + sizeof(Int8TransTypeData))
        elog(ERROR, "expected 2-element int8 array");

    transdata = (Int8TransTypeData *) ARR_DATA_PTR(transarray);
    transdata->count++;
    transdata->sum += newval;

    PG_RETURN_ARRAYTYPE_P(transarray);
}

编译:

gcc -I/usr/pgsql-9.2/include/server -fPIC -c my_avg_accum.c
gcc -shared -o my_avg_accum.so my_avg_accum.o

我在 Centos 6 上使用 Postgresql 9.2。您可能需要根据您的设置调整路径。