Go语言中 defer 使用场景及注意事项,你是要注意的!

goCenter · · 608 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

**文章来自:[Go语言圈](https://mp.weixin.qq.com/s/rt9nzMaXwTLJfMWUuy_Meg "Go语言圈")** &emsp; #### 1 简介 defer会在当前函数返回前执行传入的函数,它会经常被用于关闭文件描述符、关闭数据库连接以及解锁资源。 理解这句话主要在三个方面: - 当前函数 - 返回前执行,当然函数可能没有返回值 - 传入的函数,即 `defer` 关键值后面跟的是一个函数,包括普通函数如(`fmt.Println`), 也可以是匿名函数 `func()` &emsp; **1.1 使用场景** 使用 `defer` 的最常见场景是在函数调用结束后完成一些收尾工作,例如在 `defer` 中回滚数据库的事务: ```go func createPost(db *gorm.DB) error { tx := db.Begin() // 用来回滚数据库事件 defer tx.Rollback() if err := tx.Create(&Post{Author: "Draveness"}).Error; err != nil { return err } return tx.Commit().Error } ``` 在使用数据库事务时,我们可以使用上面的代码在创建事务后就立刻调用 `Rollback` 保证事务一定会回滚。哪怕事务真的执行成功了,那么调用 `tx.Commit()` 之后再执行 `tx.Rollback()` 也不会影响已经提交的事务。 &emsp; **1.2 注意事项** 使用`defer`时会遇到两个常见问题,这里会介绍具体的场景并分析这两个现象背后的设计原理: `defer` 关键字的调用时机以及多次调用 `defer` 时执行顺序是如何确定的`defer` 关键字使用传值的方式传递参数时会进行预计算,导致不符合预期的结果 作用域 向 `defer` 关键字传入的函数会在函数返回之前运行。 假设我们在 `for` 循环中多次调用 `defer` 关键字: ```go package main import "fmt" func main() { for i := 0; i < 5; i++ { // FILO, 先进后出, 先出现的关键字defer会被压入栈底,会最后取出执行 defer fmt.Println(i) } } ``` >//运行 $ go run main.go 4 3 2 1 0 运行上述代码会倒序执行传入 `defer` 关键字的所有表达式,因为最后一次调用 `defer` 时传入了 `fmt.Println(4)`,所以这段代码会优先打印 4。我们可以通过下面这个简单例子强化对 `defer` 执行时机的理解: ```go package main import "fmt" func main() { // 代码块 { defer fmt.Println("defer runs") fmt.Println("block ends") } fmt.Println("main ends") } ``` >//输出 $ go run main.go block ends main ends defer runs 从上述代码的输出我们会发现,`defer` 传入的函数不是在退出代码块的作用域时执行的,它只会在当前函数和方法返回之前被调用。 &emsp; **预计算参数** Go 语言中所有的函数调用都是传值的. 虽然 `defer` 是关键字,但是也继承了这个特性。假设我们想要计算 `main` 函数运行的时间,可能会写出以下的代码: ```go package main import ( "fmt" "time" ) func main() { startedAt := time.Now() // 这里误以为:startedAt是在time.Sleep之后才会将参数传递给defer所在语句的函数中 defer fmt.Println(time.Since(startedAt)) time.Sleep(time.Second) } ``` >输出 $ go run main.go 0s 上述代码的运行结果并不符合我们的预期,这个现象背后的原因是什么呢? 经过分析(或者使用`debug`方式),我们会发现: 调用 `defer` 关键字会立刻拷贝函数中引用的外部参数 所以 `time.Since(startedAt)` 的结果不是在 `main` 函数退出之前计算的,而是在 `defer` 关键字调用时计算的,最终导致上述代码输出 0s。 想要解决这个问题的方法非常简单,我们只需要向 defer 关键字传入匿名函数: ```go package main import ( "fmt" "time" ) func main() { startedAt := time.Now() // 使用匿名函数,传递的是函数的指针 defer func() { fmt.Println(time.Since(startedAt)) }() time.Sleep(time.Second) } ``` >//输出 1.0056135s &emsp; #### 2 defer 数据结构 `defer `关键字在 Go 语言源代码中对应的数据结构: ```go type _defer struct { siz int32 started bool openDefer bool sp uintptr pc uintptr fn *funcval _panic *_panic link *_defer } ``` 简单介绍一下 `runtime._defer` 结构体中的几个字段: - `siz` 是参数和结果的内存大小; - `sp` 和 `pc` 分别代表栈指针和调用方的程序计数器; - `fn` 是 `defer` 关键字中传入的函数; - `_panic` 是触发延迟调用的结构体,可能为空; - `openDefer` 表示当前 `defer` 是否经过开放编码的优化; 除了上述的这些字段之外,`runtime._defer` 中还包含一些垃圾回收机制使用的字段, 这里不做过多的说明 &emsp; #### 3 执行机制 堆分配、栈分配和开放编码是处理 defer 关键字的三种方法。 早期的 Go 语言会在堆上分配, 不过性能较差 Go 语言在 1.13 中引入栈上分配的结构体,减少了 30% 的额外开销 在1.14 中引入了基于开放编码的 `defer`,使得该关键字的额外开销可以忽略不计 堆上分配暂时不做过多的说明 &emsp; **3.1 栈上分配** 在 1.13 中对 defer 关键字进行了优化,当该关键字在函数体中最多执行一次时,会将结构体分配到栈上并调用。 除了分配位置的不同,栈上分配和堆上分配的`runtime._defer` 并没有本质的不同,而该方法可以适用于绝大多数的场景,与堆上分配的 `runtime._defer` 相比,该方法可以将 defer 关键字的额外开销降低 ~30%。 &emsp; **3.2 开放编码** 在 1.14 中通过开放编码(`Open Coded`)实现 `defer` 关键字,该设计使用代码内联优化 `defer` 关键的额外开销并引入函数数据 `funcdata` 管理 `panic` 的调用3,该优化可以将 `defer` 的调用开销从 1.13 版本的~35ns 降低至 ~6ns 左右: 然而开放编码作为一种优化 `defer` 关键字的方法,它不是在所有的场景下都会开启的,开放编码只会在满足以下的条件时启用: 函数的 `defer` 数量小于或等于8个; 函数的 `defer` 关键字不能再循环中执行 函数的 `return` 语句 与 `defer` 语句个数的成绩小于或者等于15个。

有疑问加站长微信联系(非本文作者))

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

608 次点击  ∙  1 赞  
加入收藏 微博
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传