Question

是否可以从postgres函数返回多个不同类型的结果集？

类似的东西：

CREATE OR REPLACE FUNCTION getUserById()
RETURNS setof ???
AS
$$
BEGIN

return query select id, name /* and other columns */ from users where id = 1;
return query select id, phone_number from user_phones where user_id = 1

END
$$
LANGUAGE plpgsql;

我不想使用联接，因为可以使用多个手机。另外，避免使用游标会很棒。它在MS SQL中是可能的，我想在postgres中做同样的事情。

Answer 1

在搜索之后，找不到任何更好的解决方案作为游标的使用。我假设您已经看过这个：https://blog.dsl-platform.com/multiple-result-sets-alternatives-in-postgres-3/

CREATE FUNCTION load_page(_session INT) RETURNS setof refcursor AS
$$
DECLARE c_top_items refcursor;
DECLARE c_shopping_cart refcursor;
BEGIN
    OPEN c_top_items FOR
        SELECT t.name, t.description
        FROM top_item t
        ORDER BY t.popularity DESC
        LIMIT 10;
    RETURN NEXT c_top_items;
    OPEN c_shopping_cart FOR
        SELECT c.product_id, c.product_name, c.quantity
        FROM shopping_cart c
        WHERE c.session_id = _session
        ORDER BY c.id;
    RETURN NEXT c_shopping_cart;
END;
$$ LANGUAGE plpgsql;

并致电：

BEGIN;
SELECT load_page(mySession);
FETCH ALL IN "<server cursor 1>";
FETCH ALL IN "<server cursor 2>";
COMMIT;

Answer 2

我不想使用联接，因为可以使用多个手机。

这不是在PostgreSQL中避免JOIN的理由。完全没有。

PostgreSQL允许您将电话号码聚合成一个数组：

CREATE OR REPLACE FUNCTION getUserById()
RETURNS TABLE (
    id INTEGER,
    name TEXT,
    /* and other columns */
    phone_numbers TEXT[]
)
AS
$$
    select
         users.id,
         users.name,
         /* and other columns */
         -- Remove NULL because you get an array containing just NULL
         -- if user_phones doesn't contain any matching rows.
         array_remove(array_agg(user_phones.phone_number), NULL) as phone_numbers
    from users
    left join user_phones on user_phones.user_id = users.id
    where users.id = 1
    -- Note that grouping by a table's primary key allows you to use
    -- any column from that table in the select in PostgreSQL
    group by users.id
    ;
$$
LANGUAGE SQL
STABLE
;

这更简单，更直观。

如果可以为没有电话号码的用户返回零行，则可以切换到内部联接。在这种情况下，您可以放弃array_remove来电。

我还在函数中添加了STABLE规范（因为它没有修改任何表数据）并将其切换为SQL而不是PGPLSQL（因为它＆＃39} ; s只是一个查询）。这将使PG更好地进行优化;特别是，在某些情况下，它可以内联查询并推送过滤器。实际上，你甚至可能不需要一个功能。

Answer 3

CREATE OR REPLACE FUNCTION public.TestReturnMultipleTales
( 
 param_coid integer, 
 ref1 refcursor,
 ref2 refcursor
)
RETURNS SETOF refcursor 
LANGUAGE 'plpgsql'
COST 100
VOLATILE PARALLEL UNSAFE
ROWS 1000

AS $BODY$
DECLARE
            
BEGIN
  OPEN ref1 FOR SELECT * FROM dbo.tbl1 WHERE coid = param_coid;
  RETURN NEXT ref1;

  OPEN ref2 FOR SELECT * FROM dbo.tbl2 LIMIT 5;
  RETURN NEXT ref2;
END;
$BODY$;

BEGIN;
    SELECT football_players.show_cities_multiple(123456, 'Ref1', 'Ref2');
    FETCH ALL IN "Ref1";
    FETCH ALL IN "Ref2";
COMMIT;

另外，如果想使用 C# 在 .NET 中实现它，请与我联系，我解决了这个问题。

Postgres函数：返回多个表

3 个答案: