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

都取决于前一个。

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<>小提琴这里

您可能需要在独立的

VALUES

表达式中进行显式类型转换 - 而不是附加到

VALUES

的

INSERT

表达式，其中数据类型源自目标表。参见：

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

如果多行可以具有相同的

(firstname, lastname)

，您可能需要折叠第一行

INSERT

的重复项：

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

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

data

。

将其与表中

UNIQUE

的

(firstname, lastname)

约束以及查询中的

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

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

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

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

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