如何在客户端检测PostgreSQL的smart shutdown事件并释放连接

当PostgreSQL收到智能关闭请求后，会拒绝新连接并等待现有客户端断开。客户端可以通过以下几种方法检测该事件，主动释放连接以加速关闭流程：

• 捕获特定错误码
  PostgreSQL在智能关闭阶段，对新连接请求或现有连接的新查询会返回错误码57P01（对应"admin shutdown"）。客户端只需捕获这个错误码，就能判定数据库进入了智能关闭状态，进而触发连接释放逻辑。
  比如用Python的psycopg2库时，可以这样处理：

  import psycopg2
  from psycopg2 import OperationalError
  
  try:
      # 执行查询或尝试新连接
      cur.execute("SELECT 1;")
  except OperationalError as e:
      if e.pgcode == '57P01':
          # 触发智能关闭，关闭连接
          conn.close()
  

• 定期发送心跳查询
  对于保持长连接的客户端，建议定期执行轻量心跳查询（如SELECT 1;）。如果心跳查询失败且错误码匹配57P01，即可判定数据库正在智能关闭，主动断开连接。心跳间隔可根据业务场景调整，一般建议10-30秒，平衡检测灵敏度和性能开销。

• 利用LISTEN/NOTIFY通知机制
  可以配合简单的监控脚本实现主动通知：当监控脚本检测到PostgreSQL日志中的received smart shutdown request条目后，立即连接到数据库发送NOTIFY消息（如NOTIFY shutdown_trigger;）。客户端提前通过LISTEN shutdown_trigger;监听该频道，收到通知后直接关闭连接。
  这种方法能让客户端最快感知到关闭事件，但需要额外的日志监控脚本配合。

• 使用驱动内置的连接有效性检查
  多数主流PostgreSQL客户端驱动都提供了连接有效性检查方法，比如Java JDBC的Connection.isValid(int timeout)、Go的pgx.Conn.Ping()等。当数据库进入智能关闭阶段，这些方法会返回无效状态，客户端可据此关闭连接。

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