主页 / dba / 问题 / 37433
Accepted
Chau
Asked: 2013-03-24 12:14:32 +0800 CST

如何访问 PL/pgSQL 中 RETURN 累积的行数

  • 772

表演时

RETURN QUERY ...

在 PL/pgSQL 函数中，之后是否可以直接访问累积到记录堆中的行数，这些记录在函数结束时返回？

例子：

RETURN QUERY SELECT * FROM tableA; -- 14 records
RETURN QUERY SELECT * FROM tableB; -- 8 records
RETURN QUERY SELECT * FROM tableC; -- 22 records

现在累积的记录数应该是 44。

postgresql functions

1 个回答

  1. Best Answer

    利用

    GET DIAGNOSTICS integer_var = ROW_COUNT;

    获取结果状态一章的手册中的更多内容。
    您的示例可能如下所示：

    CREATE OR REPLACE FUNCTION f_test()
  RETURNS SETOF table_a AS
$func$
DECLARE
    i  int;
    ct int := 0;
BEGIN

RETURN QUERY SELECT * FROM table_a;    -- 14 records
GET DIAGNOSTICS i = ROW_COUNT;  ct := ct + i;

RETURN QUERY SELECT * FROM table_b;    -- 8 records
GET DIAGNOSTICS i = ROW_COUNT;  ct := ct + i;

RETURN QUERY SELECT * FROM table_c;    -- 22 records
GET DIAGNOSTICS i = ROW_COUNT;  ct := ct + i;

RAISE NOTICE 'Returned % rows', ct;    -- 44

END
$func$ LANGUAGE plpgsql
    • 11

相关问题