什么是 defer?如何理解 defer 关键字?Go 中使用 defer 的一些坑。

defer 意为延迟,在 golang 中用于延迟执行一个函数。它可以帮助我们处理容易忽略的问题,如资源释放、连接关闭等。但在实际使用过程中,有一些需要注意的地方(坑),下面我们一一道来。

一些结论

首先,我们来了解 defer 的一些结论:

1、若函数中有多个 defer,其执行顺序为 先进后出,可以理解为栈。

 1package main
 2
 3import "fmt"
 4
 5func main() {
 6  for i := 0; i < 5; i++ {
 7    defer fmt.Println(i)
 8  }
 9}
10
11Output:
124
133
142
151
160

2、return 会做几件事:

给返回值赋值 调用 defer 表达式 返回给调用函数

 1package main
 2
 3import "fmt"
 4
 5func main() {
 6    fmt.Println(increase(1))
 7}
 8
 9func increase(d int) (ret int) {
10  defer func() {
11    ret++
12  }()
13
14  return d
15}
16  
17Output:
182

3、若 defer 表达式有返回值,将会被丢弃。

更多请参考官方文档

闭包与匿名函数

匿名函数:没有函数名的函数。 闭包:可以使用另外一个函数作用域中的变量的函数。

在实际开发中,defer 的使用经常伴随着闭包与匿名函数的使用。小心踩坑哦:

 1package main
 2
 3import "fmt"
 4
 5func main() {
 6    for i := 0; i < 5; i++ {
 7        defer func() {
 8            fmt.Println(i)
 9        }()
10    }
11}
12
13Output:
145
155
165
175
185

解释一下,defer 表达式中的 i 是对 for 循环中 i 的引用。到最后,i 加到 5,故最后全部打印 5。

如果将 i 作为参数传入 defer 表达式中,在传入最初就会进行求值保存,只是没有执行延迟函数而已。

1for i := 0; i < 5; i++ {
2    defer func(idx int) {
3        fmt.Println(idx)
4    }(i) // 传入的 i,会立即被求值保存为 idx
5}

巩固一下

为了巩固一下上面的知识点,我们来思考几个例子。

例1:

1func f() (result int) {
2    defer func() {
3        result++
4    }()
5    return 0
6}

例2:

1func f() (r int) {
2    t := 5
3    defer func() {
4        t = t + 5
5    }()
6    return t
7}

例3:

1func f() (r int) {
2    defer func(r int) {
3        r = r + 5
4    }(r)
5    return 1
6}

例4:

 1type Test struct {
 2    Max int
 3}
 4
 5func (t *Test) Println() {
 6    fmt.Println(t.Max)
 7}
 8
 9func deferExec(f func()) {
10    f()
11}
12
13func call() {
14    var t *Test
15    defer deferExec(t.Println)
16
17    t = new(Test)
18}

有没有得出结果?例1的答案不是 0,例2的答案不是 10,例3的答案也不是 6。

例1,比较简单,参考结论2,将 0 赋给 result,defer 延迟函数修改 result,最后返回给调用函数。正确答案是 1。

例2,defer 是在 t 赋值给 r 之后执行的,而 defer 延迟函数只改变了 t 的值,r 不变。正确答案 5。

例3,这里将 r 作为参数传入了 defer 表达式。故 func (r int) 中的 r 非 func f() (r int) 中的 r,只是参数命名相同而已。正确答案 1。

例4,这里将发生 panic。将方法传给 deferExec,实际上在传的过程中对方法求了值。而此时的 t 任然为 nil。

参考文档 [1] https://tiancaiamao.gitbooks.io/go-internals/content/zh/03.4.html

[2] http://golang.org/ref/spec#defer_statements