1 官方定义
A defer statement pushes a function call onto a list. The list of saved calls is executed after the surrounding function returns. Defer is commonly used to simplify functions that perform various clean-up actions.
defer表达式将一个函数调用保存在列表中,当包裹defer的函数"返回"后,列表中的调用会被执行。defer通常用于清理收尾工作。
注意:这里的返回加了引号,原因如下
2 实现逻辑
参考 雨痕大神的读书笔记(https://github.com/qyuhen/book)源码第20章
大致表达为:
step 1 : 在defer表达式的地方,会调用runtime.deferproc(size int32, fn *funcval)保存延时调用,注意这里保存了延时调用的参数
step 2 : 在return时,先将返回值保存起来
step 3 : 按FILO顺序调用runtime.deferreturn,即延时调用
step 4 : RET指令
因此,return并不是一个原子操作,函数返回值可能与你的预期不一样。
3 避坑提示
1. defer的参数在声明时即被确定下来,先看个例子(生产环境这样写估计会被唾沫喷死)
func calc(index string, a, b int) int {
ret := a + b
fmt.Println(index, a, b, ret)
return ret
}
func main() {
a := 1
b := 2
defer calc("1", a, calc("10", a, b))
a = 0
defer calc("2", a, calc("20", a, b))
b = 1
}
输出结果为:
10 1 2 3
20 0 2 2
2 0 2 2
1 1 3 4
原因是defer calc("1", a, calc("10", a, b)) 的第3个参数会在调用runtime.deferproc时确定,并不会在延时调用时才会被计算。
2. 有名与无名返回值
func namedReturn() (r int) {
defer func() {
r++
fmt.Println("defer in namedReturn : r = ", r)
}()
return
}
func unnamedReturn() int {
var r int
defer func() {
r++
fmt.Println("defer in unnamedReturn : r = ", r)
}()
return r
}
func main() {
fmt.Println("namedReturn : r = ", namedReturn())
fmt.Println("unnamedReturn : r = ", unnamedReturn())
}
输出结果为:
defer in namedReturn : r = 1
namedReturn : r = 1
defer in unnamedReturn : r = 1
unnamedReturn : r = 0
原因就是return会将返回值先保存起来,对于无名返回值来说,保存在一个临时对象中,defer是看不到这个临时对象的;而对于有名返回值来说,就保存在已命名的变量中。
3. 延时参数与有名返回值遮蔽
func ShelteredReturn() (r int) {
defer func(r int) {
r++
fmt.Println("defer in ShelteredReturn : r = ", r)
}(r)
return 0
}
func main() {
fmt.Println("ShelteredReturn : r = ", ShelteredReturn())
}
输出结果为:
defer in ShelteredReturn : r = 1
ShelteredReturn : r = 0
虽然r是有名返回值,但在defer func(r int)中的r是形参,与ShelteredReturn的返回值不是同一个。
参考文献
[1]. http://lib.csdn.net/article/go/33950
[2]. https://blog.golang.org/defer-panic-and-recover
[3]. https://my.oschina.net/henrylee2cn/blog/505535
[4]. https://github.com/qyuhen/book