将ResultSet传递给Postgresql函数

Question

是否可以将postgres查询的结果作为输入传递给另一个函数？

作为一个非常人为的例子,说我有一个像

SELECT id, name
FROM users
LIMIT 50

我想创建一个函数my_function,它接受第一个查询的结果集并返回最小的id.这可能在pl/pgsql中吗？

SELECT my_function(SELECT id, name FROM Users LIMIT 50); --returns 50

Answer 1

您可以使用游标，但这对于计算最小值来说非常不切实际。

为此，我将使用临时表，并传递表名以在动态 SQL中使用：

CREATE OR REPLACE FUNCTION f_min_id(_tbl regclass, OUT min_id int) AS 
$func$
BEGIN

EXECUTE 'SELECT min(id) FROM ' || _tbl
INTO min_id;

END  
$func$ LANGUAGE plpgsql;

称呼：

CREATE TEMP TABLE foo ON COMMIT DROP AS
SELECT id, name
FROM   users
LIMIT  50;

SELECT f_min_id('foo');

主要观点

• 第一个参数的类型是regclass为了防止 SQL 注入。更多信息请参见 dba.SE 上的相关答案。

• 我创建了临时表ON COMMIT DROP以将其生命周期限制为当前事务。可能是也可能不是您想要的。

• 您可以扩展此示例以获取更多参数。使用 搜索动态 SQL 的代码示例EXECUTE。

-> SQLfiddle 演示

Answer 2

我会从另一方面考虑这个问题，为结果集的每个记录调用一个聚合函数。它不那么灵活，但可以给你一个工作的提示。

作为遵循您的示例问题的示例：

CREATE OR REPLACE FUNCTION myMin ( int,int ) RETURNS int AS $$
  SELECT CASE WHEN $1 < $2 THEN $1 ELSE $2 END;
$$ LANGUAGE SQL STRICT IMMUTABLE;

CREATE AGGREGATE my_function ( int ) (
    SFUNC = myMin, STYPE = int, INITCOND = 2147483647 --maxint
);

SELECT my_function(id) from (SELECT * FROM Users LIMIT 50) x; 

Answer 3

我认为没有办法将记录集或表传递到函数中（但如果我错了，我会很高兴）。我最好的建议是传递数组：

create or replace function my_function(data int[])
returns int
as
$$
   select min(x) from unnest(data) as x
$$
language SQL;

sql 小提琴演示