golang如何优雅的编写事务代码

2022年1月23日 432点热度 0人点赞 0条评论

【导读】go 项目中如何编写事务代码?本文做了详细介绍。

前言

图片

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

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

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

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

需求

执行一个事务,需要调用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 nil
ROLLBACK:
 rollback()
 return err
}

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

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

  1. defer里有rollback的代码
  2. goto虽然好,但是不建议使用

我们可以对panic和defer进行封装,模拟一下try-catch,实现如下
图片
图片
可以看到,rollback只调用了一次,完美的进行了事务代码重构

try-catch.go代码

package exception

type 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()
}

使用代码

func
    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 = nil
            commit()
        },
        Catch: func(e interface{}) {
            rollback()
            fmt.Printf("%v panic\n", e)
            err = fmt.Errorf("%v", e)
        },
    }.Do()
    return err
}

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

转自:

cnblogs.com/chenqionghe/p/12958025.html

 - EOF -

推荐阅读(点击标题可打开)

1、Go 超时处理、非阻塞通道操作、通道的关闭和遍历

2、用Go实现TCP连接的双向拷贝

3、Go实现xunit风格的用例编写

Go 开发大全

参与维护一个非常全面的Go开源技术资源库。日常分享 Go, 云原生、k8s、Docker和微服务方面的技术文章和行业动态。

图片

关注后获取

回复 Go 获取6万star的Go资源库

分享、点赞和在看

支持我们分享更多好文章,谢谢!

8410golang如何优雅的编写事务代码

root

这个人很懒,什么都没留下

文章评论