Question

我正在尝试创建一个可以从表中逐行获取行的函数，并且对于每个行，为不同的表生成1行或更多行。例如，让我们采取这个玩具功能（注意：在这个例子中，输入和输出具有相同的字段，但在我原来的问题中，字段是不同的）：

CREATE OR REPLACE FUNCTION toy_function( a integer, b integer )
RETURNS TABLE( x integer, y integer ) AS $$
BEGIN
    FOR i IN 1..2 LOOP
        x := a + b + i;
        y := a * b * i;
        RETURN NEXT;
    END LOOP;
END;
$$ LANGUAGE plpgsql;

给出了预期的输出：

SELECT * FROM toy_function( 10, 20 );

 x  |  y  
----+-----
 31 | 200
 32 | 400
(2 rows)

但是如果我将表中的行传递给它，就像这样：

WITH data AS (
    SELECT 1 * i AS a, 2 * i AS b
    FROM GENERATE_SERIES( 1, 3, 1 ) as i
)
SELECT
    toy_function( a, b )
FROM
    data;

我得到了一个记录列表，而不是像以前那样的列：

 toy_function 
--------------
 (4,2)
 (5,4)
 (7,8)
 (8,16)
 (10,18)
 (11,36)
(6 rows)

在().*中包装函数调用会返回单独的列，但会大大减慢查询速度（在我原来的问题中，它从2秒到6秒！）。

我也试过在子查询中传递输入数据，但它失败并出现我不太明白的错误：

WITH data AS (
    SELECT 1 * i AS a, 2 * i AS b
    FROM GENERATE_SERIES( 1, 3, 1 ) as i
)
SELECT
    *
FROM
    toy_function(( SELECT * FROM data));

ERROR:  subquery must return only one column
LINE 8: toy_function(( SELECT * FROM data));

有办法做到这一点吗？要将“数据”中的行逐个传递给函数，并使用显式列从函数中获取表格吗？

Answer 1

经过几个星期的挖掘后，我找到了答案：它是横向加入。在我的示例中，我需要的查询是：

WITH data AS (
    SELECT 1 * i AS a, 2 * i AS b
    FROM GENERATE_SERIES( 1, 3, 1 ) as i
)
SELECT
    f.*
FROM
    data, LATERAL toy_function( a, b ) f;

给出了我正在寻找的结果：

 x  | y
----+----
  4 |  2
  5 |  4
  7 |  8
  8 | 16
 10 | 18
 11 | 36
(6 rows)

（注意：LATERAL关键字对于函数是可选的。）

这个新的联接被添加到postgresql 9.3，docs here，其中明确提到了这种用法：＆＃34;一个常见的应用程序为set-returns函数提供参数值＆＃34;。此外，查询的运行时现在还可以，它不会花费3倍。

相关帖子（供参考）：

How can you expand a "condensed" PostgreSQL row into separate columns?

What is the difference between LATERAL and a subquery in PostgreSQL?

Call a set-returning function with an array argument multiple times

至于在（）。*中包装函数调用时运行时间增加的原因，由于解析器中的宏扩展不好而导致运行时间增加，这在你做的时候不会发生一个横向加入。有关详细信息，请参见此处：

How to avoid multiple function evals with the (func()).* syntax in an SQL query?

Answer 2

以下是如何使用表记录作为参数的示例：

CREATE FUNCTION  your_function_name( paramater_name table_name ) RETURNS INTEGER
AS
$BODY$
   RAISE NOTICE 'You can use value from your table : field_x = % ', paramater_name.field_x;
   RETURN 0;
$BODY$
LANGUAGE 'plpgsql';

如何将表行传递给plpgsql函数？

2 个答案: