Go 语言, defer 实现原理

简介: defer 语句用于延迟函数的调用,使用 defer 关键字修饰一个函数,会将这个函数压入栈中,当函数返回时,再把栈中函数取出执行。
  1. 下面程序输出什么?
func deferTest() {
  var a = 1
  defer fmt.Println(a)
  a = 2
  return
}
复制代码

是:1

解析:延迟函数 fmt.Println(a) 的参数在 defer 语句出现的时候就已经确定下来了,所以不管后面如何修改 a 变量,都不会影响延迟函数。

  1. 下面程序输出什么?
package main
import "fmt"
func main() {
  deferTest()
}
func deferTest() {
  var arr = [3]int{1, 2, 3}
  defer printTest(&arr)
  arr[0] = 4
  return
}
func printTest(array *[3]int) {
  for i := range array {
    fmt.Println(array[i])
  }
}
复制代码

是:

4
2
3
复制代码

解析:延迟函数 printTest() 的参数在 defer 语句出现的时候就已经确定下来了,即为数组的地址,延迟函数执行的时机是在 return 语句之前,所以对数组的最终修改的值会被打印出来。

  1. 下面程序输出什么?
package main
import "fmt"
func main() {
  res := deferTest()
  fmt.Println(res)
}
func deferTest () (result int) {
  i := 1
  defer func() {
    result++
  }()
  return i
}
复制代码

是:

2
复制代码

解析:函数的 return 语句并不是原子级的,实际的执行过程为为设置返回值—>ret,defer 语句是在返回前执行,所以返回过程是:设置返回值—>执行defer—>ret。所以 return 语句先把 result 设置成 i 的值(1),defer 语句中又把 result 递增 1 ,所以最终返回值为 2 。

defer 规则

  • 延迟函数的参数在defer语句出现时就已经确定
  • 注意:对于指针类型的参数,规则仍然适用,不过延迟函数的参数是一个地址值,这种情况下,defer 后面的语句对变量的修改可能会影响延迟函数。
  • 延迟函数执行按 先进后出 顺序执行,即先出现的 defer 最后执行
  • 延迟函数可能操作主函数的具名返回值

函数返回过程

上面题目中我们已经了解到,函数的 return 语句并不是原子级的,实际上 return 语句只代理汇编指令 ret。返回过程是:设置返回值—>执行defer—>ret

func deferTest () (result int) {
  i := 1
  defer func() {
    result++
  }()
  return i
}
复制代码

上面有 defer 例子的return 语句实际执行过程是:

result = i
result++
return
复制代码

主函数拥有匿名返回值,返回字面值时

当主函数有一个匿名返回值,返回时使用字面值,例如返回 “1”,“2”,“3” 这样的值,此时 defer 语句是不能操作返回值的。

func test() int {
  var i int
  defer func() {
    i++
  }()
  return 1
}
复制代码

上面的 return 语句,直接把1作为返回值,延迟函数无法操作返回值,所以也就不能修改返回值。

主函数拥有匿名返回值,返回变量时

当主函数有一个匿名返回值,返回会使用本地或者全局变量,此时 defer 语句可以引用到返回值,但不会改变返回值。

func test() int {
  var i int
  defer func() {
    i++
  }()
  return i
}
复制代码

上面的函数,返回一个局部变量,defer 函数也有操作这个局部变量。对于匿名返回值来说,我们可以假定仍然有一个变量用来存储返回值,例如假定返回值变量为 ”aaa”,上面的返回语句可以拆分成以下过程:

aaa = i
i++
return 
复制代码

由于i是整型,会将值拷贝给变量 aaa,所以defer语句中修改 i的值,对函数返回值不造成影响。

主函数拥有具名返回值时

主函声明语句中带名字的返回值,会被初始化成一个局部变量,函数内部可以像使用局部变量一样使用该返回值。如果 defer 语句操作该返回值,可能会改变返回结果。

package main
import "fmt"
func main() {
  res := test()
  fmt.Println(res) // 1
}
func test() (i int) {
  defer func() {
    i++
  }()
  return 0
}
复制代码

上面的返回语句可以拆分成以下过程:

i = 0
i++
return 
复制代码

defer实现原理

源码包 src/src/runtime/runtime2.go:_defer 定义了defer的数据结构:

type _defer struct {
  siz     int32 // includes both arguments and results
  started bool
  heap    bool
  // openDefer indicates that this _defer is for a frame with open-coded
  // defers. We have only one defer record for the entire frame (which may
  // currently have 0, 1, or more defers active).
  openDefer bool
  sp        uintptr  // sp at time of defer
  pc        uintptr  // pc at time of defer
  fn        *funcval // can be nil for open-coded defers
  _panic    *_panic  // panic that is running defer
  link      *_defer
  // If openDefer is true, the fields below record values about the stack
  // frame and associated function that has the open-coded defer(s). sp
  // above will be the sp for the frame, and pc will be address of the
  // deferreturn call in the function.
  fd   unsafe.Pointer // funcdata for the function associated with the frame
  varp uintptr        // value of varp for the stack frame
  // framepc is the current pc associated with the stack frame. Together,
  // with sp above (which is the sp associated with the stack frame),
  // framepc/sp can be used as pc/sp pair to continue a stack trace via
  // gentraceback().
  framepc uintptr
}
复制代码
  • sp 函数栈指针
  • pc 程序计数器
  • fn 函数地址
  • link 指向自身结构的指针,用于链接多个 defer

