go语言多goroutine共享数据库连接及优雅关闭的最佳实践
在Go语言中，多个goroutine共享数据库连接并确保正确关闭是并发编程中的常见挑战。本文将分析一个新手常见的错误示例，并提供最佳解决方案。
新手通常会尝试使用defer db.Close()来关闭数据库连接，但这种方法在多goroutine场景下无效。defer语句仅在当前goroutine结束时执行，无法保证所有goroutine结束后再关闭连接。将defer db.Close()放在共享数据库连接的函数内部同样无效，因为这会导致多个goroutine尝试同时关闭同一个连接，引发错误。
一些新手尝试使用sync.WaitGroup来协调goroutine的执行，并在所有goroutine完成后关闭连接。虽然可行，但代码较为复杂。
更优雅的解决方案：使用连接池
立即学习“go语言免费学习笔记（深入）”；
为了解决这个问题，并提高数据库连接的利用率，建议使用连接池。连接池预先创建一定数量的数据库连接，goroutine从池中获取连接使用，用完后归还到池中。当程序退出时，关闭连接池即可释放所有连接。
以下是一个使用连接池的示例：
package main import ( "database/sql" "fmt" "sync" _ "github.com/go-sql-driver/mysql" // 替换成你的数据库驱动 ) type dbConn struct { conn *sql.DB mu sync.Mutex } func (dc *dbConn) GetConn() (*sql.DB, error) { dc.mu.Lock() defer dc.mu.Unlock() return dc.conn, nil } func (dc *dbConn) Close() error { dc.mu.Lock() defer dc.mu.Unlock() return dc.conn.Close() } func querydb(dc *dbConn, i int) { conn, err := dc.GetConn() if err != nil { fmt.Printf("Error getting connection: %vn", err) return } defer conn.Close() // 这里关闭的是从池中获取的连接，而不是池本身 // ... 数据库操作 ... } func main() { db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/database") if err != nil { panic(err) } defer db.Close() // 这里关闭的是连接池 dbConn := &dbConn{conn: db} var wg sync.WaitGroup for i := 0; i < 10; i++ { wg.Add(1) go func(i int) { defer wg.Done() querydb(dbConn, i) }(i) } wg.Wait() }
这个示例中，dbConn 结构体管理数据库连接，GetConn 方法从连接池获取连接，Close 方法关闭连接池。 querydb 函数从连接池获取连接，使用后归还。主函数负责创建和关闭连接池。sync.WaitGroup 用于等待所有goroutine完成。 请记得替换成你的数据库驱动和连接字符串。
使用连接池是处理Go语言多goroutine共享数据库连接并确保正确关闭的最佳实践，它不仅保证了连接的正确关闭，也提高了连接的复用率，避免了频繁创建和销毁连接的开销。
© 版权声明
文章版权归作者所有，未经允许请勿转载。
THE END
