PL/pgSQL 声明变量 - SQL

在 PL/pgSQL 中，变量是用于存储数据值的标识符。这些变量可以是任何数据类型，例如整数、浮点数、字符串、日期等。在本文中，我们将学习如何在 PL/pgSQL 中声明变量，并在示例程序中使用它们。

PL/pgSQL 中的变量声明

在 PL/pgSQL 中，我们可以使用 DECLARE 关键字来声明变量。可以在 BEGIN 关键字之后或任何子块内部声明变量。

以下是在 PL/pgSQL 中声明变量的语法：

DECLARE variable_name datatype [NOT NULL] [COLLATE collation_name] [DEFAULT initial_value];

在上面的语法中，

• variable_name 是变量的名称。
• datatype 是变量的数据类型。
• NOT NULL 和 COLLATE collation_name 是可选的参数。
• initial_value 是可选的初始值。

要声明一个名称为 num，类型为整数的变量，可以使用以下代码：

DECLARE num INTEGER;

PL/pgSQL 中的变量赋值

要在 PL/pgSQL 中给变量赋值，我们可以在变量名称之后使用等号并赋予值。

以下是在 PL/pgSQL 中赋值的语法：

variable_name := value;

在上述语法中，

• variable_name 是变量的名称。
• value 是要分配给变量的值。

以下是给变量 num 赋值的示例：

DECLARE num INTEGER;
num := 10;

PL/pgSQL 中使用变量

变量在 PL/pgSQL 中表现得像常规 SQL 查询中的列名一样。要使用变量，只需在查询或表达式中使用变量名称即可。

以下是在 PL/pgSQL 中使用变量的示例：

DECLARE num INTEGER;
num := 10;

IF num > 0 THEN
  RAISE NOTICE 'The value of num is %', num;
END IF;

完整示例

CREATE OR REPLACE FUNCTION calculate_total(price INTEGER, quantity INTEGER)
RETURNS INTEGER AS $$
DECLARE
  total INTEGER := 0;
BEGIN
  total := price * quantity;
  RETURN total;
END;
$$ LANGUAGE plpgsql;

在上面的示例中，我们声明了一个名为 total 的变量，该变量的类型为整数，并将其初始化为 0。然后，我们在函数中使用该变量计算 price 和 quantity 的乘积，并将结果返回。

结论

在 PL/pgSQL 中声明和使用变量非常简单。变量可以声明为任何数据类型，并且可以在查询和表达式中使用。在编写 PL/pgSQL 函数或存储过程时，使用变量也很有用。