postgres扩展开发

时间：2023-01-01 16:32:15

扩展开发的基本组成

demo--1.0.sql

demo.c

demo.control

Makefile

demo.c当中包含了自定义函数的实现，纯C语言，目录下可包含多个.c文件。
demo—1.0.sql对自定义函数的声明，在pg启动的时候会执行这个sql。
demo.control这个应该是版本控制，以及module_pathname告诉PG在执行到用户自定义函数的时候去这个路径下找库文件。
Makefile编译文件。

开发的基本技巧

写扩展函数的基本三部曲
第一步：（必须要的）
#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif
第二步：（必须要的）
PG_FUNCTION_INFO_V1(test_add_fun)；
第三步：（必须要的）
test_add_fun函数功能的实现。

获取外部参数函数是PG_GETARG_XXXX();
返回值的返回函数PG_RETURN_XXXX();
Ps：按照这一套你就能进行扩展开发了。

demo.c代码如下：

  #ifdef PG_MODULE_MAGIC

  PG_MODULE_MAGIC;

  #endif

  PG_FUNCTION_INFO_V1(test_add_fun);

  Datum test_add_fun(PG_FUNCTION_ARGS);

  Datum test_add_fun(PG_FUNCTION_ARGS)

 {

         int sum,a,b;

         a = PG_GETARG_INT32();

         b = PG_GETARG_INT32();

         sum = a + b;

         PG_RETURN_INT32(sum);

 }

Makfile如下:

 MODULE_big = demo

 IBASE_VERSION=1.0

 OBJS =  demo.o

 EXTENSION = demo

 DATA = demo--1.0.sql

 SHLIB_LINK += $(filter -lm, $(LIBS))

 USE_PGXS=

 ifdef USE_PGXS

 PG_CONFIG = /usr/local/postgres/bin/pg_config

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

 include $(PGXS)

 else

 subdir = contrib/demo

 top_builddir = ../..

 include $(top_builddir)/src/Makefile.global

 include $(top_srcdir)/contrib/contrib-global.mk

 endif

编译安装

[root@centos01 demo]# make

gcc -Wall -Wmissing-prototypes -Wpointer-arith -Wdeclaration-after-statement -Wendif-labels -Wmissing-format-attribute -Wformat-security -fno-strict-aliasing -fwrapv -fexcess-precision=standard -g -O0 -fpic -I. -I./ -I/usr/local/postgres/include/server -I/usr/local/postgres/include/internal -D_GNU_SOURCE   -c -o demo.o demo.c

gcc -Wall -Wmissing-prototypes -Wpointer-arith -Wdeclaration-after-statement -Wendif-labels -Wmissing-format-attribute -Wformat-security -fno-strict-aliasing -fwrapv -fexcess-precision=standard -g -O0 -fpic -shared -o demo.so demo.o -L/usr/local/postgres/lib -Wl,--as-needed -Wl,-rpath,'/usr/local/postgres/lib',--enable-new-dtags  -lm

[root@centos01 demo]# make install

/usr/bin/mkdir -p '/usr/local/postgres/lib'

/usr/bin/mkdir -p '/usr/local/postgres/share/extension'

/usr/bin/mkdir -p '/usr/local/postgres/share/extension'

/usr/bin/install -c -m   demo.so '/usr/local/postgres/lib/demo.so'

/usr/bin/install -c -m  demo.control '/usr/local/postgres/share/extension/'

/usr/bin/install -c -m  demo--1.0.sql '/usr/local/postgres/share/extension/'

创建扩展并使用

postgres=# create extension demo;

CREATE EXTENSION

postgres=# \sf test_add_fun

CREATE OR REPLACE FUNCTION public.test_add_fun(a integer, b integer)

 RETURNS integer

 LANGUAGE c

 STRICT

AS '$libdir/demo', $function$test_add_fun$function$

postgres=# select test_add_fun(,);

 test_add_fun

--------------

( row)

相关文章