关于 golang 中 db 等事务处理时代码逻辑处理

本贴最后更新于 1295 天前,其中的信息可能已经事过境迁

概述

在编码过程中,特别是在写 dbo 层的数据库操作时,我们经常会遇到写事务的地方,在很长一段时间我都是使用的正常逻辑如下:

//在新建和更新的时候注意自动创建关联和更新 func (a *ArticleRepo) Create(article interface{}) (bool, error) { db := a.db.Begin() var tags []*models.Tag user := &models.User{} ac := article.(*models.Article) if err := db.Create(&ac).Error; err != nil { db.Rollback() return false, err } for _, t := range ac.Tags { tag := &models.Tag{} if err := db.Where("name = ? ", t.Name).First(&tag).Error; err != nil { tag.Name = t.Name tag.UserId = ac.AuthorID } tags = append(tags, tag) } if err := db.Where("id = ?", ac.AuthorID).First(&user).Error; err != nil { db.Rollback() return false, errors.New("请登陆") } if err := db.Model(&user).UpdateColumn("lottery_num", gorm.Expr("lottery_num + ?", 1)).Error; err != nil { db.Rollback() return false, err } if err := db.Model(&ac).Association("Tags"). Append(tags).Error; err != nil { db.Rollback() return false, err } if err := db.Commit().Error; err != nil { db.Rollback() return false, err } return true, nil }

我发现,在处理过程中会经常要写到 db.rollback 于是在见识过比较多的常用写法后写法以“匿名函数”的写法提取出主要流程完成主体业务省下大量无用代码,同时也避免忘记 rollback 和 commit,目前的写法以 transction_test.go 中为准:

首先创建事务接受上下文,db 源,fn 方法,然后将错误 recover,进行 wraper。

transction.go:

