INSERT INTO ... FROM SELECT ... RETURNING id mappings

Question

我正在使用PostgreSQL 9.3.

我想复制一些db记录.由于我正在为表使用自动增量pk id,因此我想从生成的重复记录ID返回到原始ID的id映射.例如,假设我有一个posts包含2条记录的表:

 [{'id': 1, 'title': 'first'}
, {'id': 2. 'title': 'second'}]

使用SQL:

INSERT INTO posts (title) SELECT title FROM posts RETURNING id, ??

我希望看到像这样的映射:

 [{'id': 3, 'from_id': 1}
, {'id': 4, 'from_id': 2}]

有关如何填写上述问号以使其有效的任何想法？非常感谢!

Answer 1

这对于子句UPDATE是可见的更加简单,其中加入更新的其他行是可见的RETURNING:

• 仅使用SQL返回pre-UPDATE列值 - PostgreSQL版本

目前也不可能这样做INSERT.每个文件:

表达式可以使用table_name指定的表的任何列名

table_name是INSERT命令的目标.

您可以使用(数据修改)CTE来实现此功能.
假设每个查询title都是唯一的,否则你需要做更多:

WITH sel AS (
   SELECT id, title
   FROM   posts
   WHERE  id IN (1,2)   -- select rows to copy
   )
,    ins AS (
   INSERT INTO posts (title)
   SELECT title FROM sel
   RETURNING id, title
 )
SELECT ins.id, sel.id AS from_id
FROM   ins
JOIN   sel USING (title);

如果title每个查询不唯一(但id每个表至少是唯一的):

WITH sel AS (
   SELECT id, title, row_number() OVER (ORDER BY id) AS rn
   FROM   posts
   WHERE  id IN (1,2)   -- select rows to copy
   ORDER  BY id
   )
,    ins AS (
   INSERT INTO posts (title)
   SELECT title FROM sel ORDER  BY id  -- ORDER redundant to be sure
   RETURNING id
 )
SELECT i.id, s.id AS from_id
FROM  (SELECT id, row_number() OVER (ORDER BY id) AS rn FROM ins) i
JOIN   sel s USING (rn);

第二个查询依赖于未提供的实现细节,即按所提供的顺序插入行.它适用于所有当前版本的Postgres,可能不会破坏.

SQL小提琴.

Answer 2

如果id列posts是serial类型，它的产生一样nextval('posts_id_seq'::regclass)，你可以手动调用这个函数为每一个新行

with
sel as (
  SELECT id, title, nextval('posts_id_seq'::regclass) new_id
  FROM   posts
  WHERE  id IN (1,2)
),
ins as (
  INSERT INTO posts (id, title)
  SELECT new_id, title
  FROM sel
)
SELECT id, new_id
FROM sel

它适用于任何数据，包括非唯一的 title