PostgreSQL中如何在循环中获取值列表以用于IN操作符？

你当前的写法存在问题：循环里每次赋值v_id = rec.id_field会覆盖之前的值，最后v_id只会保留最后一条记录的id，无法形成用于IN操作的列表。正确的做法是用数组类型来存储所有收集到的id，具体步骤如下：

• 声明对应类型的数组变量：根据id_field的实际类型（比如整数、文本）声明数组，例如如果是整数类型，就写v_id integer[];
• 循环中收集值：使用array_append()函数将每个id添加到数组中，避免覆盖
• 在IN操作中使用数组：可以用= ANY()语法直接匹配数组，或者用unnest()展开数组后用IN

修正后的完整代码示例：

DO $$
declare
rec record;
v_id integer[]; -- 根据id_field的实际类型调整，比如text[]
begin
  -- 循环收集id到数组
  for rec in (select id_field from my_tbl) loop
      v_id = array_append(v_id, rec.id_field);
  end loop; 
   
  -- 使用收集到的id数组查询
  for rec_new in (select new_id_field 
                  from new_my_tbl
                  where id_field = ANY(v_id) -- 直接用数组匹配，效果等同于IN
                  ) loop
      -- 这里写你的业务逻辑
      raise notice '匹配到的new_id: %', rec_new.new_id_field;
  end loop; 
end;
$$ LANGUAGE plpgsql;

补充说明：如果一定要用IN语法，也可以写成where id_field IN (SELECT unnest(v_id))，但= ANY()的性能更优，推荐使用。

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