这篇文章主要讲解了golang编写事务的方法，内容清晰明了，对此有兴趣的小伙伴可以学习一下，相信大家阅读完之后会有帮助。

前言

新手程序员大概有如下特点

if嵌套经常超过3层、经常出现重复代码、单个函数代码特别长。只会crud，对语言特性和语言的边界不了解。不懂面向对象原则和设计模式，以为copy代码就算学会了，常见的是代码职责不明确或者写出万能类不知道数据结构和算法的重要性，以为靠硬件能解决运行慢的问题架构不懂，搭建框架不会，搭建环境不会，使用的软件底层原理一问三不知

其实吧，很多人干了很多年，看似是老手，平时工作看似很忙，其实做的都是最简单的活。
这就像去锻炼，有的人每天练的很积极，准时打卡，频繁发朋友圈，貌似是正能量，结果是几年下来体型还是那样，该减的肥肉没少，要增的肌肉没加，为什么会这样？因为从来都是挑最简单最轻松的练

貌似吐槽多了，下面演示一下如何将一坨烂事务代码重构得优雅

需求

执行一个事务，需要调用one、two、three、four、five几个方法，任意一个方法失败，都回滚事务
下面是这些方法的简单模拟，我们用尽可能少的代码模拟一个操作

//开启事务func beginTransaction() { fmt.Println("beginTransaction")}//回滚事务func rollback() { fmt.Println("rollback")}//提交事务func commit() { fmt.Println("commit")}//执行one操作func one() (err error) { fmt.Println("one ok") return nil}//执行two操作func two() (err error) { fmt.Println("two ok") return nil}//执行three操作func three() (err error) { fmt.Println("two ok") return nil}//执行four操作func four() (err error) { fmt.Println("four ok") return nil}//执行five操作func five() (err error) { err = errors.New("five panic") panic("five") return err}

烂代码示例

下面演示开启一个事务，依次执行one、two、three、four、five 5个操作，前四个成功，第五个失败

这是新手程序员常见使用事务的代码风格，其实也不光是事务，所有的代码都可能长下边这样

func demo() (err error) { beginTransaction() defer func() { if e := recover(); e != nil { err = fmt.Errorf("%v", e) fmt.Printf("%v panic\n", e) rollback() } }() if err = one(); err == nil { if err = two(); err == nil { if err = three(); err == nil { if err = four(); err == nil { if err = five(); err == nil { commit() return nil } else { rollback() return err } } else { rollback() return err } } else { rollback() return err } } else { rollback() return err } } else { rollback() return err }}

重构套路

一、提前return去除if嵌套

通过提前返回error，来去掉一些else代码，减少嵌套，如下

代码

func demo() (err error) { beginTransaction() defer func() { if e := recover(); e != nil { err = fmt.Errorf("%v", e) fmt.Printf("%v panic\n", e) rollback() } }() if err = one(); err != nil { rollback() return err } if err = two(); err != nil { rollback() return err } if err = three(); err != nil { rollback() return err } if err = four(); err != nil { rollback() return err } if err = five(); err != nil { rollback() return err } commit() return nil}

先解决嵌套

二、goto+label提取重复代码

代码

func demo() (err error) { beginTransaction() defer func() { if e := recover(); e != nil { err = fmt.Errorf("%v", e) fmt.Printf("%v panic\n", e) rollback() } }() if err = one(); err != nil { goto ROLLBACK } if err = two(); err != nil { goto ROLLBACK } if err = three(); err != nil { goto ROLLBACK } if err = four(); err != nil { goto ROLLBACK } if err = five(); err != nil { goto ROLLBACK } commit() return nilROLLBACK: rollback() return err}

三、封装try-catch统一捕获panic

上面的代码其实还有一点问题

defer里有rollback的代码goto虽然好，但是不建议使用

我们可以对panic和defer进行封装，模拟一下try-catch，实现如下

可以看到，rollback只调用了一次，完美的进行了事务代码重构

try-catch.go代码

package exceptiontype Block struct { Try func() Catch func(interface{}) Finally func()}func (t Block) Do() { if t.Finally != nil { defer t.Finally() } if t.Catch != nil { defer func() { if r := recover(); r != nil { t.Catch(r) } }() } t.Try()}

使用代码

exception.Block{Try: func() {beginTransaction()if err = one(); err != nil {panic(err)}if err = two(); err != nil {panic(err)}if err = three(); err != nil {panic(err)}if err = four(); err != nil {panic(err)}if err = five(); err != nil {panic(err)}err = nilcommit()},Catch: func(e interface{}) {rollback()fmt.Printf("%v panic\n", e)err = fmt.Errorf("%v", e)},}.Do()return err}

这样，我们就可以用非常少的代码实现事务，并且简单清晰好维护，以上为chenqionghe原创，light weight baby

看完上述内容，是不是对golang编写事务的方法有进一步的了解，如果还想学习更多内容，欢迎关注亿速云行业资讯频道。