如何为FOR循环传递动态值并遍历表中所有行？

我可以执行以下代码：

for i in 1 .. (select pays from schema.table_name where pk_id='12' and pk2_id='123')

但这仅对特定行有效，请问如何对表中所有行执行此操作？

去除WHERE条件时会出现以下错误：

ERROR: more than one row returned by a subquery used as an expression

我需要用FOR循环遍历每一行的支付次数，例如当某行的pays为50时，对应主键组合需迭代50次，且需对表中所有主键组合重复此操作。

在PostgreSQL中，要实现对表中每一行按pays值循环指定次数的需求，你可以通过以下两种方式处理：

方法1：PL/pgSQL嵌套循环 #

用外层循环遍历表中所有行，内层循环根据当前行的pays值执行对应次数的逻辑，这是最直观的循环实现方式：

DO $$
DECLARE
    rec record;
    i integer;
BEGIN
    -- 外层循环遍历所有主键组合及对应的pays值
    FOR rec IN SELECT pk_id, pk2_id, pays FROM schema.table_name LOOP
        -- 内层循环按当前行的pays值执行次数
        FOR i IN 1 .. rec.pays LOOP
            -- 这里替换成你实际要执行的操作，比如插入、更新等
            -- 示例：打印当前循环信息
            RAISE NOTICE '主键组合: (%, %), 第 % 次循环', rec.pk_id, rec.pk2_id, i;
        END LOOP;
    END LOOP;
END $$;

方法2：纯SQL批量生成（推荐） #

如果你的操作可以用SQL批量完成，建议用generate_series函数生成对应次数的行，避免显式循环，效率更高：

-- 先生成所有需要的循环行
SELECT
    t.pk_id,
    t.pk2_id,
    gs.generate_series AS loop_num
FROM schema.table_name t
CROSS JOIN generate_series(1, t.pays) gs;

你可以基于这个结果直接做后续操作，比如插入到目标表：

INSERT INTO target_table (pk_id, pk2_id, loop_count)
SELECT
    t.pk_id,
    t.pk2_id,
    gs.generate_series
FROM schema.table_name t
CROSS JOIN generate_series(1, t.pays) gs;

错误原因说明 #

你之前去掉WHERE条件报错，是因为1 .. (select pays ...)这种写法要求子查询必须返回单个值，但去掉WHERE后子查询返回了表中所有行的pays值（多行），PostgreSQL无法将多行值作为范围上限，因此抛出错误。上面两种方法都是逐行处理每个pays值，自然规避了这个问题。

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