在Go語言中，如何正確關閉多個Goroutine共享的數據庫連接？

Go語言多Goroutine共享數據庫連接的優雅關閉

在Go語言並發編程中，多個Goroutine共享數據庫連接時，如何安全地關閉連接是一個關鍵問題。不當的關閉方式可能導致數據丟失或程序崩潰。本文將探討幾種方案，並分析其優缺點。

假設我們有一個場景：多個Goroutine並發執行數據庫查詢操作，共享同一個數據庫連接。

錯誤示範：使用defer在主Goroutine關閉連接

以下代碼演示了一個常見的錯誤：

 db := openDB()
defer db.Close() // 錯誤：在主Goroutine關閉，其他Goroutine可能仍在使用for i := 0; i <p> <code>defer db.Close()</code>會在主Goroutine結束時執行，但此時其他Goroutine可能仍在使用數據庫連接，導致程序崩潰或數據錯誤。</p><p><strong>錯誤示範：在每個Goroutine中關閉連接</strong></p><p>將<code>db.Close()</code>放在每個Goroutine中也不是正確的方案：</p><pre class="brush:php;toolbar:false"> func queryDB(db *DB, i int) {
    defer db.Close() // 錯誤：每個Goroutine都嘗試關閉連接// ... 數據庫查詢操作...
}

這會導致連接被多次關閉，引發錯誤。

正確方案一：使用WaitGroup同步Goroutine

使用sync.WaitGroup可以確保所有Goroutine都完成工作後再關閉連接：

 var wg sync.WaitGroup
db := openDB()
defer db.Close() // 正確：在所有Goroutine完成後關閉for i := 0; i <p>此方法通過<code>wg.Add(1)</code>和<code>wg.Done()</code>來計數Goroutine， <code>wg.Wait()</code>阻塞直到所有Goroutine完成，確保連接在安全的時間點關閉。</p><p><strong>正確方案二：使用全局變量和主Goroutine控制</strong></p><p>另一種更簡潔的方法是將數據庫連接定義為全局變量，並在主Goroutine中統一管理：</p><pre class="brush:php;toolbar:false"> var db *DB

func main() {
    db = openDB()
    defer db.Close() // 正確：在主Goroutine統一關閉// ... 啟動Goroutine ...
}

func queryDB(i int) {
    // ... 數據庫查詢操作...
}

這種方式清晰地將連接的管理集中在主Goroutine，避免了並發訪問和關閉的問題。

選擇哪種方案取決於具體應用場景和代碼複雜度。 WaitGroup方法適用於更複雜的並發場景，而全局變量方法在簡單場景下更簡潔易懂。 關鍵在於確保數據庫連接只被關閉一次，並且在所有Goroutine完成操作之後關閉。

以上是在Go語言中，如何正確關閉多個Goroutine共享的數據庫連接？的詳細內容。更多資訊請關注PHP中文網其他相關文章！