sql - 使用 POSTGRES 从单个 SELECT 插入 2 个相关表

Question

尝试将数据从单个表 cc 移动到通过外键关联的 2 个表 aa 和 bb:

Table cc
====
data_a, data_b


Table aa
====
id, data_a


Table bb
====
id, data_b, aa_id

表 bb 有一个外键 aa_id。我需要在 aa 中使用 cc.data_a 创建一行，在 bb 中使用 cc.data_b 和新创建的 aa.id。

这是否可能，如果可能，如何实现？

Answer 1

这是你想要的吗？

with ar as (
      insert into aa (data_a)
         select distinct data_a
         from cc
         returning *
     )
insert into bb (data_b, aa_id)
    select cc.data_b, ar.id
    from cc join
         ar
         on cc.data_a = ar.data_a;

这假设您已经使用 serial id 列预先定义了 a 和 b。