如何在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中文網其他相關文章！