Question

如何创建一个函数，该函数将参数integer []参数作为参数并使用IN子句在循环中使用此参数执行查询。在循环中我想执行下一个select和我要返回的查询结果。

类似的东西：

CREATE OR REPLACE FUNCTION function_which_i_want(my_argument integer[]) RETURNS my_schema_and_table[] AS
$BODY$
DECLARE
result my_schema_and_table[];
BEGIN
FOR l IN SELECT * FROM table2 WHERE id in (my_argument) LOOP
SELECT * FROM my_schema_and_table;
END LOOP;
END;
...

我希望得到每个select in循环的联合。一个巨大的联合结果。这可能吗？请帮忙。

Answer 1

PL / pgSQL函数

填写问题中的空白，可能如下所示：

CREATE OR REPLACE FUNCTION func1(_arr integer[])
  RETURNS SETOF target_table LANGUAGE plpgsql AS
$func$
DECLARE
    l record;
BEGIN

FOR l IN
    SELECT *
    FROM   lookup_table
    WHERE  some_id = ANY(_arr)
LOOP
    RETURN QUERY
    SELECT *
    FROM   target_table
    WHERE  link_id = l.link_id;
END LOOP;

END
$func$;

如果您想要返回SELECT * FROM my_schema_and_table;的结果，则必须以不同方式处理此整个函数。将其声明为RETURNS SETOF target_table
假设您确实需要SET中的target_table行，而不是数组？
将IN构造重写为= ANY(_arr)。这是直接使用数组参数的方法。在内部，PostgreSQL无论如何都会将IN表达式重写为= ANY()。使用EXPLAIN ANALYZE进行测试，亲眼看看或将构造与unnest()一起使用，并加入到结果表中，如@Clodoaldo demonstrates。长阵列的速度更快。

简化为纯SQL函数

以上仍然是毫无意义的设计。简化为执行相同操作的SQL函数：

CREATE OR REPLACE FUNCTION func2(_arr integer[])
  RETURNS SETOF target_table LANGUAGE sql AS
$func$
SELECT t.*
FROM   (SELECT unnest($1) AS some_id) x
JOIN   lookup_table l USING (some_id) 
JOIN   target_table t USING (link_id); -- assuming both tables have link_id
$func$

呼叫：

SELECT * FROM func2('{21,31}'::int[]);

Answer 2

CREATE OR REPLACE FUNCTION function_which_i_want(my_argument integer[])
RETURNS my_schema_and_table[] AS
$BODY$
DECLARE
result my_schema_and_table[];
BEGIN

for l in 
    select t.*
    from
        table2 t
        inner join
        unnest(my_argument) m(id) on m.id = t.id
loop
    SELECT * FROM my_schema_and_table;
END LOOP;
END;

Postgres函数与list参数和in子句

2 个答案:

PL / pgSQL函数

简化为纯SQL函数