sql - 使用 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)
);

每次插入我都会得到一个 key 作为返回，我需要将该 key 插入下一个表中。
我的查询是:

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<> fiddle 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) 的 UNIQUE 约束和 ON CONFLICT 子句相结合可能是有意义的在查询中。

这不解决并发写入可能带来的并发症。为此，请参阅:

• How to use RETURNING with ON CONFLICT in PostgreSQL?

• Is SELECT or INSERT in a function prone to race conditions?