在单个查询中插入到2个表中PostgreSQL

Question

我有这3张表。

Employee
PK : id
name

completedBy
FK : employee_id
FK : order_id

Order
PK : id
date

我创建了用于创建订单的表单，其中填写了订单（日期）的信息以及完成订单的人。在我的表单中，有一个表格可供我选择员工并获取其ID。我想知道是否可以通过一个查询将表Order和completelyBy插入表中。

使用两个插入内容或使用答案中的代码之间在有效性上有什么区别吗？

Answer 1

这可以使用修改公共表表达式的数据来完成：

with new_order as (
  insert into orders (id, date) values (1, current_date)
  returning id
)
insert into completedby (employee_id, order_id)
values 
( 42 -- employee_id, 
  (select id from new_order)
);

第一部分插入orders表中并返回插入的ID。然后，第二部分completedby使用已知的employee_id并将行插入表中，并从上一步中检索order_id。

编辑

如果id在列orders表是一个serial列，你想让序列生成，你可以做到这一点，以及价值：

with new_order as (
  insert into orders (date) values (current_date)
  returning id
)
insert into completedby (employee_id, order_id)
values 
( 42 -- employee_id, 
  (select id from new_order)
);