0

0

如何在Golang中进行数据库性能测试_Golang sql包Benchmark实践

P粉602998670

P粉602998670

发布时间:2026-02-02 15:51:08

|

759人浏览过

|

来源于php中文网

原创

不能直接用 BenchmarkXxx 测数据库操作,因连接建立、驱动初始化等干扰会导致结果失真;必须外部初始化 sql.DB、预热连接池、复用 Stmt、显式设置连接池参数,并仅将 SQL 执行放入 b.N 循环。

如何在golang中进行数据库性能测试_golang sql包benchmark实践

Go 的 database/sql 包本身不提供 Benchmark 支持,性能测试必须靠自己构造可复现的压测逻辑——直接用 go test -bench 跑数据库操作会因连接建立、事务开销、SQL 编译等干扰项导致结果失真。

为什么不能直接写 BenchmarkXxx 函数跑 db.Query

标准 go test -bench 会在短时间内反复调用 Benchmark 函数,而每次调用若都新建连接、执行 SQL、关闭资源,测的就不是“查询本身”,而是“连接池初始化 + 网络往返 + 驱动解析”的混合耗时。更严重的是,多数数据库驱动(如 lib/pqmysql/mysql-driver)在首次调用时有隐式初始化行为,会导致前几次迭代极慢,后续突然变快,benchmem 统计也会不准。

实操建议:

LLaMA
LLaMA

Meta公司发布的下一代开源大型语言模型

下载
  • 所有 sql.DB 实例必须在 BenchmarkXxx 外部初始化(例如 TestMain 或全局变量),避免每次迭代重建连接池
  • 预热连接池:在 Benchmark 开始前,用 db.Ping() + 若干次 dummy 查询触发连接建立和语句预编译
  • 禁止在 Benchmark 循环体内调用 db.Close(),否则连接池被清空,后续迭代全走新连接路径
  • 使用 db.Prepare 复用 *sql.Stmt,避免每次迭代重复解析 SQL(尤其对 PostgreSQL / MySQL 的参数化查询)

sql.DB.SetMaxOpenConnsSetMaxIdleConns 怎么设才反映真实负载

默认值(0 表示无限制)会让压测结果严重偏离生产环境——你测的是“无限并发下驱动能扛多少”,而不是“服务配置了 max_open=20 时单次查询均值”。必须按目标部署场景显式设置。

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

实操建议:

  • db.SetMaxOpenConns(20) 模拟典型 Web 服务连接池上限
  • db.SetMaxIdleConns(20) 避免频繁建连/销毁,让连接真正复用
  • 若测试高并发短连接场景(如 CLI 工具),可设 MaxOpenConns=100 + MaxIdleConns=0,但需在注释中明确标注该模式
  • db.Stats() 在 Benchmark 前后打印 OpenConnectionsWaitCount,确认是否出现连接等待(WaitCount > 0 表示连接池瓶颈)

如何写出稳定、可比的 Benchmark 函数

核心原则:只让「执行 SQL」这一动作进入 b.N 循环,其余全部前置。同时要隔离事务、错误处理、结果扫描等变量。

func BenchmarkQueryUserByID(b *testing.B) {
	db, _ := sql.Open("pgx", "user=dev dbname=test")
	db.SetMaxOpenConns(20)
	db.SetMaxIdleConns(20)
	defer db.Close()

	// 预热
	db.Ping()
	_, _ = db.Exec("SELECT 1")

	// 复用 Stmt
	stmt, _ := db.Prepare("SELECT id, name FROM users WHERE id = $1")
	defer stmt.Close()

	b.ResetTimer() // 关键:重置计时器,跳过准备阶段
	for i := 0; i < b.N; i++ {
		var id int
		var name string
		err := stmt.QueryRow(i%1000 + 1).Scan(&id, &name)
		if err != nil && err != sql.ErrNoRows {
			b.Fatal(err)
		}
	}
}

注意点:

  • b.ResetTimer() 必须在预热完成后、循环开始前调用,否则包含 Prepare/Ping 耗时
  • i % 1000 + 1 控制 ID 范围,避免查不到数据导致大量 sql.ErrNoRows 干扰
  • 不忽略 err,但只 fatal 真正异常(ErrNoRows 是业务正常路径)
  • 避免在循环里做字符串拼接、JSON 序列化等非 DB 操作,否则 benchmark 测的是 Go 运行时而非数据库延迟

容易被忽略的底层影响:驱动选择与上下文取消

同一个 SELECT,用 lib/pqjackc/pgx 基准结果可能差 30%+;而没传 context.Context 的调用,在超时时会卡死整个 goroutine,导致 b.N 实际执行次数远低于预期。

实操建议:

  • 性能敏感场景优先用 pgx/v4(PostgreSQL)或 go-sql-driver/mysql(MySQL),它们比 lib/pq 少一层 interface 抽象
  • 所有 Query/Exec 必须带 context.WithTimeout,例如 stmt.QueryRowContext(ctx, id),否则网络抖动会拖垮整个 benchmark
  • go test -benchmem -benchtime=10s 延长测试时间,减少 GC 噪声影响
  • 对比不同驱动时,确保连接字符串参数一致(如 sslmode=disablebinary_parameters=yes

真正的瓶颈往往不在 SQL 本身,而在连接池配置是否匹配压测模型、Stmt 是否复用、上下文是否可控——这些细节不拉平,Benchmark 数据之间根本不可比。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

184

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

229

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

344

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

210

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

397

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

282

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

194

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

518

2025.06.17

AO3官网入口与中文阅读设置 AO3网页版使用与访问
AO3官网入口与中文阅读设置 AO3网页版使用与访问

本专题围绕 Archive of Our Own(AO3)官网入口展开,系统整理 AO3 最新可用官网地址、网页版访问方式、正确打开链接的方法,并详细讲解 AO3 中文界面设置、阅读语言切换及基础使用流程,帮助用户稳定访问 AO3 官网,高效完成中文阅读与作品浏览。

29

2026.02.02

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
MySQL 教程
MySQL 教程

共48课时 | 2万人学习

MySQL 初学入门(mosh老师)
MySQL 初学入门(mosh老师)

共3课时 | 0.3万人学习

简单聊聊mysql8与网络通信
简单聊聊mysql8与网络通信

共1课时 | 821人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

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