为测试重新填充PostgreSQL数据库种子数据的问题

这种场景我太熟了！之前做集成测试的时候也踩过pg_restore的坑，尤其是数据库不能断连的情况下，--clean选项经常因为活跃连接、对象依赖或者权限问题炸锅。给你几个亲测有效的解决方案：

1. 先强制断开所有非当前连接 #

--clean要删除现有对象，但只要有其他进程连着数据库，就会失败。先执行这段SQL踢掉所有非当前的连接：

SELECT pg_terminate_backend(pid)
FROM pg_stat_activity
WHERE datname = 'test-db'
  AND pid <> pg_backend_pid();

直接用psql命令执行更方便：

psql -d test-db -c "SELECT pg_terminate_backend(pid) FROM pg_stat_activity WHERE datname = 'test-db' AND pid <> pg_backend_pid();"

把这个命令放在pg_restore前面跑，能解决大部分因活跃连接导致的报错。

2. 优化pg_dump和pg_restore的参数组合 #

你的pg_dump参数-O -x（去掉所有者和权限）没问题，试试给pg_restore加几个关键参数：

pg_restore --clean --if-exists -j 1 -d test-db ./tests/test-db

• --if-exists：只删除确实存在的对象，避免因不存在的对象抛出错误
• -j 1：单线程恢复，避免并发恢复时的对象依赖顺序问题
• 另外，确保你用的是自定义格式备份（你已经用了-F custom，和-Fc等价），这个格式支持更灵活的恢复控制。

3. 临时禁用外键约束（解决依赖报错） #

如果报错是因为外键约束导致无法删除表，恢复前先禁用所有表的触发器（包括外键约束），恢复后再启用：

# 恢复前禁用约束
psql -d test-db -c "ALTER TABLE ALL IN SCHEMA public DISABLE TRIGGER ALL;"

# 执行恢复
pg_restore --clean -d test-db ./tests/test-db

# 恢复后启用约束
psql -d test-db -c "ALTER TABLE ALL IN SCHEMA public ENABLE TRIGGER ALL;"

这个方法能绕开外键依赖的顺序问题，让--clean顺利删除所有对象。

4. 更高效的替代方案：用事务回滚代替备份恢复 #

如果你的测试可以在单个事务中运行，这是最优解——根本不需要备份恢复：

• 测试开始前开启一个事务
• 执行所有测试逻辑
• 测试结束后直接回滚事务
  这样数据库会自动回到初始状态，速度比备份恢复快N倍，还不会有连接问题。大部分测试框架（比如pytest、JUnit）都支持在测试前后加事务管理的钩子。

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