一、defer 关键字工作原则

1. A deferred function’s arguments are evaluated when the defer statement is evaluated.

func a() {
    i := 0
    defer fmt.Println(i)
    i++
    return
}

defer 函数中的变量 i 在 defer 函数被定义的时候就已经明确，值为0。随后 defer 之外的 i++ 并不会影响 defer 函数打印的，所以打印结果为：0。

2.Deferred function calls are executed in Last In First Out order after the surrounding function returns.

func b() {
    for i := 0; i < 4; i++ {
        defer fmt.Print(i)
    }
}

多个 defer 函数被调用的顺序遵循 后进先出 的原则，所以打印结果为：3210。

3.Deferred functions may read and assign to the returning function’s named return values.

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

defer 函数可以读取并赋值给外部函数的命名返回值。所以函数 c 的返回值为：2。

二、defer return 执行流程

func f0() (r int) {
    t := 5
    defer func() {
        fmt.Printf("f0 defer t is :%d", t)//5
        fmt.Printf("f0 defer r is :%d", r)//5
        t = t + 5
        fmt.Printf("f0 defer r is :%d", r)//5
    }()
    return t
}

这里需要注意的是,return xxx语句并不是一条原子指令，defer 被插入到了 赋值 与 ret 之间。因此 return t应该被拆解程两部分，r = t 和 return。

这样 f0 函数可以被改造成下面的函数

func f0() (r int) {
    t := 5
    r = t
    defer func() {
        fmt.Printf("f0 defer t is :%d", t)//5
        fmt.Printf("f0 defer r is :%d", r)//5
        t = t + 5
        fmt.Printf("f0 defer r is :%d", r)//5
    }()
    return
}

最终函数 f0 的返回值为：5。

三、思维试验

func f1() (r int) {
    defer func(r int) {
        fmt.Printf("f1 defer r is:%d", r)//0
        r = r + 5
        fmt.Printf("f1 defer2 r is:%d", r)//5
    }(r)
    fmt.Printf("f1 r is:%d", r)//0
    return 1
}

函数返回结果：1。（defer 三原则、值传递、return xxx 拆解）

func f2() (r int) {
    defer func() {
        fmt.Printf("f2 defer r is:%d", r)//1
        r = r + 5
        fmt.Printf("f2 defer2 r is:%d", r)//6
    }()
    fmt.Printf("f2 r is:%d", r)//0
    return 1
}

函数返回结果为：6。（refer 三原则、return xxx 拆解）

参考资料

Defer, Panic, and Recover

Defer statements