从Postgres函数返回数据

时间:2013-08-23 14:12:44

标签: function postgresql user-defined-functions plpgsql postgresql-9.2

我是PostgreSQL的新手,我遇到了一些问题,让它做我想做的事。

我需要构建一个接受多个变量的函数,在内部执行多个查询,然后返回由多行和多列组成的数据集。我已经构建了几个测试函数来更好地掌握Postgres的功能,这里有一个:

 CREATE OR REPLACE FUNCTION sql_with_rows11(id integer) RETURNS character varying AS $BODY$
 declare vid integer;
 declare vendor character varying;
 BEGIN
 vid := (select v_id from public.gc_alerts where a_id = id);
 vendor := (select v_name from public.gc_vendors where v_id = vid);

   RETURN vendor;
   END;
 $BODY$
   LANGUAGE plpgsql VOLATILE
   COST 100;
 ALTER FUNCTION sql_with_rows11(integer)
   OWNER TO postgres;

我知道我可以将它组合成一个查询,但这更像是一个练习练习。这很好,我得到了供应商名称。但是,现在我需要从gc_vendors表中返回多个列。

最后,我需要构建一个函数,它将根据子查询从多个表中返回列。我已经研究过创建一个结果集函数,但我相信它一次只返回一行。我也研究了返回setof类型,但这似乎仅限于存在的表。

-------------- EDIT

嘿伙计们,谢谢你的帮助。我将功能更改为以下内容:

 CREATE OR REPLACE FUNCTION sql_with_rows14(IN v_uid character varying, IN lid           integer) 
 RETURNS table (aid int, aname character varying) AS $BODY$
 declare aid integer;
 declare aname character varying;
 BEGIN
 sql_with_rows14.aid := (select a_id from public.gc_alerts where v_id =           sql_with_rows14.v_uid);
 sql_with_rows14.aname := (select a_name from public.gc_alerts where a_id =      sql_with_rows14.aid);

   RETURN;
   END;
 $BODY$
   LANGUAGE plpgsql VOLATILE

我也尝试了RETURN NEXT,但结果相同。

当我查询它时,如果查询只返回一行,它可以正常工作。但是它不适用于多行。我也试过这样的事情,结果相同。你们能再次帮忙吗?

也试过这个:      开始      sql_with_rows14.aid:=(从public.gc_alerts中选择a_id);      sql_with_rows14.aname:=(从public.gc_alerts中选择a_name);

   RETURN NEXT;
   END;

2 个答案:

答案 0 :(得分:2)

  

我需要从gc_vendors表中返回多个列

要使用多列(而不是一组行)返回单行,您可以使用:

RETURNS row_type

..其中row_type是预定义的复合类型(如表名,自动起作用)。或者:

RETURNS record

结合OUT参数。请注意,OUT参数几乎在所有地方都可见,并避免使用naming conflicts 使用第二个选项,您的功能可能如下所示:

CREATE OR REPLACE FUNCTION sql_with_columns(
      IN  _id integer   -- the IN key word is just noise
    , OUT vid integer
    , OUT vendor text
    ) RETURNS record AS
$func$
BEGIN

SELECT INTO vid  v_id
FROM   public.gc_alerts
WHERE  a_id = id;

SELECT INTO vendor  v_name
FROM   public.gc_vendors
WHERE  v_id = vid;

RETURN;     -- just noise, since OUT parameters are returned automatically

END
$func$ LANGUAGE plpgsql

正如您所提到的,您应该将两个查询合并为一个,或者甚至使用简单的SQL语句。这只是一个展示案例。 The excellent manual has all the details.

可以也使用

RETURNS TABLE (...)

或者:

RETURNS SETOF row_type

这将允许您返回一组行(0,1或多个)。但那不是你的问题。

获取单个列而不是记录表示,请使用以下命令调用类似的函数:

SELECT * FROM sql_with_columns(...)

此处有很多示例,请尝试使用标记和一些关键字进行搜索。

使用plpgsql时,请阅读how to return from a function in the manual.

一章

答案 1 :(得分:1)

首先,考虑使用视图或简单查询。我会说,如果你可以用简单的查询来处理某些东西,你就不应该为此创建函数。在您的情况下,您可以使用此查询

select
    v.v_name, v.* -- or any other columns from gc_alerts or gc_vendors
from public.gc_alerts as a
    inner join public.gc_vendors as v on v.v_id = a.vid
where a.a_id = <your id here>

如果您希望函数返回行,可以将其声明为

CREATE OR REPLACE FUNCTION sql_with_rows11(id integer)
RETURNS table(vendor text, v_id int)
as
$$
    select
        v.v_name, v.v_id
    from public.gc_alerts as a
        inner join public.gc_vendors as v on v.v_id = a.vid
    where a.a_id = id
$$ language SQL;

或plpgsql函数:

CREATE OR REPLACE FUNCTION sql_with_rows11(id integer)
RETURNS table(vendor text, vid int)
AS
$$
    declare vid integer;
    declare vendor character varying;
BEGIN
    sql_with_rows11.vid := 1; -- prefix with function name because otherwise it would be declared variables
    sql_with_rows11.vendor := 4;
    return next; 

    sql_with_rows11.vid := 5;
    sql_with_rows11.vendor := 8;
    return next;
END;
$$ LANGUAGE plpgsql;

sql fiddle demo摆弄:)