PostgreSQL函数中如何处理其他函数返回的表结果并遍历

下面提供几种实用的方式，帮你在函数中接收、遍历返回表类型的函数结果，并基于数据调用其他函数：

方法1：直接用FOR循环逐行遍历 #

这是最常用的场景，无需提前存储结果，直接迭代返回表的每一行数据：

-- 先假设一个返回表类型的示例函数
CREATE OR REPLACE FUNCTION get_user_data()
RETURNS TABLE(user_id int, username text) AS $$
BEGIN
    RETURN QUERY SELECT id, name FROM users WHERE active = true;
END;
$$ LANGUAGE plpgsql;

-- 你的主处理函数
CREATE OR REPLACE FUNCTION process_user_data()
RETURNS void AS $$
DECLARE
    rec record; -- 定义记录变量存储每一行数据
BEGIN
    -- 直接遍历返回表的函数结果
    FOR rec IN SELECT * FROM get_user_data() LOOP
        -- 根据当前行数据调用其他函数
        PERFORM update_user_stats(rec.user_id);
        PERFORM log_user_activity(rec.user_id, rec.username);
    END LOOP;
END;
$$ LANGUAGE plpgsql;

你可以通过rec.列名直接访问返回表的字段，无需额外赋值操作，循环会自动逐行处理。

方法2：将结果存入临时表复用 #

如果需要多次使用返回的表数据，可先将结果存入临时表：

CREATE OR REPLACE FUNCTION process_user_data()
RETURNS void AS $$
DECLARE
    rec record;
BEGIN
    -- 创建临时表存储返回结果（会话结束后自动销毁）
    CREATE TEMP TABLE IF NOT EXISTS temp_users AS
    SELECT * FROM get_user_data();

    -- 第一次遍历处理
    FOR rec IN SELECT * FROM temp_users LOOP
        PERFORM update_user_stats(rec.user_id);
    END LOOP;

    -- 第二次使用临时表数据
    PERFORM bulk_log_activity((SELECT array_agg(user_id) FROM temp_users));

    -- 手动清理临时表（可选）
    DROP TABLE IF EXISTS temp_users;
END;
$$ LANGUAGE plpgsql;

临时表仅在当前会话有效，适合需要多次复用结果的场景。

方法3：将结果存入数组变量 #

如果数据量不大，也可以把返回表存入数组类型的变量中：

CREATE OR REPLACE FUNCTION process_user_data()
RETURNS void AS $$
DECLARE
    user_array get_user_data%ROWTYPE[]; -- 匹配返回表的行类型数组
    i int;
BEGIN
    -- 将返回表聚合为数组
    SELECT array_agg(t) INTO user_array FROM get_user_data() t;

    -- 遍历数组处理数据
    FOR i IN 1..array_length(user_array, 1) LOOP
        PERFORM update_user_stats(user_array[i].user_id);
        PERFORM log_user_activity(user_array[i].user_id, user_array[i].username);
    END LOOP;
END;
$$ LANGUAGE plpgsql;

get_user_data%ROWTYPE会自动匹配目标函数的返回行结构，无需手动定义字段。

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