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

Posted 2023-02-15

【中文标题】使用 Postgres 一次在 3 个表中插入数据

【英文标题】：Insert data in 3 tables at a time using Postgres

【发布时间】：2013-12-31 21:53:23

【问题描述】：

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

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

如何做到这一点？

【参考方案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?

【讨论】：

感谢重播，如果发生任何插入失败，我可以添加事务推出吗？是的，我怎么能

这是一条 SQL 语句。可以将多个语句捆绑到一个事务中，但不能将其拆分。此外，丹尼斯在他的评论中所说的话。我在我的答案中附加了一些链接。

@mmcrae：是的，你可以。相关：dba.stackexchange.com/questions/151199/…

@No_name：当然，各种方式。我建议你问一个带有定义细节的问题。您始终可以在此处链接以获取上下文。或在此处发表评论链接以引起我的注意。

@AdamHughes：确实，sample_id 和 user_id 在多个地方混淆了。该示例具有相当误导性的列名。修复、澄清并添加了一个小提琴。

【参考方案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 子句。

【讨论】：

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

【参考方案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;

【讨论】：

thanx 如果任何插入失败，我将如何在此查询中实现事务，我可以回滚

然后你重新开始一切，当然在更正查询之后，因为整个事务（或 cte）将被回滚。顺便说一句，如果您的插入偶尔会失败，那么您可能做错了什么。唯一合理的插入失败的情况是在并发事务期间遇到重复唯一键的 upsert 场景中，即使这样，如果您需要防弹，您也可以获得咨询锁或表锁。跨度>

【参考方案4】：

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

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

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

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