【Golang】defer的使用和底层实现

2,694 阅读4分钟

📚 全文字数 : 3k

⏳ 阅读时长 : 5min

📢 关键词 : defer、资源释放、底层实现

1:defer是什么

defer是Go语言的关键字,一般用于资源的释放和异常的捕捉(比如:文件打开、加锁、数据库连接、异常捕获),defer语句后将其后面跟随的语句进行延迟处理,就是说在函数执行完毕后再执行调用,也就是return的ret指令之前。

1.1 资源释放

资源的释放在代码中有很多场景,比如打开文件描述符资源后,需要进行file.close得到释放,在打开文件后就加上defer,避免在后续因为err导致的return退出忘记释放,文件资源。

func openFile() {
	file, err := os.Open("txt")
	if err != nil {
		return
	}
	defer file.Close() //合理位置
}

常见的加锁场景,业务代码中忘记释放锁,那么会导致资源得不到释放,造成死锁,但是defer就很好解决了这个问题,不管业务逻辑怎么处理,最终还是会释放锁。

func lockScene() {
	 var mutex sync.Mutex
	 mutex.Lock()
	 defer  mutex.Unlock()
	 //业务代码...
}

1.2 捕获异常

Go 语言中 recover 关键字主要用于捕获异常,让程序回到正常状态。recover 可以中止 panic 造成的程序崩溃。它是一个只能在 defer 中发挥作用的函数,在其他作用域中调用不会发挥作用

func demo()  {
 defer func() {
  if err := recover(); err !=nil{
   fmt.Println(string(Stack()))
  }
 }()
 panic("unknown")
}

2:defer语法

defer语法相对简单,直接在普通函数之前加一个defer关键字

defer demoFunc(args)

虽然defer语法简单,但是当有多个defer注册时,会以逆序执行(类似栈:先进后出),举个栗子。

func f1() {
	defer fmt.Println("defer1")
	defer fmt.Println("defer2")
	fmt.Println("start")
	fmt.Println("end")
	return
}

这段代码的字符串输出结果是:start、end、defer2、defer1。首先输出的defer字符串在正常的start、end后输出可以很好理解(defer在函数返回前执行),字符串defer2在defer1前输出(逆序执行)。

3:defer与return

Go语言中函数的 return 语句并不是原子级的,实际的执行过程为为设置返回值—>ret指令,defer 语句是在返回前执行,所以返回过程是:「设置返回值—>执行defer—>ret」

4:defer底层实现

要了解defer的实现,先看下defer的底层数据结构和各个参数表示的意义(src/runtime/runtime2.go)

type _defer struct {
   siz     int32    // 参数和返回值的内存大小
   started bool
   heap    bool       //是否分配在堆上面
   openDefer bool     // 是否经过开放编码优化
   sp        uintptr  // sp 计数器值,栈指针
   pc        uintptr  // pc 计数器值,程序计数器
   fn        *funcval // defer 传入的函数地址,也就是延后执行的函数
   _panic    *_panic  // defer 的 panic 结构体
   link      *_defer  // 同一个协程里面的defer 延迟函数,会通过该指针连接在一起
}

defer怎么实现延迟的

通过资料了解到,defer在代码中的位置在编译后会有两部分内容: 1:deferproc负责把要执行的函数保存起来,我们称之为defer注册 2:deferreturn是在defer注册完成(deferproc)后,程序执行后续业务代码,直到通过deferreturn执行注册的defer函数

为啥是逆序执行

defer结构有个link指针,是指向的一个defer单链表的头,每次咱们声明一个defer的时候,就会将该defer的数据插入到这个单链表头部的位置,取defer进行执行的时候,是从单链表的头开始去取的,这就是defer先进后出的原因。 底层代码在src/runtime/panic.go,核心代码做了说明

func deferproc(siz int32, fn *funcval) { // arguments of fn follow fn
	gp := getg() //获取goroutine结构
	if gp.m.curg != gp {
		// go code on the system stack can't defer
		throw("defer on system stack")
	}
	...
	d := newdefer(siz) //新建一个defer结构
	if d._panic != nil {
		throw("deferproc: d.panic != nil after newdefer")
	}
	d.link = gp._defer // 新建defer的link指针指向g的defer
	gp._defer = d      // 新建defer放到g的defer位置,完成插入链表表头操作
	d.fn = fn
	d.pc = callerpc
	d.sp = sp
	...
}

如图:先声明defer fun1()、再声明 defer fun2(),fun2()在单链表链表头部。

总结

1:defer关键字后面必须是函数,也叫延迟函数

2:defer是逆序执行(后进先出),延迟函数中的参数在defer声明的时候已经确定了

3:在函数return之前执行延迟函数

文末安利一波:

欢迎朋友们关注我的同名公众号📢📢:【小许code】,等你哦!🤣🤣

欢迎点赞 👍、收藏 💙、关注 💡 三连支持一下~🎈

🎈知道的越多,不知道的也越多,我是小许,下期见~🙇💻