Postgresql中PL/pgSQL代码块的语法与使用-声明与赋值、IF语句、CASE语句、循环语句

场景

PostGresSQL简介与Windows上的安装教程：

PostGresSQL简介与Windows上的安装教程_postgressql windows安装免费吗?-CSDN博客

除了标准 SQL 语句之外，PostgreSQL 还支持使用各种过程语言（例如 PL/pgSQL、C、PL/Tcl、PL/Python、PL/Perl、PL/Java 等 ）

创建复杂的过程和函数，称为存储过程（Stored Procedure）和自定义函数（User-Defined Function）。

存储过程支持许多过程元素，例如控制结构、循环和复杂的计算。

注：

博客：
霸道流氓气质_C#,架构之路,SpringBoot-CSDN博客

实现

1、PL/pgSQL 代码块结构

[ <<label>> ]
[ DECLARE
?declarations ]
BEGIN
?statements;
?...
END [ label ];

结构说明：

label 是一个可选的代码块标签，可以用于 EXIT 语句退出指定的代码块，或者限定变量的名称；

DECLARE 是一个可选的声明部分，用于定义变量；

BEGIN 和 END 之间是代码主体，也就是主要的功能代码；所有的语句都使用分号（;）结束，

END 之后的分号表示代码块结束。

代码块示例：

DO $$
DECLARE
?name text;
BEGIN
?name := '霸道';
?RAISE NOTICE 'Hello % !', name;
END $$;

运行结果：

示例说明：

以上是一个匿名块，与此相对的是命名块（也就是存储过程和函数）。

其中，DO 语句用于执行匿名块；我们定义了一个字符串变量 name，然后给它赋值并输出一个信息；

RAISE NOTICE用于输出通知消息。

$$用于替换单引号（'），因为 PL/pgSQL 代码主体必须是字符串文本，意味着代码中所有的单引号都必须转义（重复写两次），

显然这种写法很不方便，因此 PL/pgSQL 提供了$$避免单引号问题。

2、嵌套子块

PL/pgSQL 支持代码块的嵌套，也就是将一个代码块嵌入其他代码块的主体中。被嵌套的代码块被称为子块（subblock），

包含子块的代码块被称为外部块（subblock）。子块可以将代码进行逻辑上的拆分，子块中可以定义与外部块重名的变量，

而且在子块内拥有更高的优先级。

示例代码：

DO $$
<<outer_block>>
DECLARE
?name text;
BEGIN
?name := 'outer_block';
?RAISE NOTICE 'This is %', name;
?DECLARE
?name text := 'sub_block';
?BEGIN
?RAISE NOTICE 'This is %', name;
?RAISE NOTICE 'The name FROM the outer block is %', outer_block.name;
?END;
?RAISE NOTICE 'This is %', name;
?
END outer_block $$;

执行结果：

3、声明与赋值

声明变量

variable_name data_type [ NOT NULL ] [ { DEFAULT | := |= } expression ];

variable_name 是变量的名称，通常需要指定一个有意义的名称；

data_type 是变量的类型，可以是任何 SQL 数据类型；

如果指定了 NOT NULL，必须使用后面的表达式为变量指定初始值。

变量声明示例：

DO $$
DECLARE
?name text;
?age numeric(5) DEFAULT 20;
?url varchar := 'www.123.com';
BEGIN
?name := '霸道';
?RAISE NOTICE 'age= %', age;
?RAISE NOTICE 'url= %', url;
END $$;

除了基本的 SQL 数据类型之外，PL/pgSQL 还支持基于表的字段或行或者其他变量定义变量，这里不再实现。

声明常量

如果在定义变量时指定了 CONSTANT 关键字，意味着定义的是常量。常量的值需要在声明时初始化，并且不能修改。

示例代码：

通过定义PI计算圆的面积

