如何在Go语言中正确关闭多个Goroutine共享的数据库连接？

Go语言多Goroutine共享数据库连接的优雅关闭

在Go语言并发编程中，多个Goroutine共享数据库连接是常见场景。不当的连接关闭方式可能导致数据丢失或程序崩溃。本文探讨如何安全地关闭多个Goroutine共享的数据库连接。

问题分析:

直接在主Goroutine使用defer db.Close()无法保证所有子Goroutine都已完成数据库操作，可能导致连接提前关闭，引发错误。在每个子Goroutine中使用defer db.Close()则会导致连接被多次关闭，同样引发错误。

解决方案:

推荐使用计数器和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 
<p><strong>代码解释:</strong></p>
<ol>
<li>
<code>dbConn</code> 结构体包含数据库连接和<code>sync.WaitGroup</code>，用于管理Goroutine的执行。</li>
<li>
<code>openDb</code> 函数打开数据库连接并返回 <code>dbConn</code> 实例。</li>
<li>
<code>close</code> 方法使用 <code>dc.wg.Wait()</code> 等待所有 Goroutine 完成后，再关闭数据库连接。</li>
<li>
<code>querydb</code> 函数在执行数据库操作前使用 <code>dc.wg.Add(1)</code> 增加计数器，操作完成后使用 <code>defer dc.wg.Done()</code> 减少计数器。</li>
<li>
<code>main</code> 函数启动多个 Goroutine 执行 <code>querydb</code> 函数，最后使用 <code>defer dc.close()</code> 关闭数据库连接。</li>
</ol>
<p>这种方法确保所有 Goroutine 完成数据库操作后，再关闭数据库连接，避免了数据丢失和错误。  记住替换示例代码中的数据库连接字符串和驱动。  如果数据库操作时间不可预测，可能需要更复杂的机制来等待所有Goroutine完成，例如使用channel进行信号通知。</p>

以上是如何在Go语言中正确关闭多个Goroutine共享的数据库连接？的详细内容。更多信息请关注PHP中文网其他相关文章！