在 PostgreSQL 中声明局部变量？

Question

这里有一个几乎相同但没有真正回答的问题。

我正在将应用程序从 MS SQL Server 迁移到 PostgreSQL。在代码中的许多地方我使用局部变量，因此我想进行需要较少工作的更改，所以您能告诉我哪种是翻译以下代码的最佳方法吗？

-- MS SQL Syntax: declare 2 variables, assign value and return the sum of the two 
declare @One integer = 1
declare @Two integer = 2
select @One + @Two as SUM

返回：

SUM
-----------
3

(1 row(s) affected)

如果 Postgresql 8.4 甚至 9.0 包含可以简化翻译的重要功能，我将使用它。

Answer 1

Postgresql 历史上不支持命令级别的过程代码 - 仅在函数内。但是，在 Postgresql 9 中，添加了对执行内联代码块 有效地支持类似的东西，尽管语法可能有点奇怪，并且与使用 SQL Server 可以做的事情相比有很多限制。值得注意的是，内联代码块无法返回结果集，因此不能用于上面概述的内容。

一般来说，如果您想编写一些过程代码并让它返回结果，您需要将其放在函数中。例如：

CREATE OR REPLACE FUNCTION somefuncname() RETURNS int LANGUAGE plpgsql AS $
DECLARE
  one int;
  two int;
BEGIN
  one := 1;
  two := 2;
  RETURN one + two;
END
$;
SELECT somefuncname();

据我所知，PostgreSQL 有线协议不允许诸如命令返回多个结果集之类的事情。因此，您不能简单地将 T-SQL 批处理或存储过程映射到 PostgreSQL 函数。

Answer 2

您可以在 DECLARE 子句中使用 :=、= 声明局部变量，如下所示：

CREATE FUNCTION my_func()
RETURNS INT
AS $
DECLARE
  value1 INT := 1; -- Here
  value2 INT = 2; -- Here
  value3 INT DEFAULT 3; -- Here
  value4 CONSTANT INT := 4; -- Here
  value5 INT; -- Here
BEGIN
  RETURN value1 + value2 + value3;
END;
$ LANGUAGE plpgsql;

*Memos:

• := 、= 和 DEFAULT 是相同的。
• 尝试更改常量局部变量 value4 会出错。
• 未初始化的局部变量value5为NULL。
• 您可以在 PL/pgSQL 函数和过程以及 DO 语句中使用 DECLARE 子句声明局部变量。

然后，调用 my_func() 返回 6 ，如下所示：

postgres=# SELECT my_func();
 my_func
---------
       6
(1 row)