DO $$
DECLARE
?PI CONSTANT NUMERIC := 3.14159265;
?radius NUMERIC;
BEGIN
?radius := 5.0;
?RAISE NOTICE 'The area is %', PI * radius * radius;
END $$;

示例运行结果：

4、控制结构

IF语句

PL/pgSQL 提供了三种形式的 IF 语句

? IF … THEN … END IF
? IF … THEN … ELSE … END IF
? IF … THEN … ELSIF … THEN … ELSE … END IF

示例代码：

DO $$ DECLARE
i INTEGER := 3;
j INTEGER := 3;
BEGIN
?IF
??i > j THEN
???RAISE NOTICE'i 大于 j';
??ELSIF i < j THEN
???RAISE NOTICE'i 小于 j';
???ELSE RAISE NOTICE'i 等于 j';
???
??END IF;
??
?END $$;

CASE语句

类似Java中switch-case语句

简单Case语句示例：

DO $$ DECLARE
i INTEGER := 3;
BEGIN
?CASE
???i
???WHEN 1,
???2 THEN
????RAISE NOTICE'one or two';
???
???WHEN 3,
???4 THEN
????RAISE NOTICE'three or four';
????ELSE RAISE NOTICE'other value';
????
???END CASE;
???
??END $$;

搜索CASE语句示例：

DO $$ DECLARE
i INTEGER := 3;
BEGIN
?CASE
???
???WHEN i BETWEEN 0
???AND 10 THEN
????RAISE NOTICE'value is between zero and ten';
???
???WHEN i BETWEEN 11
???AND 20 THEN
????RAISE NOTICE'value is between eleven and twenty';
????ELSE RAISE NOTICE'other value';
????
???END CASE;
??
END $$;

循环语句

PostgreSQL 提供 4 种循环执行命令的语句：

LOOP、WHILE、FOR 和 FOREACH 循环，以及循环控制的 EXIT 和 CONTINUE 语句。

LOOP 用于定义一个无限循环语句，一般需要使用 EXIT 或者 RETURN 语句退出循环，

label 可以用于 EXIT 或者 CONTINUE 语句退出或者跳到执行的嵌套循环中。

LOOP循环示例：

DO $$ DECLARE
i INTEGER := 0;
BEGIN
??LOOP
??EXIT
??WHEN i = 5;
?i := i + 1;
?RAISE NOTICE'Loop: %',
?i;
?
END LOOP;

END $$;

CONTINUE 表示忽略后面的语句，直接进入下一次循环。

示例：

DO $$ DECLARE
i INTEGER := 0;
BEGIN
??LOOP
??EXIT
??WHEN i = 10;
?i := i + 1;
?CONTINUE
??WHEN MOD ( i, 2 ) = 1;
?RAISE NOTICE'Loop: %',
?i;
?
END LOOP;

END $$;

WHILE循环表示当表达式 boolean-expression 的值为真时，循环执行其中的语句；

然后重新计算表达式的值，当表达式的值假时退出循环。

WHILE循环示例：

DO $$ DECLARE
i INTEGER := 0;
BEGIN
??WHILE
??i < 5
??LOOP
??i := i + 1;
?RAISE NOTICE'Loop: %',
?i;
?
END LOOP;

END $$;

FOR 循环可以用于遍历一个整数范围或者查询结果集。

FOR 循环默认从小到大进行遍历，REVERSE 表示从大到小遍历；BY 用于指定每次的增量，默认为 1。

FOR循环示例：

DO $$ BEGIN
?FOR i IN 1..5 BY 2
?LOOP
?RAISE NOTICE'Loop: %',
?i;
?
END LOOP;

END $$;

FOREACH 循环与 FOR 循环类似，只不过变量的是一个数组

FOREACH 循环示例：

DO $$ DECLARE x INT; BEGIN FOREACH x IN ARRAY ( ARRAY [[ 1, 2, 3 ],[ 4, 5, 6 ]]
??)
?LOOP
?RAISE NOTICE'x = %',
?x;
?
END LOOP;

END $$;