Question

我们如何传递（无限量）行数组（即常量表）作为PostgreSQL函数的参数/参数？

这是一个想法：

CREATE TYPE foo AS (
    x bigint,
    y smallint,
    z varchar(64)
);

CREATE OR REPLACE FUNCTION bar(bigint, foo[]) RETURNS TABLE(a bigint, x bigint, y smallint, z varchar(64)) AS
$$
    SELECT $1, x, y, z FROM unnest($2);
$$
LANGUAGE SQL;

以下函数调用有效，但有没有办法缩短它？

SELECT * FROM bar(1, ARRAY[(1,2,'body1'),(2,1,'body2')]::foo[]);

例如，我们无法删除::foo[]广告，但有没有办法重写内容以便我们可以省略它？

我们应该使用变量参数吗？

Answer 1

我的谷歌搜索一直引导着我，所以我要发布一个可能与OP的需求不完全匹配的答案，但对于那些看到标题的人来说可能会有所帮助如何传递多个行到PostgreSQL函数？

OP的原始请求是针对类型：

CREATE TYPE foo AS (
    x bigint,
    y smallint,
    z varchar(64)
);

如果您像我一样，您可能希望将标准SELECT查询的结果传递给函数。所以想象一下我创建了一个表（而不是一个类型）：

CREATE TABLE foo AS (
    x bigint,
    y smallint,
    z varchar(64)
);

我想将结果传递给函数：

SELECT * from foo WHERE x = 12345;

结果可能是零行或多行。

根据https://www.postgresql.org/docs/9.5/static/rowtypes.html的postgres文档，创建表也会导致创建具有相同名称的复合类型。这很有用，因为它会自动处理原始问题中的CREATE TYPE foo，我现在可以将其作为数组传递给函数。

现在我可以创建一个接受 foo 类型值数组的函数（简化为专注于传入的内容，以及如何使用记录，而不是返回的内容）：

CREATE OR REPLACE FUNCTION bar(someint bigint, foos foo[]) RETURNS ...
LANGUAGE plpgsql
AS $$
DECLARE
    foo_record record;
begin

-- We are going to loop through each composite type value in the array
-- The elements of the composite value are referenced just like 
-- the columns in the original table row
FOREACH foo_record IN ARRAY foos LOOP
  -- do something, maybe like:
  INSERT INTO new_foo (
    x, y, z
  )
  VALUES (
    foo_record.x,
    foo_record.y,
    foo_record.z
  );

END LOOP;

RETURN...
END;
$$;

然后可以通过以下方式简单地调用此函数bar(bigint, foo[])：

SELECT bar(4126521, ARRAY(SELECT * from foo WHERE x = 12345));

将 foo 表中查询的所有行作为foo类型数组传递。我们看到的函数然后对每个行执行一些操作。

虽然这个例子是设计的，也许并不完全是OP所要求的，但它符合问题的标题，并且可能使其他人不必再搜索更多内容以找到他们需要的内容。

编辑命名函数参数使事情变得更容易

Answer 2

PostgreSQL没有表值变量（还没有），所以没什么好看的。传递数组效率很低，但适用于合理大小的输入。

对于更大的输入，通常工作的是传递refcursor。它很笨拙，但对于较大的数据集很有用，有时可以与临时表结合使用。

e.g。

CREATE OR REPLACE FUNCTION bar(i bigint, c refcursor) RETURNS TABLE(a bigint, x bigint, y smallint, z varchar(64)) AS
$$
DECLARE
    cursrow foo;
BEGIN
    LOOP
        FETCH NEXT FROM c INTO cursrow;
        a := i;
        x := cursrow.x;
        y := cursrow.y;
        z := cursrow.z;
        RETURN NEXT;
        IF NOT FOUND THEN
            EXIT;
        END IF;
    END LOOP;
    RETURN;
END;
$$;

用法：

demo=> BEGIN;
BEGIN
demo=> DECLARE "curs1" CURSOR FOR VALUES (1,2,'body1'), (2,1,'body2');
DECLARE CURSOR
craig=> SELECT bar(1, 'curs1');
      bar      
---------------
 (1,1,2,body1)
 (1,2,1,body2)
 (1,,,)
(3 rows)

demo=> COMMIT;
COMMIT

不漂亮。但是，plpgsql永远不会。遗憾的是它没有行值左值，因为能够写出像(x, y, z) := cursrow或ROW(x, y, z) := cursrow这样的东西会让它变得不那么难看。

RETURN NEXT有效，但仅当您返回record未命名的参数或TABLE时才会生效。

遗憾的是，你不能在子表达式中使用SQL（不是plpgsql）FETCH ALL，所以你不能写

RETURN QUERY NEXT i, cursrow.* FROM (FETCH ALL FROM c) AS cursrow;

如何将多行传递给PostgreSQL函数？

2 个答案: