Ruby使用pg gem循环查询时遭遇PG::ConnectionBad连接关闭问题

这种单独执行查询正常，但遍历多表循环时出现PG::ConnectionBad: connection is closed的情况，我遇到过好几次，大概率是这几个原因，给你几个排查和解决的方向：

1. 连接在循环中意外中断，需要检查并重新连接 #

数据库连接可能因为超时、某张表的查询触发异常（比如权限不足、表结构异常）等原因被意外关闭。你可以在每次循环前检查连接状态，一旦发现连接失效就重新建立：

array_table_names.each do |table_name|
  # 检查连接状态，PG::CONNECTION_OK代表正常，否则重连
  if @pg_conn.status != PG::CONNECTION_OK
    # 注意替换成你的实际连接参数
    @pg_conn = PG.connect(
      dbname: "your_database",
      user: "your_user",
      password: "your_password",
      host: "your_host"
    )
  end

  # 执行查询
  res_t = @pg_conn.exec("SELECT * FROM brcmanager.\"#{table_name}\" limit 2;")
  # 这里可以添加结果处理逻辑
end

2. 结果集未及时释放，导致连接资源耗尽 #

虽然pg gem会自动管理结果集，但在高频循环中，如果不手动释放结果集，可能会导致连接资源被占满，间接引发连接关闭问题。建议在处理完结果后手动释放：

array_table_names.each do |table_name|
  res_t = @pg_conn.exec("SELECT * FROM brcmanager.\"#{table_name}\" limit 2;")
  
  # 处理查询到的数据，比如打印、存储等
  puts res_t.to_a

  # 手动释放结果集资源
  res_t.clear
end

3. 数据库服务器主动关闭空闲连接 #

如果你的循环执行时间较长，或者部分表查询耗时很短导致连接短暂空闲，数据库服务器可能会因为超时设置主动关闭连接。可以在建立连接时配置保活参数，避免被主动断开：

@pg_conn = PG.connect(
  dbname: "your_database",
  user: "your_user",
  password: "your_password",
  # 开启连接保活
  keepalives: 1,
  # 空闲30秒后开始发送保活包
  keepalives_idle: 30,
  # 每10秒发送一次保活包
  keepalives_interval: 10,
  # 连续5次未收到响应则认为连接失效
  keepalives_count: 5
)

4. 优化表名转义，避免潜在的SQL注入和语法问题 #

你当前用字符串插值的方式处理表名，虽然加了双引号，但如果表名包含特殊字符，可能会引发语法错误进而导致连接异常。建议使用pg gem提供的quote_ident方法来安全转义表名：

array_table_names.each do |table_name|
  # 安全转义表名，避免语法错误和SQL注入
  quoted_table = @pg_conn.quote_ident(table_name)
  res_t = @pg_conn.exec("SELECT * FROM brcmanager.#{quoted_table} limit 2;")
end

先从检查连接状态和手动释放结果集这两点入手，应该能解决大部分这类问题。

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