你是不是觉得defer很简单、很好用,但也许你掉坑里了都不知道!
这篇文章不介绍defer的常用功能,而是介绍你在用defer时,也许会踩的坑。
defer允许我们进行一些函数执行完成后的收尾工作,并且代码更加简洁,例如:
- 关闭文件流:
// open a file
defer file.Close()
- 解锁一个加锁的资源
mu.Lock()
defer mu.Unlock()
- 打印最终报告
printHeader()
defer printFooter()
- 关闭数据库链接
// open a database connection
defer disconnectFromDB()
但是:
- 你知道defer和defer后的函数什么时候执行吗?
- 你知道defer后函数里的变量值是什么时候计算的吗?
我曾经在Stack Overflow讨论过这个问题,有兴趣的可以看下,本打算在周末写个文章分享给大家defer的坑,今天不小心浏览到一个有误解的文章,决定现在就写下来,希望大家不要踩坑。
defer陷阱测试
如果下面这段代码的结果你都知道,恭喜你,你已经了解defer的执行原理,没有必要再看这篇文章了。
func test1() (x int) {
defer fmt.Printf("in defer: x = %d\n", x)
x = 7
return 9
}
func test2() (x int) {
x = 7
defer fmt.Printf("in defer: x = %d\n", x)
return 9
}
func test3() (x int) {
defer func() {
fmt.Printf("in defer: x = %d\n", x)
}()
x = 7
return 9
}
func test4() (x int) {
defer func(n int) {
fmt.Printf("in defer x as parameter: x = %d\n", n)
fmt.Printf("in defer x after return: x = %d\n", x)
}(x)
x = 7
return 9
}
func main() {
fmt.Println("test1")
fmt.Printf("in main: x = %d\n", test1())
fmt.Println("test2")
fmt.Printf("in main: x = %d\n", test2())
fmt.Println("test3")
fmt.Printf("in main: x = %d\n", test3())
fmt.Println("test4")
fmt.Printf("in main: x = %d\n", test4())
}
你已经计算出结果了吗?看看和运行结果是不是一样的,如果不一样继续阅读本文吧:
test1
in defer: x = 0
in main: x = 9
test2
in defer: x = 7
in main: x = 9
test3
in defer: x = 9
in main: x = 9
test4
in defer x as parameter: x = 0
in defer x after return: x = 9
in main: x = 9
defer执行原理
要想知道为何是这个结果,就得先回答前面的2个问题:
- defer和defer后的函数什么时候执行吗?
- defer后函数里的变量值是什么时候计算的吗?
依次来回答,这2个问题。
问题1:defer在defer语句处执行,defer的执行结果是把defer后的函数压入到栈,等待return或者函数panic后,再按先进后出的顺序执行被defer的函数。
问题2:defer的函数的参数是在执行defer时计算的,defer的函数中的变量的值是在函数执行时计算的。
defer及defer函数的执行顺序分2步:
- 执行defer,计算函数的入参的值,并传递给函数,但不执行函数,而是将函数压入栈。
- 函数return语句后,或panic后,执行压入栈的函数,函数中变量的值,此时会被计算。
defer测试解析
这4个测试函数中,都是return 9
并且没有对返回值进行修改,所以main中都是in main: x = 9
,我相信这个大家应该是没有疑问的。接下来看每个测试函数defer的打印。
test1:defer执行时,对Printf
的入参x进行计算,它的值是0,并且传递给函数,return 9
后执行Printf
,所以结果是in defer: x = 0
。
test2:与test1类似,不同仅是,defer执行是在x=7
之后,所以x的值是7,并且传递给Printf
,所以结果是:in defer: x = 7
。
test3:defer后跟的是一个匿名函数,匿名函数能访问外部函数的变量,这里访问的是test3的x,defer执行时,匿名函数没有入参,所以把func()()
压入到栈,return语句之后,执行func()()
,此时匿名函数获得x的值是9,所以结果是in defer: x = 9
。
test4:与test3的不同是,匿名函数有一个入参n,我们把x作为入参打印,还有就是匿名函数访问外部打印x。defer执行时,x=0
,所以入栈的函数是func(int)(0)
,return语句之后执行func(int)(0)
,即n=0
,x在匿名函数内没有定义,依然访问test4中的x,此时x=9
,所以结果为:in defer x as parameter: x = 0, in defer x after return: x = 9
。
From: http://lessisbetter.site/2018/11/10/Golang-trap-of-defer/
有疑问加站长微信联系(非本文作者)