defer 语句后面是要跟一个函数的,所以 defer 的数据结构跟一般的函数类似,不同之处是 defer 结构含有一个指针,用于指向另一个 defer ,每个 goroutine 数据结构中实际上也有一个 defer 指针指向一个 defer 的单链表,每次声明一个defer 时就将 defer 插入单链表的表头,每次执行 defer 时就从单链表的表头取出一个 defer 执行。保证 defer 是按 FIFO 方式执行的。

defer的创建和执行

源码包 src/runtime/panic.go 中定义了两个方法分别用于创建defer和执行defer。

  • deferproc(): 在声明 defer 处调用,其将defer 函数存入 goroutine 的链表中;
  • deferreturn():在 return 指令,准确的讲是在 ret 指令前调用,其将 defer 从 goroutine链表中取出并执行。

归纳总结

  1. defer 定义的延迟函数的参数在 defer 语句出时就已经确定下来了
  2. defer 定义顺序与实际执行顺序相反
  3. return 不是原子级操作的,执行过程是: 保存返回值—>执行 defer —>执行ret
相关文章
|
8天前
|
JSON 中间件 Go
go语言后端开发学习(四) —— 在go项目中使用Zap日志库
本文详细介绍了如何在Go项目中集成并配置Zap日志库。首先通过`go get -u go.uber.org/zap`命令安装Zap,接着展示了`Logger`与`Sugared Logger`两种日志记录器的基本用法。随后深入探讨了Zap的高级配置,包括如何将日志输出至文件、调整时间格式、记录调用者信息以及日志分割等。最后,文章演示了如何在gin框架中集成Zap,通过自定义中间件实现了日志记录和异常恢复功能。通过这些步骤,读者可以掌握Zap在实际项目中的应用与定制方法
go语言后端开发学习(四) —— 在go项目中使用Zap日志库
|
1天前
|
安全 Java Go
探索Go语言在高并发环境中的优势
在当今的技术环境中,高并发处理能力成为评估编程语言性能的关键因素之一。Go语言(Golang),作为Google开发的一种编程语言,以其独特的并发处理模型和高效的性能赢得了广泛关注。本文将深入探讨Go语言在高并发环境中的优势,尤其是其goroutine和channel机制如何简化并发编程,提升系统的响应速度和稳定性。通过具体的案例分析和性能对比,本文揭示了Go语言在实际应用中的高效性,并为开发者在选择合适技术栈时提供参考。
|
5天前
|
运维 Kubernetes Go
"解锁K8s二开新姿势!client-go:你不可不知的Go语言神器,让Kubernetes集群管理如虎添翼,秒变运维大神!"
【8月更文挑战第14天】随着云原生技术的发展,Kubernetes (K8s) 成为容器编排的首选。client-go作为K8s的官方Go语言客户端库,通过封装RESTful API,使开发者能便捷地管理集群资源,如Pods和服务。本文介绍client-go基本概念、使用方法及自定义操作。涵盖ClientSet、DynamicClient等客户端实现,以及lister、informer等组件,通过示例展示如何列出集群中的所有Pods。client-go的强大功能助力高效开发和运维。
22 1
|
5天前
|
SQL 关系型数据库 MySQL
Go语言中使用 sqlx 来操作 MySQL
Go语言因其高效的性能和简洁的语法而受到开发者们的欢迎。在开发过程中,数据库操作不可或缺。虽然Go的标准库提供了`database/sql`包支持数据库操作,但使用起来稍显复杂。为此,`sqlx`应运而生,作为`database/sql`的扩展库,它简化了许多常见的数据库任务。本文介绍如何使用`sqlx`包操作MySQL数据库,包括安装所需的包、连接数据库、创建表、插入/查询/更新/删除数据等操作,并展示了如何利用命名参数来进一步简化代码。通过`sqlx`,开发者可以更加高效且简洁地完成数据库交互任务。
13 1
|
11天前
|
XML JSON Go
微服务架构下的配置管理:Go 语言与 yaml 的完美结合
微服务架构下的配置管理:Go 语言与 yaml 的完美结合
|
11天前
|
程序员 Go
Go 语言:面向对象还是非面向对象?揭开编程语言的本质
Go 语言:面向对象还是非面向对象?揭开编程语言的本质
|
5天前
|
算法 NoSQL 中间件
go语言后端开发学习(六) ——基于雪花算法生成用户ID
本文介绍了分布式ID生成中的Snowflake(雪花)算法。为解决用户ID安全性与唯一性问题,Snowflake算法生成的ID具备全局唯一性、递增性、高可用性和高性能性等特点。64位ID由符号位(固定为0)、41位时间戳、10位标识位(含数据中心与机器ID)及12位序列号组成。面对ID重复风险,可通过预分配、动态或统一分配标识位解决。Go语言实现示例展示了如何使用第三方包`sonyflake`生成ID,确保不同节点产生的ID始终唯一。
go语言后端开发学习(六) ——基于雪花算法生成用户ID
|
6天前
|
JSON 缓存 监控
go语言后端开发学习(五)——如何在项目中使用Viper来配置环境
Viper 是一个强大的 Go 语言配置管理库,适用于各类应用,包括 Twelve-Factor Apps。相比仅支持 `.ini` 格式的 `go-ini`,Viper 支持更多配置格式如 JSON、TOML、YAML
go语言后端开发学习(五)——如何在项目中使用Viper来配置环境
|
11天前
|
存储 编译器 Go
Go语言中的逃逸分析
Go语言中的逃逸分析
|
11天前
|
存储 Go
掌握 Go 语言的 defer 关键字
掌握 Go 语言的 defer 关键字