PL/pgSQL的：我们可以将一个查询结果到一个变量

Question

我试图像PLPGSQL创建一个函数：

CREATE OR REPLACE FUNCTION select_left_photo_ids(in_photo_id bigint[], in_album_id bigint, in_limit int) RETURNS SETOF bigint[] AS 
$$ 
DECLARE photo_count int; 
DECLARE photo_ids bigint[]; 
    BEGIN 
     SELECT photo_id INTO STRICT photo_ids FROM tbl_album_photos WHERE album_id = in_album_id AND photo_id < in_photo_id ORDER BY photo_id DESC LIMIT in_limit; 
     GET DIAGNOSTICS photo_count = ROW_COUNT; 
     IF photo_count < in_limit THEN 
      SELECT photo_id INTO STRICT photo_ids FROM (SELECT photo_id FROM tbl_album_photos WHERE album_id = in_album_id ORDER BY photo_id LIMIT in_limit) AS dummy ORDER BY photo_id DESC; 
     END IF; 
     RETURN photo_ids; 
    END; 
$$ 
LANGUAGE plpgsql; 

的想法是取了比输入照片的身份更大的带照片的身份证。如果没有。结果中的照片数量不足限制，我会尝试获取底部的n条记录。

上述功能不起作用。有人请给出一些指针/提示或链接关于如何将选择查询的结果存储在变量中。我的意思是bigint[]。

Answer 1

您需要添加一个array_agg到

1. 路线从您的查询单值，使得SELECT ... INTO会工作。
2. 并获得bigint[]结果放入您的bigint[]目标。

然后你需要添加一个派生表来让你的LIMIT工作。

select array_agg(photo_id) into strict photo_ids 
from (
    select photo_id 
    from tbl_album_photos 
    where album_id = in_album_id 
     and photo_id < in_photo_id 
    order by photo_id desc 
    limit in_limit 
) dt; 

然后，你可以问阵列是不是看着ROW_COUNT有多大：

photo_count := array_length(photo_ids, 1); 

再下一个SELECT ... INTO将类似于上述新array_agg版本。