Postgres从表中插入多个返回值作为

Question

我想插入多个（2）行到Postgres的表中，并使用WITH ROWS AS返回的ID插入到另一个表（列id_1和​​second_table），并返回即 id作为查询的最终结果。

first_table是主要数据收集，而second_table基本上只是将两个first_table条目链接在一起。

这里是我想要做的要点：

WITH row AS (
     INSERT INTO first_table (some_data, more_data) 
     VALUES (DEFAULT, DEFAULT), (DEFAULT, DEFAULT) 
     RETURNING first_table_id 
     ) 
INSERT INTO second_table (id_1, id_2, other_data) 
VALUES (???, ???, DEFAULT) 
RETURNING second_table_id 

我如何使用Node.js和node-pg模块;我知道这可以在Node.js中完成，方法是让第一个查询返回WITH ROWS AS返回的两行，然后使用下一个插入的两个ID运行另一个准备好的语句。

我不确定我是否可以通过触发器来执行此操作，因为ON INSERT只有一行。也许它会有两个，因为这些行在同一个事务中插入到first_table中，但是我一般没有丰富的Postgres/SQL经验，所以我不知道如何通过触发器来做到这一点。

我遇到的问题是，WITH ROWS AS语句返回2行（显然是打算），但我不知道如何在下一个插入中独立访问每行，并且我感觉它不是不可能。

可这在单个查询来完成，还是我有用于此触发器或Node.js的？

Answer 1

第二次插入需要是INSERT INTO ... SELECT ... FROM first_cte。如果你可以做出一个假设，那第二个生成的id比第一个（或者至少是不同的 - 比如 - 如果你使用PostgreSQL的serial，或者直接使用一个序列）大，你可以使用聚合：

WITH rows AS (
    INSERT INTO first_table (some_data, more_data) 
    VALUES (DEFAULT, DEFAULT), (DEFAULT, DEFAULT) 
    RETURNING first_table_id 
) 
INSERT INTO second_table (id_1, id_2) 
SELECT min(first_table_id), max(first_table_id) 
FROM rows 
RETURNING second_table_id 

需要注意的是，省略other_data：你不能在INSERT INTO ... SELECT ...使用DEFAULT。

对于更广泛的解决方案，你可以使用数组：

WITH rows AS (
    INSERT INTO first_table (some_data, more_data) 
    VALUES (DEFAULT, DEFAULT), (DEFAULT, DEFAULT) 
    RETURNING first_table_id 
) 
INSERT INTO second_table (id_1, id_2, other_data) 
SELECT first_table_ids[1], first_table_ids[2], 5 -- value '5' for "other_data" 
FROM (
    SELECT array_agg(first_table_id) AS first_table_ids 
    FROM rows 
) AS rows_as_array 
RETURNING second_table_id