如何在Gorm中处理PostgreSQL数据库连接超时与断开重连？

GORM 依赖底层 PostgreSQL 驱动的连接池机制，要避免连接超时，需从连接池参数和数据库层面的保活配置入手：

修改 connectDB 函数，添加连接池与保活配置 #

import (
    "database/sql"
    "time"

    "gorm.io/driver/postgres"
    "gorm.io/gorm"
)

func connectDB() (*gorm.DB, error) {
    // 先初始化底层 SQL 连接，配置连接池参数
    sqlDB, err := sql.Open("postgres", dbURL)
    if err != nil {
        return nil, err
    }

    // 配置连接池核心参数
    sqlDB.SetMaxIdleConns(5)          // 维持的最大空闲连接数
    sqlDB.SetMaxOpenConns(10)         // 允许打开的最大连接数
    sqlDB.SetConnMaxLifetime(10 * time.Minute) // 连接最大存活时间，需短于数据库的超时设置
    sqlDB.SetConnMaxIdleTime(5 * time.Minute)  // 空闲连接的最大存活时间

    // 注意：需在 dbURL 中追加 PostgreSQL 保活参数，示例：
    // dbURL = "postgres://user:pass@host:port/db?keepalives=1&keepalives_idle=300&keepalives_interval=60&keepalives_count=5"
    // 上述参数让 TCP 层定期发送保活包，防止中间网络设备断开空闲连接

    // 用配置好的 sqlDB 初始化 GORM
    db, err := gorm.Open(postgres.New(postgres.Config{
        Conn: sqlDB,
    }), &gorm.Config{})
    if err != nil {
        return nil, err
    }

    return db, nil
}

参数说明 #

• ConnMaxLifetime：设置为比数据库的 idle_in_transaction_session_timeout 短，让 GORM 主动替换旧连接，避免超时。
• TCP 保活参数：通过 DSN 配置，让网络层维持连接活跃，解决因防火墙/负载均衡超时导致的连接断开问题。

原 GetDB 仅检查 db 是否为 nil，但连接断开后 db 对象仍存在，只是底层连接失效。需添加连接健康检查，并在连接失效时触发重连：

改进 GetDB 函数，添加健康检查与并发安全 #

import "sync"

var (
    db   *gorm.DB
    mutex sync.Mutex // 避免多 goroutine 同时触发重连
)

func GetDB() (*gorm.DB, error) {
    mutex.Lock()
    defer mutex.Unlock()

    // 首次初始化连接
    if db == nil {
        var err error
        db, err = connectDB()
        return db, err
    }

    // 获取底层 SQL 连接，验证可用性
    sqlDB, err := db.DB()
    if err != nil {
        // 获取底层连接失败，直接重建
        db, err = connectDB()
        return db, err
    }

    // Ping 数据库验证连接是否存活
    if err := sqlDB.Ping(); err != nil {
        // 连接失效，关闭旧连接后重建
        _ = sqlDB.Close()
        db, err = connectDB()
        return db, err
    }

    return db, nil
}

操作层面的错误重试封装 #

在执行数据库操作时，可捕获连接类错误并自动重试：

import (
    "strings"

    "github.com/lib/pq"
)

// WithDB 封装数据库操作，遇到连接错误自动重连重试
func WithDB(fn func(*gorm.DB) error) error {
    db, err := GetDB()
    if err != nil {
        return err
    }

    err = fn(db)
    if isConnectionError(err) {
        // 清除旧连接实例，触发下次重连
        mutex.Lock()
        _ = db.Close()
        db = nil
        mutex.Unlock()

        // 重试一次
        db, err = GetDB()
        if err != nil {
            return err
        }
        return fn(db)
    }
    return err
}

// isConnectionError 判断是否为连接类错误
func isConnectionError(err error) bool {
    if err == nil {
        return false
    }
    // 匹配 PostgreSQL 连接错误码
    if pgErr, ok := err.(*pq.Error); ok {
        return pgErr.Code == "08001" || pgErr.Code == "08006"
    }
    // 匹配通用网络错误描述
    return strings.Contains(err.Error(), "connection") ||
        strings.Contains(err.Error(), "broken pipe") ||
        strings.Contains(err.Error(), "timeout")
}

使用示例：

err := WithDB(func(db *gorm.DB) error {
    return db.Create(&User{Name: "test"}).Error
})

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