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

Go语言并发编程中，多个Goroutine共享数据库连接的优雅关闭至关重要。直接使用`defer db.Close()`可能导致数据丢失或程序崩溃。本文介绍如何利用计数器和`sync.WaitGroup`安全地关闭多个Goroutine共享的数据库连接，避免因连接提前关闭或多次关闭引发的错误。通过示例代码演示了`dbConn`结构体、`openDb`、`close`和`querydb`函数的协同工作，确保所有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 < 5; i++ {
        go querydb(dc, i)
    }

    fmt.Println("Main goroutine waiting...")
    //time.Sleep(3*time.Second) // 可选：等待Goroutine完成，如果数据库操作时间不确定，则需要其他机制等待
}

代码解释:

1. dbConn 结构体包含数据库连接和sync.WaitGroup，用于管理Goroutine的执行。
2. openDb 函数打开数据库连接并返回 dbConn 实例。
3. close 方法使用 dc.wg.Wait() 等待所有 Goroutine 完成后，再关闭数据库连接。
4. querydb 函数在执行数据库操作前使用 dc.wg.Add(1) 增加计数器，操作完成后使用 defer dc.wg.Done() 减少计数器。
5. main 函数启动多个 Goroutine 执行 querydb 函数，最后使用 defer dc.close() 关闭数据库连接。

这种方法确保所有 Goroutine 完成数据库操作后，再关闭数据库连接，避免了数据丢失和错误。 记住替换示例代码中的数据库连接字符串和驱动。 如果数据库操作时间不可预测，可能需要更复杂的机制来等待所有Goroutine完成，例如使用channel进行信号通知。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。