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

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

概述

在编码过程中,特别是在写 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 发布的第二款编程语言。

    500 引用 • 1396 回帖 • 251 关注

相关帖子

欢迎来到这里!

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

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

推荐标签 标签

  • Webswing

    Webswing 是一个能将任何 Swing 应用通过纯 HTML5 运行在浏览器中的 Web 服务器,详细介绍请看 将 Java Swing 应用变成 Web 应用

    1 引用 • 15 回帖 • 651 关注
  • 微软

    微软是一家美国跨国科技公司,也是世界 PC 软件开发的先导,由比尔·盖茨与保罗·艾伦创办于 1975 年,公司总部设立在华盛顿州的雷德蒙德(Redmond,邻近西雅图)。以研发、制造、授权和提供广泛的电脑软件服务业务为主。

    8 引用 • 44 回帖 • 3 关注
  • Wide

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

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

    30 引用 • 218 回帖 • 642 关注
  • FFmpeg

    FFmpeg 是一套可以用来记录、转换数字音频、视频,并能将其转化为流的开源计算机程序。

    23 引用 • 32 回帖 • 2 关注
  • 又拍云

    又拍云是国内领先的 CDN 服务提供商,国家工信部认证通过的“可信云”,乌云众测平台认证的“安全云”,为移动时代的创业者提供新一代的 CDN 加速服务。

    20 引用 • 37 回帖 • 566 关注
  • V2EX

    V2EX 是创意工作者们的社区。这里目前汇聚了超过 400,000 名主要来自互联网行业、游戏行业和媒体行业的创意工作者。V2EX 希望能够成为创意工作者们的生活和事业的一部分。

    16 引用 • 236 回帖 • 237 关注
  • 尊园地产

    昆明尊园房地产经纪有限公司,即:Kunming Zunyuan Property Agency Company Limited(简称“尊园地产”)于 2007 年 6 月开始筹备,2007 年 8 月 18 日正式成立,注册资本 200 万元,公司性质为股份经纪有限公司,主营业务为:代租、代售、代办产权过户、办理银行按揭、担保、抵押、评估等。

    1 引用 • 22 回帖 • 810 关注
  • Latke

    Latke 是一款以 JSON 为主的 Java Web 框架。

    71 引用 • 535 回帖 • 833 关注
  • Lute

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

    29 引用 • 202 回帖 • 37 关注
  • JavaScript

    JavaScript 一种动态类型、弱类型、基于原型的直译式脚本语言,内置支持类型。它的解释器被称为 JavaScript 引擎,为浏览器的一部分,广泛用于客户端的脚本语言,最早是在 HTML 网页上使用,用来给 HTML 网页增加动态功能。

    730 引用 • 1284 回帖
  • 印象笔记
    3 引用 • 16 回帖
  • Outlook
    1 引用 • 5 回帖
  • 微服务

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

    96 引用 • 155 回帖 • 2 关注
  • Hprose

    Hprose 是一款先进的轻量级、跨语言、跨平台、无侵入式、高性能动态远程对象调用引擎库。它不仅简单易用,而且功能强大。你无需专门学习,只需看上几眼,就能用它轻松构建分布式应用系统。

    9 引用 • 17 回帖 • 638 关注
  • SEO

    发布对别人有帮助的原创内容是最好的 SEO 方式。

    36 引用 • 200 回帖 • 43 关注
  • 大疆创新

    深圳市大疆创新科技有限公司(DJI-Innovations,简称 DJI),成立于 2006 年,是全球领先的无人飞行器控制系统及无人机解决方案的研发和生产商,客户遍布全球 100 多个国家。通过持续的创新,大疆致力于为无人机工业、行业用户以及专业航拍应用提供性能最强、体验最佳的革命性智能飞控产品和解决方案。

    2 引用 • 14 回帖
  • Oracle

    Oracle(甲骨文)公司,全称甲骨文股份有限公司(甲骨文软件系统有限公司),是全球最大的企业级软件公司,总部位于美国加利福尼亚州的红木滩。1989 年正式进入中国市场。2013 年,甲骨文已超越 IBM,成为继 Microsoft 后全球第二大软件公司。

    107 引用 • 127 回帖 • 340 关注
  • 创造

    你创造的作品可能会帮助到很多人,如果是开源项目的话就更赞了!

    187 引用 • 1021 回帖
  • App

    App(应用程序,Application 的缩写)一般指手机软件。

    91 引用 • 384 回帖
  • 导航

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

    45 引用 • 177 回帖
  • 职场

    找到自己的位置,萌新烦恼少。

    127 引用 • 1708 回帖 • 2 关注
  • Maven

    Maven 是基于项目对象模型(POM)、通过一小段描述信息来管理项目的构建、报告和文档的软件项目管理工具。

    188 引用 • 319 回帖 • 234 关注
  • Typecho

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

    12 引用 • 67 回帖 • 445 关注
  • 浅吟主题

    Jeffrey Chen 制作的思源笔记主题,项目仓库:https://github.com/TCOTC/Whisper

    2 引用 • 32 回帖 • 2 关注
  • OneDrive
    2 引用
  • 30Seconds

    📙 前端知识精选集,包含 HTML、CSS、JavaScript、React、Node、安全等方面,每天仅需 30 秒。

    • 精选常见面试题,帮助您准备下一次面试
    • 精选常见交互,帮助您拥有简洁酷炫的站点
    • 精选有用的 React 片段,帮助你获取最佳实践
    • 精选常见代码集,帮助您提高打码效率
    • 整理前端界的最新资讯,邀您一同探索新世界
    488 引用 • 384 回帖 • 5 关注
  • Kotlin

    Kotlin 是一种在 Java 虚拟机上运行的静态类型编程语言,由 JetBrains 设计开发并开源。Kotlin 可以编译成 Java 字节码,也可以编译成 JavaScript,方便在没有 JVM 的设备上运行。在 Google I/O 2017 中,Google 宣布 Kotlin 成为 Android 官方开发语言。

    19 引用 • 33 回帖 • 90 关注