使用Postgres一次在3个表中插入数据

时间:2013-12-13 07:29:06

标签: sql postgresql common-table-expression sql-insert sql-returning

我想用一个查询将数据插入3个表中 我的表格如下所示:

CREATE TABLE sample (
   id        bigserial PRIMARY KEY,
   lastname  varchar(20),
   firstname varchar(20)
);

CREATE TABLE sample1(
   user_id    bigserial PRIMARY KEY,
   sample_id  bigint REFERENCES sample,
   adddetails varchar(20)
);

CREATE TABLE sample2(
   id      bigserial PRIMARY KEY,
   user_id bigint REFERENCES sample1,
   value   varchar(10)
);

我会得到一个密钥以换取每次插入,我需要在下一个表中插入该密钥 我的疑问是:

insert into sample(firstname,lastname) values('fai55','shaggk') RETURNING id;
insert into sample1(sample_id, adddetails) values($id,'ss') RETURNING user_id;
insert into sample2(user_id, value) values($id,'ss') RETURNING id;

但如果我运行单个查询,他们只会向我返回值,我不能立即在下一个查询中重复使用它们。

如何实现这一目标?

4 个答案:

答案 0 :(得分:93)

使用data-modifying CTEs

WITH ins1 AS (
   INSERT INTO sample(firstname, lastname)
   VALUES ('fai55', 'shaggk')
-- ON     CONFLICT DO NOTHING                -- optional addition in Postgres 9.5+
   RETURNING id AS user_id
   )
, ins2 AS (
   INSERT INTO sample1 (user_id, adddetails)
   SELECT user_id, 'ss' FROM ins1
   -- RETURNING user_id                      -- only if used in turn
   )
INSERT INTO sample2 (user_id, value)         -- same here
SELECT user_id, 'ss' FROM ins1;

每个插入都取决于之前的插入。 SELECT而不是VALUES确保如果从前一个插入没有返回任何行,则不会在子表中插入任何内容。 (相关:Postgres 9.5+中的ON CONFLICT条款)
它也有点短而快。


通常情况下, 在一个地方提供完整的数据行会更方便

WITH data(firstname, lastname, adddetails, value) AS (
   VALUES                                 -- provide data here
      (text 'fai55', text 'shaggk', text 'ss', text 'ss2')  -- see below
       --  more?                          -- works for multiple input rows
   )
, ins1 AS (
   INSERT INTO sample (firstname, lastname)
   SELECT firstname, lastname FROM data   -- DISTINCT? see below
   ON     CONFLICT DO NOTHING             -- required UNIQUE constraint
   RETURNING firstname, lastname, id AS sample_id
   )
, ins2 AS (
   INSERT INTO sample1 (sample_id, adddetails)
   SELECT sample_id, adddetails
   FROM   data
   JOIN   ins1 USING (firstname, lastname)
   RETURNING sample_id, user_id
   )
INSERT INTO sample2 (user_id, value)
SELECT user_id, value
FROM   data
JOIN   ins1 USING (firstname, lastname)
JOIN   ins2 USING (sample_id);

您可能需要在单独的VALUES表达式中显式类型转换(而不是附加到INSERT的VALUES表达式,其中数据类型是从目标表派生的。

如果多行可以带有相同的(firstname, lastname),则可能需要为第一个插入折叠重复:

...
INSERT INTO sample (firstname, lastname)
SELECT DISTINCT firstname, lastname FROM data
...

您可以使用(临时)表作为数据源,而不是CTE data

相关,更多细节:

答案 1 :(得分:13)

像这样的东西

with first_insert as (
   insert into sample(firstname,lastname) 
   values('fai55','shaggk') 
   RETURNING id
), 
second_insert as (
  insert into sample1( id ,adddetails) 
  values
  ( (select id from first_insert), 'ss')
  RETURNING user_id
)
insert into sample2 ( id ,adddetails) 
values 
( (select user_id from first_insert), 'ss');

由于不需要从插入到sample2生成的id,我从最后一个插入中删除了returning子句。

答案 2 :(得分:5)

通常,您使用事务来避免编写复杂的查询。

http://www.postgresql.org/docs/current/static/sql-begin.html

http://dev.mysql.com/doc/refman/5.7/en/commit.html

假设你的Postgres标签是正确的,你也可以使用CTE。例如:

with sample_ids as (
  insert into sample(firstname, lastname)
  values('fai55','shaggk')
  RETURNING id
), sample1_ids as (
  insert into sample1(id, adddetails)
  select id,'ss'
  from sample_ids
  RETURNING id, user_id
)
insert into sample2(id, user_id, value)
select id, user_id, 'val'
from sample1_ids
RETURNING id, user_id;

答案 3 :(得分:3)

您可以在Sample表上创建一个after insert插件,以插入另外两个表。

我看到这样做的唯一问题是你没有办法插入adddetails,它总是空的,或者在这种情况下是ss。没有办法将样品列插入样品表中不实际的样品中,因此您无法将其与天生插入物一起发送。

另一种选择是创建一个存储过程来运行插入。

你有问题taged mysql和postgressql我们在这里讨论哪个数据库?