如何将表行传递给 plpgsql 函数?
Posted
技术标签:
【中文标题】如何将表行传递给 plpgsql 函数?【英文标题】:How to pass table rows to a plpgsql function? 【发布时间】:2015-08-13 13:21:46 【问题描述】:我正在尝试创建一个函数,该函数可以从表中一一获取行,并为每一行为不同的表生成 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));
有没有办法做到这一点?要将“数据”中的行一一传递给函数,并从函数中取出一个带有显式列的表?
【问题讨论】:
【参考方案1】:又挖了几个星期,我找到了答案:这是 LATERAL JOIN。 在我的示例中,我需要的查询是:
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,文档here,他们明确提到了这种用法:“一个常见的应用程序正在为一个集合返回函数提供一个参数值”。此外,查询的运行时间现在还可以,不需要 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
至于在 ().* 中包装函数调用时增加运行时间的原因,事实证明这是因为解析器中的宏扩展错误,而当您执行 LATERAL 连接时不会发生这种情况。有关详细信息,请参见此处:
How to avoid multiple function evals with the (func()).* syntax in an SQL query?
【讨论】:
【参考方案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 函数?的主要内容,如果未能解决你的问题,请参考以下文章
如何将参数从 bat 文件或命令行传递给 plpgsql 函数?
将数组从 node-postgres 传递给 plpgsql 函数