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

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

霞舞

发布： 2025-03-18 10:32:01

原创

516人浏览过

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

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

在Go语言并发编程中，多个goroutine共享数据库连接是常见场景。不当的连接关闭处理可能导致资源泄漏或其他问题。本文将分析问题并提供解决方案。

问题分析：

直接在主goroutine中使用defer db.Close()无法保证所有goroutine执行完毕后再关闭连接，因为主goroutine可能在子goroutine执行前就已结束。将db.Close()放在每个querydb函数中也不可行，因为这会导致多个goroutine竞争关闭同一个连接，引发错误。

解决方案：

使用WaitGroup同步goroutine的执行，确保所有goroutine完成任务后再关闭连接。

改进后的代码示例：

package main

import (
    "fmt"
    "sync"
    "time"
)

// 模拟数据库连接
type dbConn struct{}

func (c *dbConn) Close() error {
    fmt.Println("Closing database connection...")
    return nil
}

func openDb() *dbConn {
    fmt.Println("Opening database connection...")
    return &dbConn{}
}

func querydb(db *dbConn, i int, wg *sync.WaitGroup) {
    defer wg.Done()
    fmt.Printf("Goroutine %d querying database...\n", i)
    time.Sleep(time.Second) // 模拟查询操作
}

func main() {
    db := openDb()
    var wg sync.WaitGroup
    for i := 0; i < 5; i++ {
        wg.Add(1)
        go querydb(db, i, &wg)
    }
    wg.Wait() // 等待所有goroutine完成
    err := db.Close()
    if err != nil {
        fmt.Printf("Error closing database connection: %v\n", err)
    }
}

登录后复制

代码解释：