总结
豆包 AI 助手文章总结
首页 > 后端开发 > Golang > 正文

如何在Go语言中多个goroutine共享数据库连接并确保正确关闭?

聖光之護
发布: 2025-03-20 08:42:15
原创
242人浏览过

如何在go语言中多个goroutine共享数据库连接并确保正确关闭?

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

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

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

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

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

立即学习go语言免费学习笔记(深入)”;

为了解决这个问题,并提高数据库连接的利用率,建议使用连接池。连接池预先创建一定数量的数据库连接,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共享数据库连接并确保正确关闭?的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
豆包 AI 助手文章总结
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号