我想为PostgreSQL编写一个C函数。对于此功能,我将需要使用libpq查询一些数据,因此我首先编写了一个虚拟函数来测试此部分:

#define _XOPEN_SOURCE

#include <libpq-fe.h>
#include <postgres.h>
#include <fmgr.h>
#include <funcapi.h>
#include <executor/executor.h>

#include "test.h"

PG_FUNCTION_INFO_V1(getAnnotation);
Datum getAnnotation(PG_FUNCTION_ARGS) {
    // Connection to the database
    PGconn *conn = PQsetdbLogin("localhost",
                         "5432",
                         "",
                         "",
                         "postgres",
                         "postgres",
                         "password");

    // Databases names
    PGresult *res = PQexec (conn, "SELECT user FROM activity LIMIT 1;");
    VarChar* i = PQgetvalue(res, 0, 0);
    PG_RETURN_VARCHAR_P(i);
}

它只是应该返回我的一个表的第一行的第一列。很简单吧?好吧,这是行不通的。

当我尝试在psql中使用它时,它说:
ERROR:  could not load library "/usr/local/lib/postgresql/test.so": Error relocating /usr/local/lib/postgresql/test.so: PQexec: symbol not found

奇怪的是PQsetdbLoginPQexec都在libpq-fe.h文件中,但是只有第二个会导致错误。如果我在PQexec行中注释,那么PQsetdbLogin也会引发错误。

这是我用来构建代码的Makefile:
PG_CPPFLAGS = -I$(libpq_srcdir)
LDFLAGS_INTERNAL = -L$(libdir)
SHLIB_LINK_INTERNAL = $(libpq)
SHLIB_PREREQS = submake-libpq

EXTENSION = test
DATA = test--0.1.sql
MODULES = test

# REGRESS = ... # Script for tests

PG_CONFIG = pg_config
PGXS := $(shell $(PG_CONFIG) --pgxs)
include $(PGXS)

如您所见,我在其中链接了libpq,因此一切正常。

我在Docker容器中使用PostgreSQL 9.6。

最佳答案

如果要链接到外部库,则需要SHLIB_LINK,但这仅在使用MODULE_big而不是MODULES时有效。

一个有效的Makefile将是

PG_CPPFLAGS = -I$(libpq_srcdir)
SHLIB_LINK = $(libpq)

EXTENSION = test
DATA = test--0.1.sql
MODULE_big = test
OBJS = test.o

PG_CONFIG = pg_config
PGXS := $(shell $(PG_CONFIG) --pgxs)
include $(PGXS)

但是您的功能还有其他问题:
  • VarCharvarlena,但是PQgetvalue返回char *

    您必须使用以下代码将值转换为varlena:

    VarChar *result;
    char *c = PQgetvalue(res, 0, 0);
    result = (VarChar *) palloc(strlen(c) + VARHDRSZ);
    strncpy(VARDATA(result), c, strlen(c));
    SET_VARSIZE(result, strlen(c) + VARHDRSZ);
    
  • 在服务器功能中编写客户端代码通常是错误的主意。

    如果您需要的是在当前 session 中运行查询,请使用Server Programming Interface
  • 我想您的函数只是示例代码,但是您需要在后端终止之前关闭连接,否则会发生连接泄漏。
  • 关于PostgreSQL C函数: libpq doesn't link?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57871869/

    10-10 04:14