在 plpgsql 的 FOR 循环中使用自定义 return 类型

Question

下面的代码我用returns一个整数1:

CREATE TYPE my_test AS (
   foo Integer
  );

CREATE FUNCTION foo_out() 
  RETURNS SETOF Integer AS
$$ 
BEGIN
RETURN QUERY
SELECT 1 as foo;
END
$$
LANGUAGE plpgsql;

CREATE FUNCTION foo1()
  RETURNS SETOF my_test
AS $$
DECLARE
    x my_test;
BEGIN
    FOR x IN    
        SELECT foo_out()
    LOOP     
        RETURN NEXT x;
    END LOOP;
END;
$$
LANGUAGE 'plpgsql';

select * from foo1();

但是为什么同样的代码return:

ERROR:  invalid input syntax for integer: (1)

如果我将 return 类型更改为：

CREATE FUNCTION foo_out() 
  RETURNS SETOF my_test

哪个也应该是 integer!?可能是系统在整数类型和包含整数的自定义类型之间存在差异。应该改变什么才能在这里使用自定义类型？

Answer 1

my_test 是复合类型，行类型，包含单个整数字段。不是而是integer。

当分配给 FOR loop 中的记录或行类型时，查询形成行的所有输出列。您的 SELECT 需要 return 单个未嵌套的 integer 列，然后将其嵌套到与您的自定义类型匹配的行类型中 my_test.
您还可以在 FOR 循环中分配给标量变量列表，在这种情况下，查询中的列从左到右分配给变量原样 - 不形成一行.

如果列本身是行类型，则您有一层到多层的嵌套。包含整数字段的行的文本表示形式是“(1)”（带括号！），这就是您在错误消息中看到的内容。

您可以通过 extracting the integer field from the row with attribute notation:

解决这个问题

SELECT (foo_out()).*

或者（对于多列更有效）分解为：

SELECT * FROM foo_out()

示例代码

CREATE FUNCTION foo_out() 
  RETURNS SETOF <b>my_test</b>
  LANGUAGE sql AS
'<b>SELECT ''(1)''::my_test</b>';

CREATE FUNCTION foo1()
  RETURNS SETOF my_test
  LANGUAGE plpgsql AS
$func$
DECLARE
   x my_test;
BEGIN
   FOR x IN    
      <b>SELECT * FROM foo_out()</b>
   LOOP     
      RETURN NEXT x;
   END LOOP;
END
$func$;

db<>fiddle here
_旧sqlfiddle

不要引用语言名称 plpgsql。这是一个标识符。

请记住很少需要循环，因为使用基于集合的方法可以更有效地解决大多数问题（仅限 SQL）。

Craig 和 Pavel 的相关回答：

Passing array of a composite type to stored procedure

在 plpgsql 的 FOR 循环中使用自定义 return 类型

Use of custom return types in a FOR loop in plpgsql

postgresql

for-loop

types

row

plpgsql

示例代码