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

发布时间 2023-12-19 17:39:35作者: 霸道流氓

场景

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

https://blog.csdn.net/BADAO_LIUMANG_QIZHI/article/details/113981563

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

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

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

注:

博客:
https://blog.csdn.net/badao_liumang_qizhi

实现

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 $$;