PostgreSQL - RETURNING INTO array

Question

我希望将一次更新的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

- JS_Riddler

---

这真的取决于您想用收集到的ID做什么。通常，使用带有数据修改CTE的单个查询是最佳解决方案。 - Erwin Brandstetter

Answer 1

有几种方法可以解决这个问题。

比如你想为每个UPDATE影响的id调用一些f(id)。

在 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_id收集到一个数组中，你可以这样做：

$$
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
$$

- Nick Barnes

Answer 2

这里有一个例子：

CREATE OR REPLACE FUNCTION exemplary ( parent_ids bigint[] )
RETURNS VOID AS $$
DECLARE
    _found_ids bigint[];
BEGIN

    WITH matching_children AS (
        UPDATE child
        SET foo = 1
        WHERE parent_id = ANY ( parent_ids )
        RETURNING id
    )
    SELECT array_agg ( id )
    FROM matching_children
    INTO _found_ids;

    RAISE NOTICE '%', _found_ids;

    RETURN;

END $$ LANGUAGE plpgsql;

- Tomasz Siorek