18

我有以下形式的两个表（即，每个 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)
4

2 回答 2

30

最终版本

...在从 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()。

此外，请考虑用单个联合表替换和之间的1:1 关系。foobar

数据修改 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 就像任何其他的一样 - 可以包括您要插入的表。首先读取行，然后插入。

于 2012-05-06T16:23:05.250 回答
0

如果idofbar是串行的并且具有默认值nextval('bar_id_seq'::regclass)，您可以手动调用此函数以获取 cte 中的新 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
于 2017-11-08T13:55:15.797 回答