go中defer的理解--deferreturn返回值之间执行顺序

Posted 2021-01-03 ricklz

defer可以读取有名返回值

func c() (i int) {
defer func() { i++ }()
return 1
}

输出结果是2. 在开头的时候，我们知道defer是在return调用之后才执行的。 这里需要明确的是defer代码块的作用域仍然在函数之内，结合上面的函数也就是说，defer的作用域仍然在c函数之内。因此defer仍然可以读取c函数内的变量(如果无法读取函数内变量，那又如何进行变量清除呢....)。

当执行return 1 之后，i的值就是1. 此时此刻，defer代码块开始执行，对i进行自增操作。 因此输出2.

A. 无名返回值的情况

package main

import (
        "fmt"
)

func main() {
        fmt.Println("return:", a()) // 打印结果为 return: 0
}

func a() int {
        var i int
        defer func() {
                i++
                fmt.Println("defer2:", i) // 打印结果为 defer: 2
        }()
        defer func() {
                i++
                fmt.Println("defer1:", i) // 打印结果为 defer: 1
        }()
        return i
}

　　B. 有名返回值的情况

package main

import (
        "fmt"
)

func main() {
        fmt.Println("return:", b()) // 打印结果为 return: 2
}

func b() (i int) {
        defer func() {
                i++
                fmt.Println("defer2:", i) // 打印结果为 defer: 2
        }()
        defer func() {
                i++
                fmt.Println("defer1:", i) // 打印结果为 defer: 1
        }()
        return i // 或者直接 return 效果相同
}

先来假设出结论，帮助大家理解

1. 多个defer的执行顺序为“后进先出”；

2. defer、return、返回值三者的执行逻辑应该是：return最先执行，return负责将结果写入返回值中；接着defer开始执行一些收尾工作；最后函数携带当前返回值退出。

如何解释两种结果的不同：

上面两段代码的返回结果之所以不同，其实从上面第2条结论很好理解。 
a()int 函数的返回值没有被提前声明，其值来自于其他变量的赋值，而defer中修改的也是其他变量，而非返回值本身，因此函数退出时返回值并没有被改变。 
b()(i int) 函数的返回值被提前声明，也就意味着defer中是可以调用到真实返回值的，因此defer在return赋值返回值 i 之后，再一次地修改了 i 的值，最终函数退出后的返回值才会是defer修改过的值。

C. 下面我们再来看第三个例子，验证上面的结论：

package main

import (
        "fmt"
)

func main() {
        fmt.Println("c return:", *(c())) // 打印结果为 c return: 2
}

func c() *int {
        var i int
        defer func() {
                i++
                fmt.Println("c defer2:", i) // 打印结果为 c defer: 2
        }()
        defer func() {
                i++
                fmt.Println("c defer1:", i) // 打印结果为 c defer: 1
        }()
        return &i
}

　　虽然 c()*int 的返回值没有被提前声明，但是由于 c()*int 的返回值是指针变量，那么在return将变量 i 的地址赋给返回值后，defer再次修改了 i 在内存中的实际值，因此函数退出时返回值虽然依旧是原来的指针地址，但是其指向的内存实际值已经被成功修改了。 
即，我们假设的结论是正确的！