Go: defer与return小记

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

本文来自:简书

感谢作者:语鬼_7022

查看原文:Go: defer与return小记

• 编辑
• 预览