package dbx import ( "context" "fmt" "github.com/pkg/errors" ) func NewTransaction(ctx context.Context, db Database, fn func(ctx context.Context, tx Transaction) error) (err error) { tx, err := db.Begin() if err != nil { return errors.Wrap(err, "begin") } // recover defer func() { if r := recover(); r != nil { var ok bool err, ok = r.(error) if !ok { err = fmt.Errorf("%v", r) } err = errors.WithMessage(err, "recover") } if err != nil { if e := tx.Rollback(); e != nil { err = errors.WithMessagef(err, "rollback %v", e) } return } err = errors.Wrap(tx.Commit(), "tx commit") }() return fn(ctx, tx) }

sqlconn.go:

package dbx import ( "context" "database/sql" "github.com/jmoiron/sqlx" ) type ( Database interface { Begin() (Transaction, error) Ping() error } Transaction interface { Commit() error Rollback() error Get(dst interface{}, query string, args ...interface{}) error Select(dst interface{}, query string, args ...interface{}) error Exec(query string, args ...interface{}) (sql.Result, error) ExecContext(ctx context.Context, query string, args ...interface{}) (sql.Result, error) GetContext(ctx context.Context, dst interface{}, query string, args ...interface{}) error SelectContext(ctx context.Context, dst interface{}, query string, args ...interface{}) error } ) type DatabaseSQLX struct { *sqlx.DB } func (db *DatabaseSQLX) Begin() (Transaction, error) { tx, err := db.Beginx() return &Tx{Tx: tx}, err } type Tx struct { *sqlx.Tx } func (t *Tx) Commit() error { return t.Tx.Commit() } func (t *Tx) Rollback() error { return t.Tx.Rollback() }

mysql.go

package dbx import ( "fmt" "github.com/jmoiron/sqlx" "github.com/pkg/errors" ) const DefaultMySQLDSN = "%s:%s@tcp(%s:%d)/%s?charset=utf8&parseTime=true" func ConnectMySQL(format string, data DSN) (err error) { dsn := fmt.Sprintf(format, data.User, data.Password, data.Host, data.Port, data.DBName) db, err := sqlx.Connect("mysql", dsn) if err != nil { return errors.WithStack(err) } db.SetMaxIdleConns(5) db.SetMaxOpenConns(30) DB = &DatabaseSQLX{DB: db} return errors.WithStack(DB.Ping()) }

db.go

type DSN struct { Host string Port int User string Password string DBName string }

trainsaction.go

func TestNewTransaction(t *testing.T) { ctx := context.Background() mockDB := &MockDB{} t.Run("success", func(t *testing.T) { defer resetTrans() assert.Equal(t, nil, NewTransaction(ctx, mockDB, func(ctx context.Context, tx Transaction) error { return nil })) assert.Equal(t, false, rollback) assert.Equal(t, true, commit) }) t.Run("return error", func(t *testing.T) { defer resetTrans() returnErr := errors.New("return error") err := NewTransaction(ctx, mockDB, func(ctx context.Context, tx Transaction) error { return errors.WithStack(returnErr) }) assert.Equal(t, true, rollback) assert.Equal(t, false, commit) assert.Equal(t, returnErr, errors.Cause(err)) }) t.Run("panic error", func(t *testing.T) { defer resetTrans() panicErr := errors.New("panic error") err := NewTransaction(ctx, mockDB, func(ctx context.Context, tx Transaction) error { panic(panicErr) }) assert.Equal(t, true, rollback) assert.Equal(t, false, commit) assert.Equal(t, panicErr, errors.Cause(err)) }) }
  • golang

    Go 语言是 Google 推出的一种全新的编程语言,可以在不损失应用程序性能的情况下降低代码的复杂性。谷歌首席软件工程师罗布派克(Rob Pike)说:我们之所以开发 Go,是因为过去 10 多年间软件开发的难度令人沮丧。Go 是谷歌 2009 发布的第二款编程语言。

    498 引用 • 1395 回帖 • 258 关注

相关帖子

欢迎来到这里!

我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。

注册 关于
请输入回帖内容 ...
xhaoxiong
站在巨人的肩膀上学习与创新

推荐标签 标签

  • RIP

    愿逝者安息!

    8 引用 • 92 回帖 • 394 关注
  • Pipe

    Pipe 是一款小而美的开源博客平台。Pipe 有着非常活跃的社区,可将文章作为帖子推送到社区,来自社区的回帖将作为博客评论进行联动(具体细节请浏览 B3log 构思 - 分布式社区网络)。

    这是一种全新的网络社区体验,让热爱记录和分享的你不再感到孤单!

    132 引用 • 1115 回帖 • 122 关注
  • 导航

    各种网址链接、内容导航。

    43 引用 • 177 回帖 • 3 关注
  • AWS
    11 引用 • 28 回帖 • 9 关注
  • Dubbo

    Dubbo 是一个分布式服务框架,致力于提供高性能和透明化的 RPC 远程服务调用方案,是 [阿里巴巴] SOA 服务化治理方案的核心框架,每天为 2,000+ 个服务提供 3,000,000,000+ 次访问量支持,并被广泛应用于阿里巴巴集团的各成员站点。

    60 引用 • 82 回帖 • 611 关注
  • Word
    13 引用 • 40 回帖
  • WordPress

    WordPress 是一个使用 PHP 语言开发的博客平台,用户可以在支持 PHP 和 MySQL 数据库的服务器上架设自己的博客。也可以把 WordPress 当作一个内容管理系统(CMS)来使用。WordPress 是一个免费的开源项目,在 GNU 通用公共许可证(GPLv2)下授权发布。

    66 引用 • 114 回帖 • 203 关注
  • Caddy

    Caddy 是一款默认自动启用 HTTPS 的 HTTP/2 Web 服务器。

    12 引用 • 54 回帖 • 167 关注
  • Flutter

    Flutter 是谷歌的移动 UI 框架,可以快速在 iOS 和 Android 上构建高质量的原生用户界面。 Flutter 可以与现有的代码一起工作,它正在被越来越多的开发者和组织使用,并且 Flutter 是完全免费、开源的。

    39 引用 • 92 回帖
  • jsoup

    jsoup 是一款 Java 的 HTML 解析器,可直接解析某个 URL 地址、HTML 文本内容。它提供了一套非常省力的 API,可通过 DOM,CSS 以及类似于 jQuery 的操作方法来取出和操作数据。

    6 引用 • 1 回帖 • 490 关注
  • 开源中国

    开源中国是目前中国最大的开源技术社区。传播开源的理念,推广开源项目,为 IT 开发者提供了一个发现、使用、并交流开源技术的平台。目前开源中国社区已收录超过两万款开源软件。

    7 引用 • 86 回帖
  • CentOS

    CentOS(Community Enterprise Operating System)是 Linux 发行版之一,它是来自于 Red Hat Enterprise Linux 依照开放源代码规定释出的源代码所编译而成。由于出自同样的源代码,因此有些要求高度稳定的服务器以 CentOS 替代商业版的 Red Hat Enterprise Linux 使用。两者的不同在于 CentOS 并不包含封闭源代码软件。

    239 引用 • 224 回帖
  • Postman

    Postman 是一款简单好用的 HTTP API 调试工具。

    4 引用 • 3 回帖 • 4 关注
  • Lute

    Lute 是一款结构化的 Markdown 引擎,支持 Go 和 JavaScript。

    28 引用 • 197 回帖 • 29 关注
  • 单点登录

    单点登录(Single Sign On)是目前比较流行的企业业务整合的解决方案之一。SSO 的定义是在多个应用系统中,用户只需要登录一次就可以访问所有相互信任的应用系统。

    9 引用 • 25 回帖 • 2 关注
  • 微服务

    微服务架构是一种架构模式,它提倡将单一应用划分成一组小的服务。服务之间互相协调,互相配合,为用户提供最终价值。每个服务运行在独立的进程中。服务于服务之间才用轻量级的通信机制互相沟通。每个服务都围绕着具体业务构建,能够被独立的部署。

    96 引用 • 155 回帖
  • Rust

    Rust 是一门赋予每个人构建可靠且高效软件能力的语言。Rust 由 Mozilla 开发,最早发布于 2014 年 9 月。

    58 引用 • 22 回帖 • 1 关注
  • 外包

    有空闲时间是接外包好呢还是学习好呢?

    26 引用 • 233 回帖 • 1 关注
  • Shell

    Shell 脚本与 Windows/Dos 下的批处理相似,也就是用各类命令预先放入到一个文件中,方便一次性执行的一个程序文件,主要是方便管理员进行设置或者管理用的。但是它比 Windows 下的批处理更强大,比用其他编程程序编辑的程序效率更高,因为它使用了 Linux/Unix 下的命令。

    124 引用 • 74 回帖 • 1 关注
  • iOS

    iOS 是由苹果公司开发的移动操作系统,最早于 2007 年 1 月 9 日的 Macworld 大会上公布这个系统,最初是设计给 iPhone 使用的,后来陆续套用到 iPod touch、iPad 以及 Apple TV 等产品上。iOS 与苹果的 Mac OS X 操作系统一样,属于类 Unix 的商业操作系统。

    87 引用 • 139 回帖 • 1 关注
  • WiFiDog

    WiFiDog 是一套开源的无线热点认证管理工具,主要功能包括:位置相关的内容递送;用户认证和授权;集中式网络监控。

    1 引用 • 7 回帖 • 604 关注
  • Wide

    Wide 是一款基于 Web 的 Go 语言 IDE。通过浏览器就可以进行 Go 开发,并有代码自动完成、查看表达式、编译反馈、Lint、实时结果输出等功能。

    欢迎访问我们运维的实例: https://wide.b3log.org

    30 引用 • 218 回帖 • 639 关注
  • 面试

    面试造航母,上班拧螺丝。多面试,少加班。

    325 引用 • 1395 回帖
  • HBase

    HBase 是一个分布式的、面向列的开源数据库,该技术来源于 Fay Chang 所撰写的 Google 论文 “Bigtable:一个结构化数据的分布式存储系统”。就像 Bigtable 利用了 Google 文件系统所提供的分布式数据存储一样,HBase 在 Hadoop 之上提供了类似于 Bigtable 的能力。

    17 引用 • 6 回帖 • 68 关注
  • ngrok

    ngrok 是一个反向代理,通过在公共的端点和本地运行的 Web 服务器之间建立一个安全的通道。

    7 引用 • 63 回帖 • 648 关注
  • Typecho

    Typecho 是一款博客程序,它在 GPLv2 许可证下发行,基于 PHP 构建,可以运行在各种平台上,支持多种数据库(MySQL、PostgreSQL、SQLite)。

    12 引用 • 67 回帖 • 444 关注
  • Access
    1 引用 • 3 回帖 • 5 关注