Rails 数据库连接池水合

Question

我正在开发一个响应时间 SLA 非常严格的 Rails 7 应用程序。 在正常运行时，我完全符合 SLA。 我痛苦地缺乏的地方是第一个请求。 我添加了一个初始化程序，它将加载 ActiveRecord 并确保我的所有数据库模型都已加载。 它滋润了一些各种 memory 缓存等。这让我走了很远。 我的第一次响应时间减少了大约 60%。 但是，我一直在试图弄清楚有几件事仍在减慢第一响应时间。

1. 第一个 API 请求会检查我是否需要进行 Rails 迁移。 我还没有想出如何将此检查移动到 init。
2. 第一个 API 请求似乎正在使用新的数据库池……而不是在初始阶段使用的那个。 我已经尝试完全补水池以防止 API 在 Rails 启动时创建它们，但我还没有弄清楚。

在初始化程序中，我可以这样做：

connections = []
ActiveRecord::Base.connection.pool.size.times do
  connections << ActiveRecord::Base.connection.pool.checkout
end

connections.each { ActiveRecord::Base.connection.pool.checkin(_1) }

根据我的 PG 日志，这会打开连接，Rails 会执行所有这些键入查询，设置 session 属性等。但是，当我 go 触发我的第一个 API 调用时，我的池是空的。

Answer 1

最后，最终成为普遍问题的是我需要通过正确的连接为泳池补水。 on_worker_boot是因为这是在 puma 后面运行的。

on_worker_boot do
  ActiveRecord::Base.connected_to(role: :reading) do
    # spin up db connections
    connections = []
    (ActiveRecord::Base.connection.pool.size - 1).times do
      connections << ActiveRecord::Base.connection.pool.checkout
    end

    connections.each { |x| ActiveRecord::Base.connection.pool.checkin(x) }
  end
end