PostgreSQL 11 新特性之存储过程

作者: 不剪发的Tony老师
毕业于北京航空航天大学,十多年数据库管理与开发经验,目前在一家全球性的金融公司从事数据库架构设计。CSDN学院签约讲师以及GitChat专栏作者。csdn上的博客收藏于以下地址:https://tonydong.blog.csdn.net

文章目录

PostgreSQL 11 增加了一个新的模式对象:存储过程(Stored Procedure)。存储过程和函数(Function)类似,不过它没有返回值。

存储过程最大的优势就是能够支持事务控制,也就是可以在定义中使用 COMMIT 或者 ROLLBACK 语句。

使用 CREATE\ALTER\DROP PROCEDURE 命令创建\修改\删除存储过程,使用 CALL 命令调用存储过程。支持存储过程的服务器端编程语言包括:PL/pgSQL、PL/Perl、PL/Python、PL/Tcl 以及 SPI 。

先来看一个简单示例:

CREATE TABLE tbl(id int, name text);

CREATE PROCEDURE proc1(pid integer, pname text)
LANGUAGE sql
AS $$
INSERT INTO tbl(id, name) VALUES (pid, pname);
$$;

   

使用 CALL 语句调用存储过程:

CALL proc1(1, 'the first value');
CALL proc1(pname=>'the second value', pid=>2);

    
查看一下结果:

SELECT * FROM tbl;
 id |       name       
----+------------------
  1 | the first value
  2 | the second value
(2 rows)


存储过程增加了对事务的支持:

CREATE OR REPLACE PROCEDURE proc2()
 LANGUAGE plpgsql
AS $$
BEGIN
 FOR i IN 0..9 LOOP
  INSERT INTO tbl(id, name) VALUES (i, 'value: '|| i);
  IF i % 2 = 0 THEN
   COMMIT;
  ELSE
   ROLLBACK;
  END IF;
 END LOOP;
END;
$$ ;

   
    
调用存储过程 proc2,即使没有参数,仍然需要加上括号(()):

CALL proc2();

   
查看结果:

SELECT * FROM tbl;
 id |       name       
----+------------------
  1 | the first value
  2 | the second value
  0 | value: 0
  2 | value: 2
  4 | value: 4
  6 | value: 6
  8 | value: 8
(7 rows)

    
只有id 为偶数的记录成功插入表中,奇数都被 ROLLBACK 语句回退了。

在 psql 中,使用 \df 命令查看存储过程:

\df
                         List of functions
 Schema | Name  | Result data type |   Argument data types   | Type
--------+-------+------------------+-------------------------+------
 public | add   | integer          | integer, integer        | func
 public | proc1 |                  | pid integer, pname text | proc
 public | proc2 |                  |                         | proc
(3 rows)

   
对于存储过程,Type 的值为 proc。使用 \sf 命令可以查看存储过程的定义:

\sf proc1
CREATE OR REPLACE PROCEDURE public.proc1(pid integer, pname text)
 LANGUAGE sql
AS $procedure$
INSERT INTO tbl(id, name) VALUES (pid, pname);
$procedure$

PROCDEURE 目前还不支持自治事务(autonomous transaction)。

相关参考文档:
CREATE PROCEDURE
ALTER PROCEDURE
DROP PROCEDURE
CALL