在Go语言并发编程中,多个Goroutine共享数据库连接是常见场景。不当的连接关闭方式可能导致数据丢失或程序崩溃。本文探讨如何安全地关闭多个Goroutine共享的数据库连接。
问题分析:
直接在主Goroutine使用defer db.Close()无法保证所有子Goroutine都已完成数据库操作,可能导致连接提前关闭,引发错误。在每个子Goroutine中使用defer db.Close()则会导致连接被多次关闭,同样引发错误。
解决方案:
立即学习“go语言免费学习笔记(深入)”;
推荐使用计数器和sync.WaitGroup来协调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 finished\n", 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完成,如果数据库操作时间不确定,则需要其他机制等待 }
代码解释:
这种方法确保所有 Goroutine 完成数据库操作后,再关闭数据库连接,避免了数据丢失和错误。 记住替换示例代码中的数据库连接字符串和驱动。 如果数据库操作时间不可预测,可能需要更复杂的机制来等待所有Goroutine完成,例如使用channel进行信号通知。
以上就是如何在Go语言中正确关闭多个Goroutine共享的数据库连接?的详细内容,更多请关注php中文网其它相关文章!
每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号