Question

我想为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

奇怪的部分是PQsetdbLogin和PQexec都在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。

Answer 1

如果要链接到外部库，则需要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)

但是您的功能还有其他问题：

VarChar是varlena，但是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);

在服务器功能中编写客户端代码通常是错误的主意。

如果您需要在当前会话中运行查询，请使用Server Programming Interface。
我想您的函数只是示例代码，但是您需要在后端终止之前关闭连接，否则会发生连接泄漏。

PostgreSQL C函数：libpq不链接？

1 个答案: