INSERT INTO ... FROM SELECT ... 返回 id 映射

Question

我使用的是 PostgreSQL 9.3。

我想复制一些数据库记录。由于我对表使用了自动增量 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 子句可见：

• Return pre-UPDATE column values using SQL only

INSERT 目前不可能做到这一点。 The manual:

表达式可以使用由 table_name

命名的表的任何列名

table_name 是INSERT 命令的目标。

您可以使用(data-modifying) CTEs 来实现此功能。
假设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，并且可能不会中断。

db小提琴here
_旧sqlfiddle

Answer 2

如果posts 的id 列是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

Answer 3

恕我直言，最简单的解决方案是简单地将一列添加到您的表中，您可以在其中放置被克隆行的 id。