Go语言多Goroutine共享数据库连接的优雅关闭
在Go语言并发编程中，多个Goroutine共享数据库连接是常见场景。不当的连接关闭方式可能导致或程序崩溃。本文探讨如何安全地关闭多个Goroutine共享的数据库连接。
问题分析:
直接在主Goroutine使用defer db.Close()无法保证所有子Goroutine都已完成数据库操作，可能导致连接提前关闭，引发错误。在每个子Goroutine中使用defer db.Close()则会导致连接被多次关闭，同样引发错误。
解决方案:
立即学习“”；
推荐使用计数器和sync.WtGroup来协调Goroutine的执行和数据库连接的关闭。
示例代码:
package main import ( "fmt" "sync" "time" "database/sql" _ "github.com/go-sql-driver/mysql" // 替换成你的数据库驱动 ) type dbConn struct { conn *sql.DB wg *sync.WaitGroup } func openDb() (*dbConn, error) { db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database") // 替换成你的数据库连接字符串 if err != nil { return nil, err } return &dbConn{conn: db, wg: &sync.WaitGroup{}}, nil } func (dc *dbConn) close() error { dc.wg.Wait() // 等待所有 Goroutine 完成 return dc.conn.Close() } func querydb(dc *dbConn, i int) { defer dc.wg.Done() dc.wg.Add(1) // 计数器加1 // ... 数据库操作 ... time.Sleep(time.Second) // 模拟数据库操作耗时 fmt.Printf("Goroutine %d finishedn", i) } func main() { dc, err := openDb() if err != nil { fmt.Println("Error opening database:", err) return } defer dc.close() // 确保最后关闭连接 for i := 0; i < 5; i++ { go querydb(dc, i) } fmt.Println("Main goroutine waiting...") //time.Sleep(3*time.Second) // 可选：等待Goroutine完成，如果数据库操作时间不确定，则需要其他机制等待 }
登录后复制
代码解释:
- dbConn 结构体包含数据库连接和sync.WaitGroup，用于管理Goroutine的执行。
- openDb 函数打开数据库连接并返回 dbConn 实例。
- close 方法使用 dc.wg.Wait() 等待所有 Goroutine 完成后，再关闭数据库连接。
- querydb 函数在执行数据库操作前使用 dc.wg.Add(1) 增加计数器，操作完成后使用 defer dc.wg.Done() 减少计数器。
- main 函数启动多个 Goroutine 执行 querydb 函数，最后使用 defer dc.close() 关闭数据库连接。
这种方法确保所有 Goroutine 完成数据库操作后，再关闭数据库连接，避免了数据丢失和错误。 记住替换示例代码中的数据库连接字符串和驱动。 如果数据库操作时间不可预测，可能需要更复杂的机制来等待所有Goroutine完成，例如使用channel进行信号通知。
以上就是如何在Go语言中正确关闭多个Goroutine共享的数据库连接？的详细内容，更多请关注php中文网其它相关文章！
微信扫一扫打赏
支付宝扫一扫打赏
