将 SELECT 结果作为参数传递给 postgreSQL 函数

发布于07月22日

我有一个带有以下字段的表"UserState":

SELECT userid FROM "UserState" WHERE ctime>'2014-07-14'::timestamp

我有一个plpgsql函数，它必须把这个查询的结果作为参数:

get_timeinstate(SELECT userid FROM "UserState" WHERE ctime>'2014-07-14'::timestamp);

如何正确地创建函数以将查询结果作为参数传递到那里？

$func$
BEGIN
 RETURN QUERY
 SELECT 
...myanotherquery...
 WHERE "UserState".userid IN (HERE I NEED TO INSERT MY QUERY RESULT)
END;
$func$

推荐答案

将返回的用户id设置为数组传递.创建接受整数数组的函数

create function get_timeinstate (
    user_id_set integer[],
    another_param...

然后通过array_agg生成的数组调用它

get_timeinstate(
    (
        select array_agg(userid)
        from "UserState"
        where ctime>'2014-07-14'::timestamp
    ),
    another_param
);

在函数内部:

where "UserState".userid = any (user_id_set)

顺便说一句，如果您使用的是plpgsql，您可以将查询放在函数中，只传递日期:

create function get_timeinstate (
    p_ctime timestamp,
    another_param...
$func$
declare
    user_id_set integer[] := (
        select array_agg(userid)
        from "UserState"
        where ctime > p_ctime
    );
begin
    return query
    select 
    ...myanotherquery...
    where "UserState".userid = any (user_id_set)
end;
$func$