PostgreSQL用C完成存储过程实例
最近给客户写了一个PostgreSQL用C写的存储过程的例子,在此记录一下。
目的:用C完成一个存储过程例子,存储过程实现对表某一段进行update。
准备工作
1、安装数据库
2、建立表test
highgo=# create table test(id int, name text, label int);
CREATE TABLE
3、建立C文件,C代码如下:
#include "postgres.h"
#include "executor/spi.h"
#include "utils/builtins.h"
#ifdef PG_MODULE_MAGIC
PG_MODULE_MAGIC;
#endif
int mydelete(int key);
int
mydelete(int key)
{
char command[128]; //视命令长短建立相应大小的数组
int ret;
int proc; //对表数据操作的行数
/* 将命令赋值到command */
sprintf(command, "update test set label = 0 where id = %d and label = 1; ", key);
SPI_connect(); //内部链接
ret = SPI_exec( command, 0); //执行操作
proc = SPI_processed; //为行数赋值
SPI_finish(); //中断连接
return (proc); //将操作行数作为返回结果
}
数据库api参考文档:http://www.postgresql.org/docs/9.4/static/spi.html
编译到安装
4、gcc编译
gcc -fpic -I/opt/HighGo/db/20150401/include/postgresql/server/ -shared -o myapi.so myapi.c
5、复制到lib目录下
cp myapi.so /opt/HighGo/db/20150401/lib/postgresql/
6、加载到服务器
highgo=# load 'myapi';
LOAD
7、建立函数
highgo=# create function mydele(integer) returns integer as '$libdir/myapi.so','mydelete' language c strict;
CREATE FUNCTION
highgo=#
8、效果
highgo=# insert into test values (1,'jim',1);
INSERT 0 1
highgo=# insert into test values (2,'tom',1);
INSERT 0 1
highgo=# select * from test;
id | name | label
----+------+-------
1 | jim | 1
2 | tom | 1
highgo=# select mydele(1);
mydele
--------
1
(1 row)
highgo=# select * from test;
id | name | label
----+------+-------
2 | tom | 1
1 | jim | 0
------------------------------------华丽丽的分割线------------------------------------
------------------------------------华丽丽的分割线------------------------------------
PostgreSQL 的详细介绍:请点这里
PostgreSQL 的下载地址:请点这里