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: %v\n", 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共享数据库连接并确保正确关闭的最佳实践，它不仅保证了连接的正确关闭，也提高了连接的复用率，避免了频繁创建和销毁连接的开销。
