首页 > 后端开发 > Golang > 正文

Golang并发处理数据库访问安全示例

P粉602998670
发布: 2025-10-01 15:37:02
原创
839人浏览过
Go的database/sql包支持并发安全的数据库访问,通过连接池管理多个goroutine对同一sql.DB实例的安全使用。示例中10个goroutine并发执行查询,依赖连接池分配独立连接,确保操作隔离。需注意应用层竞态条件:共享map等非线程安全结构必须用sync.Mutex或sync.Map保护。事务(sql.Tx)不可跨goroutine使用,每个事务应在单个goroutine内完成,避免并发提交或回滚错误。为防资源耗尽,应限制并发量,如通过带缓冲channel实现信号量机制,控制同时运行的goroutine数量。综上,合理利用连接池、避免共享可变状态、隔离事务上下文、控制并发度是保障Go数据库并发安全的关键实践。

golang并发处理数据库访问安全示例

在Go语言中,并发处理数据库访问时,确保数据安全和程序稳定性非常重要。Go的database/sql包本身是线程安全的,底层通过连接池管理并发请求,但应用层仍需注意操作逻辑的正确性,避免竞态条件。

使用 database/sql 连接池处理并发

Go的标准库database/sql已经为并发访问做了优化,它维护一个数据库连接池,多个goroutine可以安全地共用同一个*sql.DB实例。

示例:

package main
<p>import (
"database/sql"
"log"
"sync"</p><pre class='brush:php;toolbar:false;'>_ "github.com/go-sql-driver/mysql"
登录后复制

)

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

func main() { db, err := sql.Open("mysql", "user:password@tcp(127.0.0.1:3306)/testdb") if err != nil { log.Fatal(err) } defer db.Close()

var wg sync.WaitGroup

for i := 0; i < 10; i++ {
    wg.Add(1)
    go func(id int) {
        defer wg.Done()
        var name string
        err := db.QueryRow("SELECT name FROM users WHERE id = ?", id).Scan(&name)
        if err != nil {
            log.Printf("查询失败 ID %d: %v", id, err)
            return
        }
        log.Printf("用户 %d: %s", id, name)
    }(i)
}

wg.Wait()
登录后复制

}

在这个例子中,多个goroutine同时使用同一个db对象执行查询,这是安全的,因为*sql.DB会从连接池中分配空闲连接。

避免共享可变状态

虽然*sql.DB是安全的,但如果你在多个goroutine之间共享其他变量(如结构体、切片等),需要额外同步。

错误示例:多个goroutine写入同一map

var resultMap = make(map[int]string)
var mu sync.Mutex // 必须加锁保护
<p>for i := 0; i < 10; i++ {
go func(id int) {
var name string
db.QueryRow("SELECT name FROM users WHERE id = ?", id).Scan(&name)</p><pre class='brush:php;toolbar:false;'>    mu.Lock()
    resultMap[id] = name  // 写map必须加锁
    mu.Unlock()
}(i)
登录后复制

}

说明:map不是并发安全的,必须使用sync.Mutexsync.Map来保护共享数据。

怪兽AI知识库
怪兽AI知识库

企业知识库大模型 + 智能的AI问答机器人

怪兽AI知识库 51
查看详情 怪兽AI知识库

事务中的并发注意事项

数据库事务(*sql.Tx)不能被多个goroutine同时使用,每个事务必须在单个goroutine中完成。

错误做法:

tx, _ := db.Begin()
go func() { tx.Exec("INSERT ...") }() // ❌ 不允许跨goroutine使用tx
go func() { tx.Commit() }()
登录后复制

正确方式:将事务操作封装在单一goroutine内:

go func() {
    tx, err := db.Begin()
    if err != nil {
        log.Println(err)
        return
    }
    defer tx.Rollback()
<pre class='brush:php;toolbar:false;'>_, err = tx.Exec("INSERT INTO users(name) VALUES(?)", "alice")
if err != nil {
    log.Println(err)
    return
}

err = tx.Commit()
if err != nil {
    log.Println(err)
}
登录后复制

}()

每个事务独立在自己的goroutine中开始、执行、提交,避免共享*sql.Tx

控制并发数量避免资源耗尽

虽然连接池能管理连接,但大量并发可能耗尽数据库连接或内存。建议限制并发数。

使用带缓冲的channel控制并发:

semaphore := make(chan struct{}, 5) // 最多5个并发
<p>for i := 0; i < 20; i++ {
wg.Add(1)
go func(id int) {
defer wg.Done()
semaphore <- struct{}{} // 获取令牌
defer func() { <-semaphore }() // 释放令牌</p><pre class='brush:php;toolbar:false;'>    var name string
    db.QueryRow("SELECT name FROM users WHERE id = ?", id).Scan(&name)
    log.Printf("ID %d: %s", id, name)
}(i)
登录后复制

}

这样即使启动20个goroutine,也最多5个同时执行数据库操作,防止压垮数据库。

基本上就这些。只要合理使用连接池、避免共享非线程安全结构、不在goroutine间共享事务、控制并发量,就能安全高效地在Go中并发访问数据库。

以上就是Golang并发处理数据库访问安全示例的详细内容,更多请关注php中文网其它相关文章!

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

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

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

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