function – PostgreSQL – RETURNING INTO数组

我想将更新的RETURNING值存储到数据结构中,以便我可以在后续查询中使用它.

在这个例子中,我给出了一个“parent_ids”列表,我想找到父节点在该数组中的所有子节点.然后,我希望更新它们的一些价值,并做其他的事情.

CREATE OR REPLACE FUNCTION plpgsql_is_really_great(parent_ids bigint[])
  RETURNS void AS
$$
DECLARE
    found_ids bigint[];
BEGIN    
    UPDATE child SET
        foo = bar
    FROM 
        (SELECT id
            FROM child
            WHERE parent_id=ANY(parent_ids)
        ) as children_ids
    WHERE
        child.id = children_ids.id
    RETURNING children_ids.id INTO found_ids;  -- ???

    -- do more stuff with found_ids
$$LANGUAGE plpgsql

解决方法

有几种方法可以解决这个问题.假设你想为每个id调用一些f().

在PL / pgsql中：

$$
DECLARE found_id BIGINT;
BEGIN
  FOR found_id IN (UPDATE child SET foo=bar RETURNING id) LOOP
    PERFORM f(found_id);
  END LOOP;
END
$$

在纯sql中：

WITH updated(found_id) AS (
  UPDATE child SET foo=bar RETURNING id
)
SELECT f(found_id) FROM updated;

如果要收集数组中的所有found_ids,可以简单地：

$$
DECLARE array_var BIGINT[];
BEGIN
  WITH updated(found_id) AS (
    UPDATE child SET foo=bar RETURNING id
  )
  SELECT array_agg(found_id) FROM updated INTO array_var;
END
$$