Function PostgreSQL-返回到数组中

我想将更新的返回值存储到数据结构中，以便在后续查询中使用它

在这个例子中，我得到了一个“parent_id”列表，我想找到父元素在该数组中的所有子元素。然后，我希望更新它们的一些值，并做其他事情

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（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;

如果要收集数组中找到的所有
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
$$

以下是一个例子：

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;

这实际上取决于您想对收集的ID做什么。通常，使用数据修改CTE的单个查询是最佳解决方案。