在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.")
}
代码解释：
-
wg.Add(1): 在每个goroutine启动前增加计数器。 -
defer wg.Done(): 在每个goroutine结束后减少计数器。 -
wg.Wait(): 阻塞主goroutine，直到计数器变为0，表示所有goroutine都已完成。 -
defer db.Close(): 确保数据库连接在主函数结束时关闭。
这种方法保证了数据库连接在所有goroutine完成其数据库操作后被安全地关闭，避免了资源泄漏和并发冲突。 请记住替换代码中的占位符数据库连接字符串和驱动程序。
