PostgreSQL如何判断变量存储的记录数大于1并执行后续操作

你原有写法存在两个核心问题：

• 直接SELECT ... INTO 普通变量的写法仅能接收单行结果，集返回函数（即你调用的public.perform_some_action）返回多行时，该写法只会取第一行丢弃剩余行，无法拿到全量数据。
• 先count再查询的写法会触发两次集返回函数执行，对应两次重复API调用，不符合需求。

以下两种实现方案都只会触发一次原函数调用，可根据场景选择：

方案1：临时表存储结果（推荐，适配任意数据量） #

通过临时表一次性缓存函数返回的全量结果，后续计数、遍历都读取临时表，完全不会重复触发原函数执行：

-- 建临时表存储结果，事务结束自动删除无残留
CREATE TEMP TABLE tmp_action_results ON COMMIT DROP AS
SELECT * FROM public.perform_some_action(id)
WHERE user_id = driver_id_arg;

-- 判断记录数，读取临时表计数无额外调用
IF (SELECT COUNT(*) FROM tmp_action_results) > 1 THEN
    -- 遍历记录执行任务b，可直接将整行/指定字段传入其他PG函数
    FOR row_record IN SELECT * FROM tmp_action_results LOOP
        -- 替换为你的任务b逻辑，示例为调用其他PG函数
        PERFORM public.your_other_function(row_record.*, row_record.target_field);
    END LOOP;
END IF;

方案2：数组缓存结果（适合数据量较小的场景） #

如果返回结果量级不大，可以直接用数组存储全量结果，不需要建表：

DECLARE
    -- 声明和函数返回结构一致的数组、行变量
    result_arr public.perform_some_action[];
    row_record public.perform_some_action;
BEGIN
    -- 单次调用函数，将所有结果聚合为数组存入变量
    SELECT array_agg(res) INTO result_arr
    FROM public.perform_some_action(id) res
    WHERE user_id = driver_id_arg;

    -- 通过数组长度判断记录数
    IF cardinality(result_arr) > 1 THEN
        -- 遍历数组执行任务b
        FOREACH row_record IN ARRAY result_arr LOOP
            PERFORM public.your_other_function(row_record.*);
        END LOOP;
    END IF;
END;

注意事项 #

• 不要在SQL中多次直接引用public.perform_some_action，PostgreSQL不会对集返回函数的结果做默认缓存，每引用一次就会执行一次函数，对应重复的API调用。
• 临时表添加ON COMMIT DROP参数后，会在当前事务提交/回滚后自动清理，不会和其他会话的同名临时表冲突，也不会产生垃圾数据。

内容的提问来源于stack exchange，提问作者Alwaysblue