0

0

如何使用Golang对数据库操作进行测试_结合事务和回滚保证隔离

P粉602998670

P粉602998670

发布时间:2025-12-22 20:24:00

|

906人浏览过

|

来源于php中文网

原创

Golang数据库测试核心是事务回滚隔离:每个测试开启事务、执行后强制回滚,配合DBTX接口抽象与testdb等工具实现干净、可并行的测试;需注意DDL限制、引擎兼容性及外部依赖mock。

如何使用golang对数据库操作进行测试_结合事务和回滚保证隔离

用 Golang 对数据库操作做测试时,核心难点是避免测试间相互干扰、不污染真实数据,同时保证每次测试的独立性和可重复性。最稳妥的方式是:在每个测试中开启事务,执行业务逻辑,然后回滚(rollback),而不是提交(commit)。这样既模拟了真实数据库交互,又确保测试干净、隔离、可并行。

使用 testdb + 事务包装器简化测试

手动管理事务容易出错(比如忘记回滚、panic 后未恢复)。推荐用 testdb(或类似轻量工具)自动包裹事务。它本质是在测试开始前开启事务,测试结束时强制回滚,无论成功或失败。

  • 安装:go get github.com/ory/dockertest/v3(用于启动临时 PostgreSQL/MySQL 容器)或直接用内存 SQLite(适合简单场景)
  • 更轻量的选择是自己封装一个 TestTxDB:接收 *sql.DB,调用 Begin(),返回带 Rollback() 的 wrapper 结构体
  • 关键点:wrapper 必须实现 driver.Connector 或直接透传 *sql.Tx 给你的 DAO 层,让业务代码“以为”在操作普通 DB

在测试中注入事务连接

你的数据访问层(如 UserRepository)不应硬依赖 *sql.DB,而应接受可替换的 executor 接口(例如 Queryer 或自定义 DBTX)。

  • 定义接口:type DBTX interface { QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error); ExecContext(context.Context, string, ...interface{}) (sql.Result, error) }
  • 测试时传入 *sql.Tx 实例(它实现了 DBTX),生产环境传 *sql.DB
  • 这样 DAO 方法无需修改,就能自然运行在事务上下文中

注意事务边界与驱动兼容性

不是所有操作都能在事务中安全执行。例如:

Thiings
Thiings

免费的拟物化图标库

下载

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

  • DDL 语句(CREATE TABLEALTER COLUMN)在 PostgreSQL 中不能在事务块内执行(除非用 BEGIN; SET LOCAL ... 等变通)
  • 某些 MySQL 存储引擎(如 MyISAM)不支持事务,务必用 InnoDB
  • SQLite 默认支持事务,但需确认 PRAGMA journal_mode = WAL 等设置不影响测试稳定性
  • 若测试涉及 time.Sleep 或外部 HTTP 调用,事务可能超时;建议用 mock 替代外部依赖

完整测试示例(PostgreSQL + dockertest)

以下是一个最小可行测试片段:

func TestUserRepository_Create(t *testing.T) {
    pool, err := dockertest.NewPool("")
    require.NoError(t, err)
    resource, err := pool.Run("postgres", "15-alpine", []string{"POSTGRES_PASSWORD=secret"})
    require.NoError(t, err)
    defer func() { _ = pool.Purge(resource) }()

    db, err := sql.Open("postgres", fmt.Sprintf("host=localhost port=%s user=postgres password=secret dbname=postgres sslmode=disable", resource.GetPort("5432/tcp")))
    require.NoError(t, err)
    defer db.Close()

    // 每个测试用独立事务
    tx, err := db.Begin()
    require.NoError(t, err)
    defer tx.Rollback() // 关键:无论如何都回滚

    repo := NewUserRepository(tx) // 注入事务连接
    user := User{Name: "alice", Email: "a@example.com"}
    err = repo.Create(context.Background(), &user)
    require.NoError(t, err)

    var count int
    err = tx.QueryRow("SELECT COUNT(*) FROM users WHERE email = $1", user.Email).Scan(&count)
    require.NoError(t, err)
    assert.Equal(t, 1, count)
}

相关专题

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

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

180

2024.02.23

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

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

228

2024.02.23

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

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

340

2024.02.23

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

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

209

2024.03.05

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

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

393

2024.05.21

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

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

197

2025.06.09

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

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

191

2025.06.10

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

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

233

2025.06.17

AO3中文版入口地址大全
AO3中文版入口地址大全

本专题整合了AO3中文版入口地址大全,阅读专题下面的的文章了解更多详细内容。

1

2026.01.21

热门下载

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

精品课程

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

共48课时 | 1.9万人学习

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

共3课时 | 0.3万人学习

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

共1课时 | 805人学习

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

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