Postgresql返回setof记录

时间:2012-06-04 07:44:45

标签: postgresql select boolean row plpgsql

我有自定义类型:

create type some_type as (
    some_bool_param     boolean, 
    str                 varchar
);

我创建一个包含此类字段的表并插入一些数据:

create table test_table (
    strs some_type
);

insert into test_table(strs) values
  ((false, 'First str'))
, ((false, 'Second str '))
, ((false, 'Third str'))
, ((false, 'Yet another str'));

现在我尝试返回setof some_type数据:

create or replace function get_str() returns setof some_type as
$$
declare
    r some_type;
begin
    for r in 
        select * from test_table loop
    return next r;
    end loop;
    return;
end;

我致电get_str()

select * from get_str();

但是得到一个错误:

ERROR:  error in boolean type value: "(f,"First str")"
CONTEXT:  PL/pgSQL function "get_str" line 4 at FOR by result of SELECT

我该如何解决?

3 个答案:

答案 0 :(得分:4)

create or replace function get_str() returns setof some_type as
$$
declare
    r some_type;
begin
    for r in 
        select strs from test_table loop
    return next r;
    end loop;
    return;
end;

以防万一:声明一个表也声明了它的行类型,因此这里不需要单独的CREATE TYPE。这也可行:

create table test_table (
    some_bool_param     boolean, 
    str                 varchar
);

insert into test_table values (false, 'First str');
insert into test_table values (false, 'Second str ');
insert into test_table values (false, 'Third str');
insert into test_table values (false, 'Yet another str');

create or replace function get_str()
returns setof test_table as
$$
    SELECT  *
    FROM    test_table;
$$
LANGUAGE sql;

答案 1 :(得分:3)

对于这种功能,你不再需要使用当前版本的PostgreSQL循环。请改用RETURN QUERY

CREATE OR REPLACE FUNCTION get_str()
  RETURNS SETOF test_table AS
$func$
BEGIN
   RETURN QUERY
   TABLE test_table;  -- shorthand for: SELECT * FROM test_table
END
$func$  LANGUAGE plpgsql;

或者提供像@Quassnoi这样的SQL函数。 请注意,这会返回嵌套的复合类型(SETOF test_table,而不是SETOF some_type)。

奇怪,这也有效:

CREATE OR REPLACE FUNCTION get_str5()
  RETURNS SETOF some_type AS
$func$
   TABLE test_table;
$func$ LANGUAGE sql;

外部行包装器被静默删除,至少可以说是Postgres有点古怪。

关于错误

你的功能会像这样:

CREATE OR REPLACE FUNCTION get_str()
  RETURNS TABLE (strs some_type) AS
$func$
BEGIN
   FOR strs IN
       SELECT (t.strs).* FROM test_table t
   LOOP
      RETURN NEXT r;
   END LOOP;
END
$func$ LANGUAGE plpgsql;

请注意粗体部分。 FOR循环中对行或记录类型的分配是通过子字段子字段完成的。这通常很方便,但在使用嵌套复合类型时会感到困惑。

您在表格行(some_type) - 两个嵌套级别中有一个复合类型(test_table)。这通常不是人们想要的,但这就是你向我们展示的东西。

您需要分解/解开两次

  • 您的原始代码仅使用select * from test_table分解外部行包装。

  • @ Quassnoi的建议修复select strs from test_table以同样的方式失败:strs通过直接引用从表行中分解,但strs仍然是需要的行类型在分配之前自行分解。

答案 2 :(得分:2)

根据它的声明,函数get_str()应该返回一组带有some_type结构的记录,该结构是2列,1个布尔值和1个varchar。

Wile函数实际上返回一组记录,其中包含1列,类型为some_type

尝试将return next r;替换为return next r.strs;