defer介绍
defer是golang的一个特色功能,被称为“延迟调用函数”。当外部函数返回后执行defer。类似于其他语言的 try… catch … finally… 中的finally,当然差别还是明显的。在使用defer之前我们应该多了解defer的特性,这样才能避免使用上的误区。
1. 最简单的defer
func test(){
defer func(){ ("defer") }()
//todo
//...
return
//defer执行时机
}
我们可以稍微改动一下上述代码再次确认defer的执行时机。
func main() {
(test())
}
func test() (i int) {
defer func() { i++ }()
defer func() { (i) }()
//todo
//...
(0)
return 1
//defer执行时机
}
output:
0
1
2
从上面示例可以发现defer执行是在return之后,且按照defer声明的先进后出顺序执行。以下是真实场景中比较常见的用法。
释放占用的资源
func test() error {
file, err := ("path")
if err != nil {
return err
}
//放在判断err状态之后
defer ()
//todo
//...
return nil
//defer执行时机
}
捕捉处理异常
func test2() {
defer func() {
if err := recover(); err != nil {
(err)
}
}()
file, err := ("path")
if err != nil {
panic(err)
}
defer ()
//todo
//...
return
//defer执行时机
}
输出日志 等收尾工作
func test3() {
t1 := ()
defer func() {
("耗时: %f s", ().Sub(t1).Seconds())
}()
//todo
//...
return
//defer执行时机
}
2. 复杂的defer
当我们已经深深记住defer的执行时机并打算翻过这一页时,事情的发展又开始偏离初始。请看如下代码
func test() {
i := 0
defer (i) //输出 0
defer func(x int) { (x) }(i) //输出 0
defer func(x *int) { (*x) }(&i) //输出 1
defer func() { (i) }() //输出 1
i++
//todo
//...
(i) //输出 1
return
}
output:
1
1
1
0 // 值未修改
0 // 值未修改
通常认为defer就像真的被挪到了return之后。但defer的本质依然是函数调用。当执行到defer定义时,首先会对参数进行求值,然后参数被压入函数调用栈,此时不会进入defer函数体,而是直到函数返回时才调用defer函数体。参数被压入函数调用栈时,如果参数是值类型,那么将复制值,如果参数是指针,那么将复制指针而不是复制指针指向的值。defer函数体内的变量是在return后执行因此不受影响。因此我们在使用defer时一定要明确函数的参数类型(如果有的话),其次要明确defer函数体内的变量引用是否正确。
以下是常见错误
func test4() error {
f, err := ("")
if err != nil {
return err
}
defer func() { () }()//错误: 关闭是B文件,f引用被重新赋值
f, err = ("")
if err != nil {
return err
}
defer func() { () }() //关闭是B文件
list := []int{1, 2}
for _, i := range list {
defer (i) //输出 2 1 //i为值类型参数被复制
defer func() { (i) }() //错误: 输出 2 2 //函数体内对i引用,留最终值
}
return nil
}
3. 更复杂点的defer
且看如下代码
type Test struct {
name string
}
func (this *Test) Point() { // this 为指针
()
}
func (this Test) Value() { //this 为值类型
()
}
func test5() {
ts := []Test{{"a"}, {"b"}, {"c"}}
for _, t := range ts {
defer () //输出 c c c
defer () //输出 c b a
}
}
看似相同的代码却输出了完全不同的结果。要理解这种差别还是得从调用函数的本质来说。golang对struct 的方法调用是这样的
defer func (this Type, para) result
struct 的方法在定义时this 采用的值类型还是指针决定defer在调用时首个参数(隐藏)的复制的不同。上述代码中Point方法定义时使用了指针作为this,因此输出是for循环最终赋值的t引用。Value方法定义时使用值类型作为this,因此输出是for循环执行的每步复制后的t。
最终只有理解了以上出现的问题,到此defer所带来困扰才能远离我们。