一、defer用法

defer 是golang中独有的流程控制语句,用于延迟指定语句的运行时机,运行于函数的内部,当他所属函数运行完之后它才会被调用。

例如以下使用defer的代码:

func main() {
    defer fmt.Println("HelloDefer")
    fmt.Println("HelloWorld")
}

输出结果为:

HelloWorld
HelloDefer

它会先打印出HelloWorld ,然后再打印出HelloDefer。一个函数中如果有多个defer ,运行顺序和函数中的调用顺序相反,因为它们都是被写在了栈中:

func deferTest(){
    defer fmt.Println("HelloDefer1")
    defer fmt.Println("HelloDefer2")
    fmt.Println("HelloWorld")
}

运行结果:

fmt.Println("HelloDefer2")
fmt.Println("HelloDefer1")
fmt.Println("HelloWorld")

二、defer和return

在包含有return语句的函数中,defer的运行顺序位于return之后,但是defer所运行的代码片段会生效:

func deferReturn() int {
    i := 1

    defer func() {
        fmt.Println("Defer")
        i += 1
    }()

    return func() int {
        fmt.Println("Return")
        return i
    }()
}

func main(){
    fmt.Println(deferReturn)
}

运行结果:

Return
Defer
1

这里很明显就能看到defer是在return之后运行的!但是有一个问题是defer里执行了语句i += 1 ,按照这个逻辑的话返回的i 值应该是2 而不是1 。这个问题是由于return的运行机制导致的:return返回一个对象时,如果返回类型不是指针或者引用类型,那么return返回的就不是这个对象本身,而是这个对象的副本。也就是说,实际上当defer执行的时候,return已经把值返回了,返回的是一个副本值,此时defer对值的修改就无效了。

如果返回的是引用值,defer的改动是有效的。我们可以把返回值得int类型改成*int:

func deferAndReturn() *int {
    i := 1
    defer func() {
        fmt.Println("Defer:\t", i, &i)
        i += 1
    }()
    return func() *int {
        fmt.Println("Return:\t", i, &i)
        return &i
    }()
}

func main() {
    x := deferAndReturn()
    fmt.Println("main:\t", *x, x)
}

程序的输出为:

Return:  1 0xc000014060
Defer:   1 0xc000014060
main:    2 0xc000014060

可以看到,在return和defer执行前,i的值都是1,并且地址一样。返回到main函数后,i的地址也是和函数体的一样,但是值是2了,这就说明defer中对i的修改生效了。

三、defer和panic

panic会在defer运行完之后才把恐慌扩散到其他函数:

func deferPanic(){
    defer fmt.Println("HelloDefer")
    panic("Hey, I'm panic")
}

结果:

HelloDefer  //会先输出defer中的语句
panic: Hey, I'm panic

goroutine 1 [running]:
main.deferPanic()
        /Users/maqian/code/go/src/awesomeProject/defer/defer.go:21 +0xb9
main.main()
        /Users/maqian/code/go/src/awesomeProject/defer/defer.go:29 +0x82

四、defer和for循环

不要在defer内使用外部变量,可能会造成一些意想不到的错误:

func deferTest() {
    for i := 0; i < 5; i++ {
        defer func() {
            fmt.Printf("%d", i)
        }()
    }
}

它的输出的结果是55555 ,原理很简单,因为defer会在for循环运行完后才会调用,for循环运行完时i的值为5,所以打印的i值会是55555。

正确的做法是不要在defer函数中使用共享变量,而是手动传入参数:

func deferTest() {
    for i := 0; i < 5; i++ {
        defer func(i int) {
            fmt.Printf("%d", i)
        }(i)
    }
}

此时就会打印出43210而不是55555了。

最后修改:2020 年 02 月 22 日
喜欢就给我点赞吧