在Go语言中，如何正确关闭多个Goroutine共享的数据库连接？-Golang-PHP中文网

在Go语言中，如何正确关闭多个Goroutine共享的数据库连接？

心靈之曲

发布： 2025-03-22 10:30:11

原创

1001人浏览过

在go语言中，如何正确关闭多个goroutine共享的数据库连接？

Go语言多Goroutine共享数据库连接的优雅关闭

在Go语言并发编程中，多个Goroutine共享数据库连接时，如何安全地关闭连接是一个关键问题。不当的关闭方式可能导致数据丢失或程序崩溃。本文将探讨几种方案，并分析其优缺点。

假设我们有一个场景：多个Goroutine并发执行数据库查询操作，共享同一个数据库连接。

错误示范：使用defer在主Goroutine关闭连接

以下代码演示了一个常见的错误：

立即学习“go语言免费学习笔记（深入）”；

db := openDB()
defer db.Close() // 错误：在主Goroutine关闭，其他Goroutine可能仍在使用

for i := 0; i < 10; i++ {
    go func(i int) {
        queryDB(db, i)
    }(i)
}
// ... 其他代码 ...

登录后复制

defer db.Close() 会在主Goroutine结束时执行，但此时其他Goroutine可能仍在使用数据库连接，导致程序崩溃或数据错误。

错误示范：在每个Goroutine中关闭连接

将db.Close()放在每个Goroutine中也不是正确的方案：

func queryDB(db *DB, i int) {
    defer db.Close() // 错误：每个Goroutine都尝试关闭连接
    // ... 数据库查询操作 ...
}

登录后复制

这会导致连接被多次关闭，引发错误。

正确方案一：使用WaitGroup同步Goroutine

使用sync.WaitGroup可以确保所有Goroutine都完成工作后再关闭连接：

var wg sync.WaitGroup
db := openDB()
defer db.Close() // 正确：在所有Goroutine完成后关闭

for i := 0; i < 10; i++ {
    wg.Add(1)
    go func(i int) {
        defer wg.Done()
        queryDB(db, i)
    }(i)
}
wg.Wait() // 等待所有Goroutine完成

登录后复制

此方法通过wg.Add(1)和wg.Done()来计数Goroutine，wg.Wait()阻塞直到所有Goroutine完成，确保连接在安全的时间点关闭。

正确方案二：使用全局变量和主Goroutine控制

另一种更简洁的方法是将数据库连接定义为全局变量，并在主Goroutine中统一管理：

var db *DB

func main() {
    db = openDB()
    defer db.Close() // 正确：在主Goroutine统一关闭

    // ... 启动Goroutine ...
}

func queryDB(i int) {
    // ... 数据库查询操作 ...
}

登录后复制

这种方式清晰地将连接的管理集中在主Goroutine，避免了并发访问和关闭的问题。

选择哪种方案取决于具体应用场景和代码复杂度。WaitGroup方法适用于更复杂的并发场景，而全局变量方法在简单场景下更简洁易懂。关键在于确保数据库连接只被关闭一次，并且在所有Goroutine完成操作之后关闭。

以上就是在Go语言中，如何正确关闭多个Goroutine共享的数据库连接？的详细内容，更多请关注php中文网其它相关文章！