我想为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。
最佳答案
如果要链接到外部库,则需要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);
如果您需要的是在当前 session 中运行查询,请使用Server Programming Interface。
关于PostgreSQL C函数: libpq doesn't link?,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57871869/