Go语言中如何共享数据库连接并确保goroutine安全关闭？

Go语言多goroutine并发访问数据库时，如何优雅地共享连接并确保正确关闭是常见问题。直接使用`defer db.Close()`无效，`sync.WaitGroup`协调虽然可行但复杂。本文介绍最佳实践：使用连接池。连接池预先创建连接，goroutine按需获取和归还，程序退出时统一关闭，有效避免资源竞争和泄漏，提升数据库连接利用率。文章将通过代码示例详解连接池的实现和使用方法，并对比其他方法的优劣。

Go语言多goroutine共享数据库连接及优雅关闭的最佳实践

在Go语言中，多个goroutine共享数据库连接并确保正确关闭是并发编程中的常见挑战。本文将分析一个新手常见的错误示例，并提供最佳解决方案。

新手通常会尝试使用defer db.Close()来关闭数据库连接，但这种方法在多goroutine场景下无效。defer语句仅在当前goroutine结束时执行，无法保证所有goroutine结束后再关闭连接。将defer db.Close()放在共享数据库连接的函数内部同样无效，因为这会导致多个goroutine尝试同时关闭同一个连接，引发错误。

一些新手尝试使用sync.WaitGroup来协调goroutine的执行，并在所有goroutine完成后关闭连接。虽然可行，但代码较为复杂。

更优雅的解决方案：使用连接池

为了解决这个问题，并提高数据库连接的利用率，建议使用连接池。连接池预先创建一定数量的数据库连接，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共享数据库连接并确保正确关闭的最佳实践，它不仅保证了连接的正确关闭，也提高了连接的复用率，避免了频繁创建和销毁连接的开销。

到这里，我们也就讲完了《Go语言中如何共享数据库连接并确保goroutine安全关闭？》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！