FOREACH 循环的 PostgreSQL PL/pgSQL 语法错误

use*_*258 2 postgresql plpgsql

我正在尝试通过编写一些简单的程序来学习 PL/pgSQL。为了了解 FOREACH 循环,我编写了以下内容:

CREATE OR REPLACE FUNCTION test(int[]) RETURNS void AS $$  
DECLARE                                                     
    window INT;                                             
BEGIN                                                       
    FOREACH window IN ARRAY $1                              
    LOOP                                                    
        EXECUTE 'SELECT $1' USING window;                   
    END LOOP;                                               
$$ LANGUAGE plpgsql;                                        

SELECT test(ARRAY [30,60]);    
Run Code Online (Sandbox Code Playgroud)

我希望该代码片段首先打印30,然后打印60。但是,我收到以下错误。

psql:loop.sql:11: ERROR:  syntax error at end of input
LINE 7:         EXECUTE 'SELECT $1' USING window;
                                                ^
psql:loop.sql:13: ERROR:  function test(integer[]) does not exist
LINE 1: SELECT test(ARRAY [30,60]);
               ^
HINT:  No function matches the given name and argument types. You might need
 to add explicit type casts.
Run Code Online (Sandbox Code Playgroud)

所以函数定义有语法错误,但我不明白错误是什么以及如何修复它。我将不胜感激任何帮助。谢谢!

小智 9

您的函数被声明为returns void因此您无法从中返回任何内容。如果要返回多个值,则需要使用returns setof integer

但它的问题还不止于此。

  1. 你应该给你的参数一个名称(不是错误,而是良好的编码风格)
  2. 要从需要使用的函数返回值return。要返回多个值(因为returns setof),您需要使用return next
  3. 不需要动态SQL返回值,直接返回变量即可。
  4. 也不是错误,而是:window是一个关键字,我不会使用具有该名称的变量。

应用所有这些,您的函数应该如下所示:

CREATE OR REPLACE FUNCTION test(p_input int[]) 
  RETURNS setof integer
as
$$  
DECLARE                                                     
  l_value INT;                                             
BEGIN                                                       
    FOREACH l_value IN ARRAY p_input                             
    LOOP                                                    
        return next l_value;
    END LOOP;                                               
end;
$$ 
LANGUAGE plpgsql;                                        
Run Code Online (Sandbox Code Playgroud)

我不确定您是否知道,但是已经有一个内置函数可以实现相同的功能:unnest().