Rails中PostgreSQL连接管理及Upsert自定义连接优化咨询

嘿，针对你的两个问题，我来给你梳理一下Rails里PostgreSQL连接管理的门道：

其实Rails已经帮你把大部分连接管理的脏活累活都干了，核心靠的是数据库连接池。默认情况下，Rails会根据config/database.yml里的pool配置（默认是5）维护一组预建立的连接——当请求进来时，从池里捞一个可用连接用，请求结束后自动放回池里，完全不用你手动操心。

这里有几个关键细节要注意：

• 每个Rails进程（比如Puma的worker进程）都有自己独立的连接池，所以设置pool数值时，得结合你的进程数和并发请求量来算，别拍脑袋瞎设
• 如果池里的连接都被占满了，Rails会等一段时间（默认5秒，可通过wait_timeout配置调整），要是还拿不到连接就会抛出ActiveRecord::ConnectionTimeoutError，这时候就得考虑调大池的大小或者优化请求的并发度了

举个配置示例，你可以在config/database.yml里这么写：

production:
  adapter: postgresql
  database: my_app_production
  pool: 10  # 根据实际并发需求调整，比如Puma开4个worker，总连接数就是4*10=40，别超过PostgreSQL的max_connections上限
  wait_timeout: 10  # 延长等待连接的超时时间到10秒

首先得说：如果你为了Upsert手动创建多个PostgreSQL连接，这其实是绕远路了——不仅容易导致连接泄漏，还浪费了Rails自带连接池的优势。

更优的姿势：用Rails内置的Upsert方法 #

从Rails 6开始，ActiveRecord已经原生支持Upsert操作了，直接用就行，自动复用连接池里的连接，根本不用手动建连接：

# 单个记录的Upsert，遇到email冲突就更新name
User.upsert({ email: 'foo@example.com', name: 'Foo' }, unique_by: :email)

# 批量Upsert，效率更高
User.upsert_all([
  { email: 'foo@example.com', name: 'Foo' },
  { email: 'bar@example.com', name: 'Bar' }
], unique_by: :email)

真要手动建连接？那这些规则必须遵守 #

如果因为某些特殊场景（比如用了原生SQL做复杂Upsert）不得不手动创建连接，那一定要记住：

• 能复用就复用：别每次操作都新建连接，要么自己整个小型连接池，要么直接拿Rails连接池里的原生连接（ActiveRecord::Base.connection.raw_connection就能拿到）
• 必须手动关闭！：Rails的连接池管不了你自己建的连接，不关闭的话，数据库的连接数会一直涨，直到达到上限，新请求直接被拒。比如用pg gem的话：

conn = PG.connect(dbname: 'my_db')
# 执行你的Upsert SQL
conn.exec("INSERT INTO users (email, name) VALUES ('foo@example.com', 'Foo') ON CONFLICT (email) DO UPDATE SET name = EXCLUDED.name")
conn.close  # 这步绝对不能忘！

另外，为了避免因为代码抛出异常导致连接没关闭，最好用ensure块兜底：

conn = PG.connect(dbname: 'my_db')
begin
  # 执行你的Upsert操作
ensure
  conn.close unless conn.closed?  # 不管有没有异常，都确保连接关闭
end

最后提个小建议 #

平时可以用PostgreSQL的这个SQL查看当前活跃连接数，及时发现连接泄漏的问题：

SELECT count(*) FROM pg_stat_activity;

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