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

Bertha 。 2024-02-05 16:49 13阅读 0赞

## 场景 ##

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

[PostGresSQL简介与Windows上的安装教程\_postgressql windows安装免费吗?-CSDN博客][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博客][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 $$;

运行结果：

![326afc279ed8417cae2f83a64e0428c5.png][]

示例说明：

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

其中，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 $$;

执行结果：

### ![fe86cc08f9c44af4804be55e9897c008.png][] ###

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

示例运行结果：

### ![67cdbc4dcd884cb38e9bf4a4ff72585e.png][] ###

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

[PostGresSQL_Windows_postgressql windows_-CSDN]: https://blog.csdn.net/BADAO_LIUMANG_QIZHI/article/details/113981563
[C_SpringBoot-CSDN]: https://blog.csdn.net/badao_liumang_qizhi
[326afc279ed8417cae2f83a64e0428c5.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/01/30/187e6adbb9804b03a3d17ae53ec913a8.png
[fe86cc08f9c44af4804be55e9897c008.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/01/30/c36bb50dce2a4d7d9007dd4b8ebb6dad.png
[67cdbc4dcd884cb38e9bf4a4ff72585e.png]: https://image.dandelioncloud.cn/pgy_files/images/2024/01/30/4e3f9a619e154c49878e04bebe5f5eeb.png