PL/pgSQL：如何将表的列分配给声明的参数？

Question

看看这个简单的 PL/pgSQL 函数：

create or replace function test() returns void()
as $$
declare 
a int;
begin
select more_than_one_value into a from my_table;
-- do some other job using a
end;
$$ 
language plpgsql;

调用此函数时，只选择一个行，因为select into 只将一个值传递给a，尽管more_than_one_value 列有很多值。

是否有任何便捷的方法可以从表中选择多个值，然后将它们分配给变量 a ？我知道create temp table as select 是一种方法。我正在寻找替代品。

Answer 1

您可以使用几个选项：

• 您可以在一个INTO 子句中使用多个变量：

  SELECT f1, f2, f3, f4 INTO v1, v2, v3, v4 FROM ...

• 您可以使用记录类型：

  DECLARE
   a my_table%rowtype
  BEGIN
   SELECT * INTO a FROM my_table WHERE ...
   INSERT INTO another_table VALUES (a.*);
   ...

• 或者完全忘记变量：

  INSERT INTO another_table
   SELECT * FROM my_table WHERE ...

Answer 2

为此，您需要一个 循环。

多行

如果 "many values" 应该表示 多行，我建议使用带有隐式光标的 FOR loop：

CREATE OR REPLACE FUNCTION test_rows()
  RETURNS void AS
$func$
DECLARE
   i int;  -- scalar variable
BEGIN
   FOR i IN  -- repeated assignment
      SELECT more_than_one_value FROM my_table
   LOOP
      -- do something with i
   END LOOP;
END
$func$  LANGUAGE plpgsql;

在许多情况下，有更好的基于集合的解决方案（纯 SQL）来代替循环。

more_than_one_value 列的数据类型必须与变量（或参数）的类型匹配，否则赋值时会出错。

i 在这里是一个变量，而不是一个参数。后一个术语是指在调用中传递给函数的变量。

• Postgres FOR LOOP
• FOR loop on PLpgSQL function result

数组

如果 "many values" 应该表示数组，我建议FOREACH:

CREATE OR REPLACE FUNCTION test_arr()
  RETURNS void AS
$func$
DECLARE
   i int[];   -- array variable
   elem int;  -- scalar variable
BEGIN
   SELECT INTO i more_than_one_value
   FROM my_table LIMIT 1; -- assign single array column

   FOREACH elem IN ARRAY i
   LOOP
      -- do something with elem
   END LOOP;

END
$func$  LANGUAGE plpgsql;

• Postgres - array for loop