如何在PostgreSQL中声明可存储整列数据的变量？

你遇到的问题本质是：单个numeric(1)类型的变量只能容纳单一数值，所以当查询返回多行结果时，它只会保留最后一行的值。要把整列的所有数据都存起来，必须更换变量的声明类型，下面给你两种最常用的实现方式：

方式一：使用数组类型存储所有值 #

这是最直接的方法，把变量声明为对应类型的数组，然后用聚合函数array_agg()把整列的值打包成数组存入变量。

示例代码：

DECLARE
    var_tmp numeric(1)[]; -- 声明为numeric(1)类型的数组
BEGIN
    -- 用array_agg把table_tmp.number的所有值聚合为数组，存入var_tmp
    SELECT array_agg(number) INTO var_tmp FROM table_tmp;

    -- 后续可以直接使用这个数组，比如访问第n个元素
    RAISE NOTICE '第一个值: %', var_tmp[1];
    RAISE NOTICE '所有值: %', var_tmp;
END;

如果列里有NULL值，你可以选择过滤掉它们：

SELECT array_agg(number) FILTER (WHERE number IS NOT NULL) INTO var_tmp FROM table_tmp;

方式二：使用游标逐行处理数据 #

如果你的数据量很大，或者需要逐行处理每一条记录，用游标会更节省内存，不需要一次性把所有数据加载到变量里。

示例代码：

DECLARE
    -- 声明一个游标，指向你要查询的结果集
    cur_number CURSOR FOR SELECT number FROM table_tmp;
    var_single numeric(1); -- 用来存储游标每次取出的单个值
BEGIN
    OPEN cur_number; -- 打开游标
    LOOP
        FETCH cur_number INTO var_single; -- 从游标里取出一行的值
        EXIT WHEN NOT FOUND; -- 当没有更多数据时退出循环

        -- 在这里添加你的处理逻辑，比如打印、计算等
        RAISE NOTICE '当前处理的值: %', var_single;
    END LOOP;
    CLOSE cur_number; -- 关闭游标
END;

两种方式的适用场景：

• 数组方式：适合需要一次性获取所有数据，后续进行批量操作的场景（比如统计、批量更新）。
• 游标方式：适合数据量较大，或者需要逐个处理每条记录的场景，避免占用过多内存。

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