如何在 PostgreSQL 事务中运行依赖查询

Question

我是一个 postgres 事务，我想向两个表中的每一个插入 1 条记录，比如说 A 和 B，B 有一个外键指向在同一事务中插入到 A 的记录。

我如何在这里引用插入的记录，以便我可以在表 B 上插入

Answer 1

你没有指定实际的表结构，所以我必须编造，但是可以在一个语句中使用修改 CTE的数据来做到这一点

with new_a_row as (
  insert into a (col1, col2, col3) 
  values ('one', 'two', 42)
  returning id --<<< this is the generated primary key of the table A
)
insert into b (b_col_one, b_col_two, fk_col_to_a)
select 100, 'something', id
from new_a_row;

或者使用lastval()并在两个语句中执行此操作

begin;
insert into a (col1, col2, col3) 
values ('one', 'two', 42);

insert into b (b_col_one, b_col_two, fk_col_to_a)
values (100, 'something', lastval());
commit;

第一个解决方案将适用于任何“生成策略”，不仅是identity或serial列，而且如果您使用gen_random_uuuid()作为默认值，也适用。

第二种解决方案需要identity或旧的serial类型列。