go defer 学习笔记

Golang
452
0
0
2022-06-19

什么是defer?

在Go中,一个函数调用可以跟在一个defer关键字后面,形成一个延迟函数调用。

当一个函数调用被延迟后,它不会立即被执行。它将被推入由当前协程维护的一个延迟调用堆栈。 当一个函数调用(可能是也可能不是一个延迟调用)返回并进入它的退出阶段后,所有在此函数调用中已经被推入的延迟调用将被按照它们被推入堆栈的顺序逆序执行。 当所有这些延迟调用执行完毕后,此函数调用也就真正退出了。

举个简单的例子:

package main

import "fmt"

func sum(a, b int) {
    defer fmt.Println("sum函数即将返回")
    defer fmt.Println("sum函数finished")
    fmt.Printf("参数a=%v,参数b=%v,两数之和为%v\n", a, b, a+b)
}

func main() {
    sum(1, 2)
}

output:

参数a=1,参数b=2,两数之和为3
sum函数finished
sum函数即将返回

事实上,每个协程维护着两个调用堆栈。

  • 一个是正常的函数调用堆栈。在此堆栈中,相邻的两个调用存在着调用关系。晚进入堆栈的调用被早进入堆栈的调用所调用。 此堆栈中最早被推入的调用是对应协程的启动调用。
  • 另一个堆栈是上面提到的延迟调用堆栈。处于延迟调用堆栈中的任意两个调用之间不存在调用关系。

defer函数参数估值

  • 对于一个延迟函数调用,它的实参是在此调用被推入延迟调用堆栈的时候被估值的。
  • 一个匿名函数体内的表达式是在此函数被执行的时候才会被逐个估值的,不管此函数是被普通调用还是延迟调用。
  • 例子1:
package main

import  "fmt"

func  Print(a int) {

fmt.Println("defer函数中a的值=", a)

}

func  main() {

a := 10

defer  Print(a)

a = 1000

fmt.Println("a的值=", a)

}

output:

a的值= 1000
defer函数中a的值= 10

defer Print(a) 被加入到延迟调用堆栈的时候,a 的值是5,故defer Print(a) 输出的结果为5

例子2:

package main

import "fmt"

func main() {
    func() {
        for i := 0; i < 3; i++ {
            defer fmt.Println("a=", i)
        }
    }()

    fmt.Println()
    func() {
        for i := 0; i < 3; i++ {
            defer func() {
                fmt.Println("b=", i)
            }()
        }
    }()
}

output:

a= 2
a= 1
a= 0

b= 3
b= 3
b= 3

第一个匿名函数循环中的 i 是在 fmt.Println函数调用被推入延迟调用堆栈的时候估的值,因此输出结果是 2,1,0 , 第二个匿名函数中的 i 是匿名函数调用退出阶段估的值(此时 i 已经变成3了),故结果输出:3,3,3。

其实对第二个匿名函数调用略加修改,就能使它输出和匿名函数一相同的结果:

package main

import "fmt"

func main() {
    func() {
        for i := 0; i < 3; i++ {
            defer fmt.Println("a=", i)
        }
    }()

    fmt.Println()
    func() {
        for i := 0; i < 3; i++ {
            defer func(i int) {
                fmt.Println("b=", i)
            }(i)
        }
    }()
}

output:

a= 2
a= 1
a= 0

b= 2
b= 1
b= 0

恐慌(panic)和恢复(defer + recover)

Go不支持异常抛出和捕获,而是推荐使用返回值显式返回错误。 不过,Go支持一套和异常抛出/捕获类似的机制。此机制称为恐慌/恢复(panic/recover)机制。

我们可以调用内置函数panic来产生一个恐慌以使当前协程进入恐慌状况。

进入恐慌状况是另一种使当前函数调用开始返回的途径。 一旦一个函数调用产生一个恐慌,此函数调用将立即进入它的退出阶段,在此函数调用中被推入堆栈的延迟调用将按照它们被推入的顺序逆序执行。

通过在一个延迟函数调用之中调用内置函数recover,当前协程中的一个恐慌可以被消除,从而使得当前协程重新进入正常状况。

在一个处于恐慌状况的协程退出之前,其中的恐慌不会蔓延到其它协程。 如果一个协程在恐慌状况下退出,它将使整个程序崩溃。看下面的两个例子:

package main

import (
    "fmt" 
    "time"
)

func div(a, b int) int {
    return a / b
}

func main() {
    go func() {
        fmt.Println(div(1, 0))
    }()
    time.Sleep(time.Second)
    fmt.Println("程序正常退出~~~")
}

output:

panic: runtime error: integer divide by zero

goroutine 6 [running]:
main.div(...)
        /Users/didi/Desktop/golang/defer.go:9
main.main.func1()
        /Users/didi/Desktop/golang/defer.go:14 +0x12
created by main.main
        /Users/didi/Desktop/golang/defer.go:13 +0x39
exit status 2

div函数发生panic(除数为0),因为所在协程没有恐慌恢复机制,导致整个程序崩溃。

如果div函数所在协程加上恐慌恢复(defer + recover),程序便可正常退出。

package main

import (
    "fmt" 
    "time"
)

func div(a, b int) int {
    return a / b
}

func main() {
    go func() {
        defer func() {
            v := recover()
            if v != nil {
                fmt.Println("恐慌被恢复了:", v)
            }
        }()
        fmt.Println(div(1, 0))
    }()
    time.Sleep(time.Second)
    fmt.Println("程序正常退出~~~")
}

output:

恐慌被恢复了: runtime error: integer divide by zero

程序正常退出~~~

参考文章:gfw.go101.org/article/control-flow...