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

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

心靈之曲

发布： 2025-03-14 10:28:23

原创

760人浏览过

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

在Go语言并发编程中，正确管理共享数据库连接至关重要。本文探讨如何在多个goroutine共享同一数据库连接时，确保连接的可靠关闭，避免资源泄漏。

问题：多个goroutine并发访问共享数据库连接，如何保证连接在所有goroutine结束后正确关闭？

错误方法及原因分析：

在主函数中使用defer db.Close(): 这仅在主goroutine结束时执行，无法保证所有子goroutine都已完成数据库操作。

立即学习“go语言免费学习笔记（深入）”；
在每个goroutine中使用defer db.Close(): 这会导致多个goroutine尝试同时关闭同一个连接，引发错误。
等待所有goroutine结束后关闭连接: 需要一种机制来协调所有goroutine的结束，并确保只有一个goroutine负责关闭连接。

推荐解决方案：使用WaitGroup同步goroutine

sync.WaitGroup 提供了一种优雅的机制来等待一组goroutine完成。

package main

import (
    "database/sql"
    "fmt"
    "sync"
    _ "github.com/go-sql-driver/mysql" // 替换成你的数据库驱动
)

var db *sql.DB
var wg sync.WaitGroup

func openDb() (*sql.DB, error) {
    // ... 数据库连接配置 ...
    db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database")
    if err != nil {
        return nil, err
    }
    return db, nil
}

func querydb(i int) {
    defer wg.Done()
    // ... 数据库操作 ...
    row := db.QueryRow("SELECT ...")
    var result string
    err := row.Scan(&result)
    if err != nil {
        fmt.Printf("Error in goroutine %d: %v\n", i, err)
    } else {
        fmt.Printf("Goroutine %d: %s\n", i, result)
    }
}

func main() {
    var err error
    db, err = openDb()
    if err != nil {
        fmt.Println("Error opening database:", err)
        return
    }
    defer db.Close()

    for i := 0; i < 5; i++ {
        wg.Add(1)
        go querydb(i)
    }

    wg.Wait() // 等待所有goroutine完成
    fmt.Println("All goroutines finished.")
}

登录后复制

代码解释：