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


81

我想通过一个查询将数据插入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;

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

如何实现呢?

Answers:


131

使用修改数据的CTE

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

每个都INSERT取决于之前的一个。如果没有从前一个表返回任何行SELECTVALUES则要确保没有在子表中插入任何内容INSERT。(自Postgres 9.5+起,您可能会添加一个ON CONFLICT。)
这种方式也更短,更快。

通常,在一处提供完整的数据行更为方便:

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

db <>在这里拨弄

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

如果多个行可以具有相同的行(firstname, lastname),则可能需要折叠第一行的重复项INSERT

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

您可以使用(临时)表代替CTE作为数据源data

将其与表中的UNIQUE约束(firstname, lastname)ON CONFLICT查询中的子句结合起来可能很有意义。

有关:


1
重播的thanx如果发生任何失败插入,是否可以添加事务推出。是的,我如何
Faisal

3
这是一条SQL语句。一个人可以将几条语句捆绑到一个事务中,但不能将其分开。另外,丹尼斯在评论中说了什么。我在答案中附加了一些链接。
Erwin Brandstetter 2013年

2
@mmcrae:是的,可以。相关:dba.stackexchange.com/questions/151199/...
欧文Brandstetter修改

1
@No_name:当然,有多种方式。我建议您提出一个定义细节的问题。您可以随时在此处链接以获取上下文。或在此处发表评论并链接回去以引起我的注意。
Erwin Brandstetter

1
这是错字吗?在您的回答中INSERT INTO sample1 (user_id, adddetails),不是(sample_id, addetails)吗?
亚当·休斯

17

像这样

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不需要从insert into生成的id ,因此我returning从最后一个insert中删除了该子句。


我喜欢这种选择内部值的方法。它更一致,也可以在with语句中删除返回别名
mattdlockyer

6

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

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

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

您也可以使用CTE,前提是您的Postgres标签正确。例如:

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;

1
thanx如果有任何插入失败,我将如何在该查询中实现交易,我可以回滚
Faisal 2013年

1
当然,在更正了查询之后,您将重新开始所有事情,因为整个事务(或CTE)将被回滚。顺便说一句,如果您的插入偶尔会失败,则可能是您做错了。插入失败的唯一合理案例是在向上插入的情况下,该情况在并发事务期间遇到重复的唯一键,即使这样,如果需要使事情变得安全,您也可以得到咨询锁或表锁。
Denis de Bernardy 2013年

3

您可以在Sample表上创建一个after after触发器,以插入到其他两个表中。

我这样做的唯一问题是您将无法插入adddetails,该方法将始终为空或在这种情况下为ss。无法将一列插入样品表中并非实际的样品中,因此您无法将其与初始插入物一起发送。

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

您有标记为mysql和postgressql的问题,我们在这里谈论哪个数据库?

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.