Go语言defer语句的三种机制整理

Go语言中的defer语句用于在函数退出时执行一些操作，可以是函数调用、变量赋值等。defer语句的执行时机有三种机制，下面对这三种机制进行详细介绍。

1. 栈（Stack）机制

栈机制是defer语句最常见的执行机制。在函数中调用多个defer语句时，defer语句的执行顺序符合后进先出的栈结构规律。具体来说，当函数执行defer语句时，会把defer语句压入函数栈中，并在函数返回时按照LIFO（后进先出）的顺序依次执行这些defer语句。示例如下：

func foo() {
    defer fmt.Println("defer 1")
    defer fmt.Println("defer 2")
    fmt.Println("foo")
}

foo()

输出结果为：

foo
defer 2
defer 1

2. Copy机制

如果在函数中调用了一个期望语句无副作用的函数，为避免出现因为复制多次执行该函数而对程序产生影响的问题，Go语言使用copy机制来宏观地避免这种副作用的发生。也就是说，如果一个函数中调用了一个期望语句无副作用的函数，那么在函数返回时，会将该函数所产生的 defer 语句文本复制到一个新的位置，以确保每次执行该函数时，都是按照最近一次调用该函数时的defer语句顺序执行的。示例如下：

func demo() {
    fmt.Println("start")
    defer fmt.Println("defer 1")
    defer fmt.Println("defer 2")
    fmt.Println("end")
}

for i := 0; i < 3; i++ {
    demo()
}

输出结果为：

start
end
defer 2
defer 1
start
end
defer 2
defer 1
start
end
defer 2
defer 1

3. Return语句机制

在函数执行到return语句时，Go语言会进行两个步骤：首先，计算出所有的defer语句，按照栈的方式依次压入函数栈中，然后，执行函数的返回操作。在执行返回操作时，会按照常规顺序执行之前压入栈中defer语句。下面的示例代码演示了这种机制：

func deferDemo() (result int) {
    defer func() {
        result++
    }()
    return 0
}

fmt.Println(deferDemo()) // 输出 1

这个示例中，当函数执行到 return 0 语句时，展开的代码是 result=0; return。这时候会计算出defer语句result++，并将其压入函数栈中，然后执行函数返回操作，相当于执行result=0; result++；最后返回result的值为1。