gorm golang 并发连接数据库报错的解决方法

下面是“gorm golang 并发连接数据库报错的解决方法”的完整攻略。

问题现象

使用 Golang 并发访问数据库时，使用 gorm 作为 ORM 库时可能会出现报错，报错信息可能类似如下：

panic: sql: database is closed

解决方法

在使用 Golang 和 gorm 并发访问数据库时，需要遵循一些规则，否则会引起一些奇奇怪怪的报错。以下是一些可行的解决方法。

1. 数据库连接池

并发操作数据库时，由于每个 goroutine 都需要访问数据库，因此需要在连接时进行池化管理，确保每个 goroutine 都能够获得可用的数据库连接。

db, err := gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local")
if err != nil {
    // 处理错误...
}

// 设置最大连接池数
db.DB().SetMaxOpenConns(100)

// 设置闲置连接池数
db.DB().SetMaxIdleConns(10)

// 设置最大闲置时间，超过该时间的连接将会被回收
db.DB().SetConnMaxLifetime(time.Hour)

defer db.Close()

// 在查询或修改时，使用 gorm 自带的事务管理功能
db.Transaction(func(tx *gorm.DB) error {
    // 数据库操作...
})

2. 复用 Gorm DB 实例

在并发访问数据库时，每个 goroutine 操作时使用自己单独的数据库连接可能会引发死锁等问题。因此可以考虑创建一个公共的数据库连接池，并在每个 goroutine 中共享这个连接池。

var db *gorm.DB

func init() {
    var err error
    db, err = gorm.Open("mysql", "user:password@tcp(127.0.0.1:3306)/test?charset=utf8&parseTime=True&loc=Local")
    if err != nil {
        log.Fatalf("gorm.Open err: %v", err)
    }

    // SetMaxIdleConns 设置空闲连接池中连接的最大数量
    db.DB().SetMaxIdleConns(5)
}

func main() {
    // 可以开启多个 goroutine 并发执行数据库操作
    for i := 0; i < 10; i++ {
        go dbQuery()
    }

    time.Sleep(time.Second * 5)
}

func dbQuery() {
    db.Transaction(func(tx *gorm.DB) error {
        // 数据库操作...
    })
}

在上面的示例中，我们共享了一个 Gorm 的 DB 实例，并在每个 goroutine 中使用该实例进行数据库访问。具体来说，例子中我们通过 go 关键字开启了多个 goroutine 并行处理数据库操作。同时，我们使用了 Gorm 的事务管理功能 db.Transaction()，确保了每个 goroutine 中的多个数据库操作以事务的方式正确执行。

需要特别注意的是，因为我们复用同一个数据库连接，因此在指定查询或修改的数据表之前，需要在 Gorm 的实例上使用 Set() 函数指定要操作的数据表。