使用C语言写PostgreSQL函数

引言:

使用C语言写PostgreSQL中的函数可以很方便的扩展PostgreSQL数据库的功能。由C语言写的函数,使用方法与PostgreSQL中内置的函数基本没有差别,性能基本也没有什么差别。本文一步一步教你如何使用C语言写PostgreSQL函数:

1. 先根据一个简单函数的生成一个动态库,看PostgreSQL能否使用

源文件为cfunc.c,内容如下:

#include "postgres.h"
int add_one(int arg)
{
    return arg + 1;
}

然后我们编译生成cfunc.so,并把此文件拷贝到PostgreSQL的库路径下:

gcc -I`pg_config --includedir-server` -c cfunc.c
gcc -shared -o cfunc.so cfunc.o
cp cfunc.so `pg_config --libdir`/.

我们来测试一下:

osdba@osdba-laptop:~/src/ctest/pgfunc$ psql
psql (9.3.2)
Type "help" for help.

osdba=# load 'cfunc.so';
ERROR:  incompatible library "/home/osdba/pgsql9.3.2/lib/cfunc.so": missing magic block HINT: Extension libraries are required to use the PG_MODULE_MAGIC macro. 

发现报“missing magic block”的错误。那么什么是“magic block”呢? PostgreSQL使用的动态库中要求打上一个特殊的标记,当PostgreSQL装载这个动态库时,会检查这个标记,如果没有发现这个标记或发现这个标记与需求的标记不同(如版本不对上)则会报上面这个错误。

2. 在动态库中加上PostgreSQL的“magic block”

那么如何在动态库中增加这个“magic block”标记呢?方法是在源代码中加下面两行:

#include "postgres.h"
#include "fmgr.h"
#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif

这样之后,我们的源代码最终如下:

#include "postgres.h"
#include "fmgr.h"
#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif
int add_one(int arg)
{
    return arg + 1;
}

再编译一下:

osdba@osdba-laptop:~/src/ctest/pgfunc$ gcc -I`pg_config --includedir-server` -c cfunc.c osdba@osdba-laptop:~/src/ctest/pgfunc$ gcc -shared -o cfunc.so cfunc.o
/usr/bin/ld: cfunc.o: relocation R_X86_64_32 against `.rodata' can not be used when making a shared object; recompile with -fPIC cfunc.o: error adding symbols: Bad value collect2: error: ld returned 1 exit status 

根据报错信息,我们知道在编译时,需要加“-fPIC”的参数,再重试一下:

osdba@osdba-laptop:~/src/ctest/pgfunc$ gcc -I`pg_config --includedir-server` -fPIC  -c cfunc.c
osdba@osdba-laptop:~/src/ctest/pgfunc$ gcc -shared -o cfunc.so cfunc.o
osdba@osdba-laptop:~/src/ctest/pgfunc$ cp cfunc.so `pg_config --libdir`/.

编译通过,我们再重试一下:

osdba@osdba-laptop:~/src/ctest/pgfunc$ psql
psql (9.3.2)
Type "help" for help.

osdba=# load 'cfunc.so';
LOAD
osdba=#

装载成功了,我们来使用一下这个动态库,使用之前需要先创建addone函数,创建语句如下:

CREATE FUNCTION add_one(integer) RETURNS integer AS '$libdir/cfunc1', 'add_one' LANGUAGE C STRICT;

执行情况如下:

osdba=# CREATE FUNCTION add_one(integer) RETURNS integer AS '$libdir/cfunc', 'add_one' LANGUAGE C STRICT;
CREATE FUNCTION osdba=# 

创建成功了,使用一下:

osdba=# select add_one(1);
 add_one 
---------
 2
(1 row)
osdba=# select add_one(23);
 add_one 
---------
      24
(1 row)

3. 总结注意事项

总结一下:

  1. 程序中的源代码中要加宏PG_MODULE_MAGIC;
  2. 程序中要加#include "fmgr.h"
  3. 编译时要加-fPIC选项。

如果没有注意到以上三点,load动态库时,都可能报“missing magic block”错误。

4. 结尾

实际上面描述的是一个最简单的使用C语言写PostgreSQL函数的方法。为了让写出来函数能更好的与数据库之间交互,PostgreSQL搞了一个更高级的调用规则,前面的方法为“Version 0 Calling Conventions”,更高级的调用规则称之为“Version 1 Calling Conventions”,这个调用规则下,函数的声明都有统一的如下形式:

Datum funcname(PG_FUNCTION_ARGS)

同时为了告诉PostgreSQL使用的是版本1的调用规则,需要加函数:

PG_FUNCTION_INFO_V1(funcname);

具体的使用方法见 PostgreSQL官方文件 ,我在这里只做抛砖引玉。


http://www.tuicool.com/articles/j6JFFr

相关文章