在Flyway中如何获取插入行ID并用于另一表的插入操作？

当然可以实现！不管是PostgreSQL（serial4是PostgreSQL的自增ID类型）本身的SQL语法，还是Flyway的脚本支持，都能搞定插入时获取自增ID并直接用于后续关联插入的需求，完全不用手动调整序列值。

方法一：用PostgreSQL的RETURNING子句（纯SQL方案） #

PostgreSQL原生支持在INSERT语句后加RETURNING子句，直接返回插入行的指定字段，你可以把关联插入逻辑放在同一个事务里，两种实现方式：

1. 用CTE一次性完成关联插入 #

这种方式最简洁，一次执行就能完成两张表的插入：

WITH inserted_country AS (
    -- serial4类型的id会自动生成，不需要手动指定
    INSERT INTO country (iso) VALUES ('GBR')
    RETURNING id
)
INSERT INTO player (name, country_id)
SELECT 'bob', id FROM inserted_country;

如果像你例子里那样手动指定了id值，也可以用同样的方式返回：

WITH inserted_country AS (
    INSERT INTO country (id, iso) VALUES (10, 'GBR')
    RETURNING id
)
INSERT INTO player (name, country_id)
SELECT 'bob', id FROM inserted_country;

2. 用会话变量暂存ID（适合多步骤操作） #

如果需要拆分步骤执行，Flyway支持在SQL脚本里用PostgreSQL的DO块来定义变量暂存ID：

DO $$
DECLARE
    country_id INTEGER;
BEGIN
    INSERT INTO country (id, iso) VALUES (10, 'GBR')
    RETURNING id INTO country_id;
    
    INSERT INTO player (name, country_id) VALUES ('bob', country_id);
END $$;

方法二：Flyway回调扩展（复杂场景） #

如果需要在多个Flyway迁移脚本之间复用这个ID，可以用Flyway的Java回调功能，在插入后把ID存到上下文里，但这种方式适合复杂业务场景，一般纯SQL方案就足够解决你的问题。

注意事项 #

• 尽量别手动执行setval调整序列，除非你能明确把控后续的ID范围，不然很容易触发主键冲突；
• 要是手动指定了serial4类型的id值，一定要确认这个值没被序列的nextval占用，不然后续自动生成ID会报错；
• 所有操作最好放在同一个事务里，保证数据一致性。

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