如何用单条SQL语句实现多表关联插入（引用前表序列ID）

当然可以用单条SQL语句实现这种链式插入，你之前的写法问题在于没有正确引用CTE返回的值，以及错误嵌套了数据修改型CTE。正确的写法是利用顶层CTE的链式引用，每个后续的INSERT通过SELECT从前面的CTE获取返回的ID：

WITH ins1 AS (
    INSERT INTO tab1 (col) VALUES ('val')
    RETURNING id AS tab1id
),
ins2 AS (
    INSERT INTO tab2 (col)
    SELECT tab1id FROM ins1
    RETURNING id AS tab2id
)
INSERT INTO finaltab (col)
SELECT tab2id FROM ins2;

为什么之前的写法不行？ #

1. 第一个尝试的问题：ins2里用VALUES(tab1id)是无效的，因为tab1id是ins1 CTE的列，VALUES子句无法直接引用外部CTE的变量，必须通过SELECT从ins1中查询获取。
2. 第二个尝试的问题：PostgreSQL不允许在子查询内部使用数据修改型CTE（比如INSERT/UPDATE/DELETE的CTE），这类CTE必须放在语句的顶层，所以嵌套写法会报错。

扩展：插入多行的情况 #

如果需要一次性插入多条关联数据，只要保持CTE的链式查询逻辑即可，比如：

WITH ins1 AS (
    INSERT INTO tab1 (col) VALUES ('val1'), ('val2')
    RETURNING id AS tab1id, col AS tab1col
),
ins2 AS (
    INSERT INTO tab2 (col, tab1_col)
    SELECT tab1id, tab1col FROM ins1
    RETURNING id AS tab2id
)
INSERT INTO finaltab (col)
SELECT tab2id FROM ins2;

内容的提问来源于stack exchange，提问作者zjmarlow