代码之家 › 专栏 › 技术社区 › Somil

如何从postgres函数返回新创建的ids数组?

pgadmin-4 postgresql

Somil · 技术社区 · 7 年前

我正在使用postgres函数在一个表中插入多行。我必须返回插入行的所有ID。

我的职能如下

    CREATE OR REPLACE FUNCTION insert_multiple_rows(temp integer)
        RETURNS integer[]
        LANGUAGE 'plpgsql'
        VOLATILE
        PARALLEL UNSAFE
        COST 100
    AS $BODY$DECLARE
    company_ids_list integer[];
    BEGIN
    INSERT INTO company VALUES
    (default,now(),temp),
    (default,now(),temp),
    (default,now(),temp),
    (default,now(),temp)
     RETURNING id INTO company_ids_list;

    RETURN company_ids_list;
    END;
    $BODY$;

当我试图用

 select insert_multiple_rows(58);

获取错误

ERROR:  query returned more than one row
CONTEXT:  PL/pgSQL function create_company_budget_allocation(integer) 
line 4 at SQL statement
SQL state: P0003

1 回复 | 直到 7 年前

Laurenz Albe 7 年前

SELECT .. INTO 或 INSERT ... RETURNING ... INTO 只能填充单个变量。如果查询返回多行,您将得到观察到的错误。

您可以使用 INSERT CTE和使用中的语句 array_agg 要构造结果数组,请执行以下操作:

WITH ins AS (
   INSERT INTO company VALUES
      (default,now(),temp),
      (default,now(),temp),
      (default,now(),temp),
      (default,now(),temp)
   RETURNING id)
SELECT array_agg(id) INTO company_ids_list
FROM ins;

推荐文章

Johnny T · 基于当前值的SQL合并表[重复]

6 月前

Kallydi · jdbc无法连接到docker中的postgresql数据库

6 月前

dietzi96 · Pandas DataFrame.to_sql随机和静默地失败,没有错误消息

6 月前

Peter Schofield · 类型转换Postgresql

6 月前

BarÄ±Å UÅaklÄ± · 如何在ON冲突更新中使用CTE中的值

7 月前

Kevin Smeeks · Pyspark JDBC分区读取

7 月前

Micah Armantrout · 如何使用sql脚本将数据插入到代码优先的数据库中?

7 月前

Vanortton · 使用Docker开始应用程序开发还是在开发过程中配置Docker?[关闭]

7 月前

Andrus · 如何在sql中查找第二个匹配项

7 月前

MWood · Postgres jsonpath用于引用单例元素或元素数组

7 月前