在单个查询中将行插入多个表，从涉及的表中进行选择

Question

我有两个以下表格的表格（即每个foo都链接到一个条形图）。

CREATE TABLE foo (
    id INTEGER PRIMARY KEY,
    x INTEGER NOT NULL,
    y INTEGER NOT NULL,
    ...,
    bar_id INTEGER UNIQUE NOT NULL,
    FOREIGN key (bar_id) REFERENCES bar(id)
);

CREATE TABLE bar (
    id INTEGER PRIMARY KEY,
    z INTEGER NOT NULL,
    ...
);

使用嵌套查询复制满足特定条件的foo中的行很容易：

INSERT INTO foo (...) (SELECT ... FROM foo WHERE ...)

但我无法弄清楚如何使相关行的拷贝在bar中的每一行foo和插入的ID bar到新foo行。 有没有办法在单个查询中执行此操作？

期望结果的具体例子：

-- Before query:

foo(id=1,x=3,y=4,bar_id=100)  .....  bar(id=100,z=7)
foo(id=2,x=9,y=6,bar_id=101)  .....  bar(id=101,z=16)
foo(id=3,x=18,y=0,bar_id=102) .....  bar(id=102,z=21)


-- Query copies all pairs of foo/bar rows for which x>3:

-- Originals
foo(id=1,x=3,y=4,bar_id=101)  .....  bar(id=101,z=7)
foo(id=2,x=9,y=6,bar_id=102)  .....  bar(id=102,z=16)
foo(id=3,x=18,y=0,bar_id=103) .....  bar(id=103,z=21)

-- "Copies" of foo(id=2,...) and foo(id=3,...), with matching copies of
-- bar(id=102,...) and bar(id=103,...)
foo(id=4,x=9,y=6,bar_id=104)  .....  bar(id=104,z=16)
foo(id=5,x=18,y=0,bar_id=105) .....  bar(id=105,z=21)

Answer 1

最终版本

...来自OP的更多信息之后。 考虑这个演示：

-- DROP TABLE foo; DROP TABLE bar;

CREATE TEMP TABLE bar (
 id serial PRIMARY KEY  -- using a serial column!
,z  integer NOT NULL
);

CREATE TEMP TABLE foo (
 id     serial PRIMARY KEY  -- using a serial column!
,x      integer NOT NULL
,y      integer NOT NULL
,bar_id integer UNIQUE NOT NULL REFERENCES bar(id)
);

插入值 - 先bar 。
如果你在这样的问题中提供测试数据会非常有用 ！

INSERT INTO bar (id,z) VALUES
 (100, 7)
,(101,16)
,(102,21);

INSERT INTO foo (id, x, y, bar_id) VALUES
 (1, 3,4,100)
,(2, 9,6,101)
,(3,18,0,102);

将序列设置为当前值或我们得到重复的密钥违规：

SELECT setval('foo_id_seq', 3);
SELECT setval('bar_id_seq', 102);

检查：

-- SELECT nextval('foo_id_seq')
-- SELECT nextval('bar_id_seq')
-- SELECT * from bar;
-- SELECT * from foo;

查询：

WITH a AS (
    SELECT f.x, f.y, bar_id, b.z
    FROM   foo f
    JOIN   bar b ON b.id = f.bar_id
    WHERE  x > 3
    ),b AS (
    INSERT INTO bar (z)
    SELECT z
    FROM   a
    RETURNING z, id AS bar_id
    )
INSERT INTO foo (x, y, bar_id)
SELECT a.x, a.y, b.bar_id
FROM   a
JOIN   b USING (z);

这应该是您上次更新描述的内容。

该查询假定z是UNIQUE 。 如果z不是唯一的，它会变得更复杂。 在这种情况下，请参考此相关答案中的查询2，以获得使用窗口函数row_number()的现成解决方案。

另外，考虑用单个联合表替换foo和bar之间的1：1关系 。

---

数据修改CTE

更多信息后的第二个答案。

如果要在单个查询中向foo 和 bar添加行，则可以使用自PostgreSQL 9.1以来修改CTE的数据 ：

WITH x AS (
    INSERT INTO bar (col1, col2)
    SELECT f.col1, f.col2
    FROM   foo f
    WHERE  f.id BETWEEN 12 AND 23 -- some filter
    RETURNING col1, col2, bar_id  -- assuming bar_id is a serial column
    )
INSERT INTO foo (col1, col2, bar_id)
SELECT col1, col2, bar_id
FROM   x;

我从foo绘制值，将它们插入bar ，让它们与自动生成的bar_id一起返回并将其插入foo 。 您也可以使用任何其他数据。

这是一个在sqlfiddle上玩的工作演示 。

---

基本

在澄清之前提供基本信息的原始答案。
基本形式是：

INSERT INTO foo (...)
SELECT ... FROM foo WHERE ...

不需要括号。 你可以对任何表做同样的事情

INSERT INTO foo (...)
SELECT ... FROM bar WHERE ...

您可以加入到SELECT中插入的表：

INSERT INTO foo (...)
SELECT f.col1, f.col2, .. , b.bar_id
FROM   foo f
JOIN   bar b USING (foo_id);  -- present in foo and bar

它只是一个SELECT，就像其他任何一样 - 可以包括你插入的表。 首先读取行，然后插入。

Answer 2

如果bar id是serial并且具有默认值nextval('bar_id_seq'::regclass)您可以手动调用此函数以获取nextval('bar_id_seq'::regclass)新ID

with
s_bar as (
  SELECT id, z, nextval('bar_id_seq'::regclass) new_id
  FROM   bar
  WHERE  ...
),
s_foo as (
  SELECT x, y, bar_id
  FROM   foo
  WHERE  ...
),
i_bar as (
  INSERT INTO bar (id, z)
  SELECT new_id, z
  FROM   s_bar
),
i_foo as (
  INSERT INTO foo (x, y, bar_id)
  SELECT f.x, f.y, b.new_id
  FROM   s_foo f
  JOIN   s_bar b on b.id = f.bar_id
)
SELECT 1