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

Question

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

但是，如果我运行单个查询，它们只会将值返回给我，而我无法立即在下一个查询中重用它们。

如何实现这一目标？

Answer 1

使用数据修改 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依赖于前一个。 如果没有从先前的INSERT返回行，则SELECT而不是VALUES可确保在子表中没有插入任何内容。 （自 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表达式中进行显式类型转换 - 而不是附加到INSERT的VALUES表达式，其中数据类型是从目标表派生的。 见：

• 更新多行时强制转换 NULL 类型

如果多行可以带有相同的(firstname, lastname) ，您可能需要为第一个INSERT折叠重复项：

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

您可以使用（临时）表作为数据源而不是 CTE data 。

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

相关：

• 如何在 PostgreSQL 中使用 RETURNING 和 ON CONFLICT？
• 函数中的 SELECT 或 INSERT 是否容易出现竞争条件？

Answer 2

像这样的东西

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子句。

Answer 3

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

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;

Answer 4

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

我在这样做时看到的唯一问题是，您无法插入 adddetails，它将始终为空或在这种情况下为 ss。 无法将一列插入到样本表中实际上并不存在的样本中，因此您不能将其与初始插入一起发送。

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

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