[英]Insert data in 3 tables at a time using Postgres
我想通過一個查詢將數據插入到 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;
但是,如果我運行單個查詢,它們只會將值返回給我,而我無法立即在下一個查詢中重用它們。
如何實現這一目標?
使用數據修改 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
表達式,其中數據類型是從目標表派生的。 見:
如果多行可以帶有相同的(firstname, lastname)
,您可能需要為第一個INSERT
折疊重復項:
...
INSERT INTO sample (firstname, lastname)
SELECT DISTINCT firstname, lastname FROM data
...
您可以使用(臨時)表作為數據源而不是 CTE data
。
將其與表中(firstname, lastname)
的 UNIQUE 約束和查詢中的ON CONFLICT
子句結合使用可能是有意義的。
相關:
像這樣的東西
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
子句。
通常,您會使用事務來避免編寫復雜的查詢。
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;
您可以在 Sample 表上創建一個 after insert 觸發器以插入到其他兩個表中。
我在這樣做時看到的唯一問題是,您無法插入 adddetails,它將始終為空或在這種情況下為 ss。 無法將一列插入到樣本表中實際上並不存在的樣本中,因此您不能將其與初始插入一起發送。
另一種選擇是創建一個存儲過程來運行您的插入。
您有標記為 mysql 和 postgressql 的問題,我們在這里談論的是哪個數據庫?
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.