Rails(Puma)中Sequel多PostgreSQL连接异常问题求助

我有一个基于Rails（使用Puma服务器）的应用，搭配多个PostgreSQL数据库，采用Sequel作为数据库驱动gem。应用启动时的多数据库连接代码如下：

require "sequel"

module MyApp
  class Databases
   class << self
     attr_accessor :first_db, :second_db

     def start_connections
       @first_db = Sequel.connect(ENV.fetch("FIRST_DB_CREDENTIONALS"))
       @second_db = Sequel.connect(ENV.fetch("SECOND_DB_CREDENTIONALS"))
     end

     def disconnect_all
       first_db.disconnect
       second_db.disconnect
     end
   end
  end
end

MyApp::Databases.start_connections

但偶尔会有请求在从第一个或第二个数据库获取数据时失败，报错信息为：

"PG::ConnectionBad: PQconsumeInput() server closed the connection unexpectedly. This probably means the server terminated abnormally before or while processing the request: ..."

我想知道如何修复该错误？是否是连接超时设置的问题？

我已经做了以下尝试，但问题仍持续存在：

• 在Puma配置中添加before_fork钩子断开所有连接：

before_fork do
  MyApp::Databases.disconnect_all
end

• 尝试在每个请求结束时手动关闭连接

1. 启用Sequel连接有效性校验 #

Sequel默认不会自动检测失效连接，开启test: :all配置，让它在取出连接前先执行简单查询验证可用性，失效则自动重建：

def start_connections
  @first_db = Sequel.connect(ENV.fetch("FIRST_DB_CREDENTIONALS"), test: :all)
  @second_db = Sequel.connect(ENV.fetch("SECOND_DB_CREDENTIONALS"), test: :all)
end

2. 调整PostgreSQL端超时参数 #

数据库可能会回收闲置过久的连接，检查并修改postgresql.conf中的以下参数：

# 闲置事务超时回收，避免无效连接占用
idle_in_transaction_session_timeout = 5min
# TCP保活探测，及时识别断开的连接
tcp_keepalives_idle = 30s

修改后重启PostgreSQL服务。

3. 匹配连接池与Puma线程数 #

确保Sequel连接池大小不小于Puma的最大线程数，避免连接耗尽。连接时指定池大小：

def start_connections
  pool_size = ENV.fetch("DB_POOL_SIZE", 5).to_i
  @first_db = Sequel.connect(ENV.fetch("FIRST_DB_CREDENTIONALS"), test: :all, pool_size: pool_size)
  @second_db = Sequel.connect(ENV.fetch("SECOND_DB_CREDENTIONALS"), test: :all, pool_size: pool_size)
end

同时保证Puma配置的max_threads不超过连接池大小。

4. 优化全局连接的线程安全 #

当前全局实例存储连接在多线程环境下可能有竞争问题，改用延迟加载+Sequel内置连接池管理：

module MyApp
  class Databases
    class << self
      def first_db
        @first_db ||= Sequel.connect(ENV.fetch("FIRST_DB_CREDENTIONALS"), test: :all)
      end

      def second_db
        @second_db ||= Sequel.connect(ENV.fetch("SECOND_DB_CREDENTIONALS"), test: :all)
      end

      def disconnect_all
        first_db.disconnect
        second_db.disconnect
      end
    end
  end
end

这样每次调用都会从连接池获取可用连接，而非直接使用全局实例。

5. 捕获异常自动重试 #

在数据库操作代码中捕获PG::ConnectionBad，断开并重连后重试：

def with_db_retry(db, retries: 2)
  retries.times do
    begin
      yield db
      return
    rescue PG::ConnectionBad => e
      db.disconnect
      db.reconnect
    end
  end
  yield db # 最后一次尝试，抛出异常便于排查
end

# 使用示例
with_db_retry(MyApp::Databases.first_db) do |db|
  db[:users].where(id: 1).first
end

内容的提问来源于stack exchange，提问作者J. Doe