使用 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

使用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 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 都依赖于前一个。 SELECT 而不是 VALUES 确保如果没有从先前的 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小提琴here

您可能需要在独立的VALUES 表达式中进行显式类型转换，而不是附加到INSERT 的VALUES 表达式，其中数据类型派生自目标表。见：

• Casting NULL type when updating multiple rows

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

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

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

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

相关：

• How to use RETURNING with ON CONFLICT in PostgreSQL?
• Is SELECT or INSERT in a function prone to race conditions?

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

假设您的 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;

Answer 4

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

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

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

你的问题是 mysql 和 postgressql 我们在这里讨论的是哪个数据库？