Go:defer关键字

        在Go语言中，defer语句用于延迟执行一个函数或方法，直到包含它的函数执行完毕时才执行。defer通常用于资源清理、文件关闭、锁释放等操作，确保这些操作即使在函数发生错误时也能得到执行。

一 defer的执行顺序

        defer语句将函数调用推迟到当前函数返回之后执行，无论函数是正常执行完毕还是发生了错误。

func main() {
	fmt.Println("Start")

	// 推迟执行的函数
	defer fmt.Println("This is deferred")

	fmt.Println("End")
}

输出结果：

        我们可以将defer理解为一个压栈操作，因此如果存在多个defer语句，那么它们的执行顺序是先进后出的，也就是排在前面的defer语句会在最后执行。

        代码示例如下： 

package main

import (
	"fmt"
)

func main() {
	defer fmt.Println("First deferred")
	defer fmt.Println("Second deferred")
	defer fmt.Println("Third deferred")

	fmt.Println("Main function")
}

        那么，如果defer关键字修饰的语句遇到了return，两者的执行顺序又是如何呢？ 我们不妨猜一下，如果defer语句是将函数调用推迟到当前函数返回之后执行，那么一定会在函数的生命周期结束后，才轮到defer语句，我们可以代码验证一下这个猜想：

func fun1() int {
	fmt.Println("defer function called ...")
	return 0
}
func fun2() int {
	fmt.Println("return function called ...")
	return 0
}
func fun3() int {
	defer fun1()
	return fun2()
}
func main() {
	fun3()
}

        运行结果如下：

 

        需要注意的是，虽然defer语句会在return语句执行之后，但在实际返回之前执行。

package main

import (
	"fmt"
)

func testDefer() (result int) {
	defer func() {
		result++
	}()
	return 10
}

func main() {
	fmt.Println("Result:", testDefer()) // 输出：Result: 11
}

        在这个例子中，testDefer函数返回10，但由于defer语句在返回前已经执行了，因此 result值增加了1，最终返回11。 

二 defer与参数评估

        defer语句通常与匿名函数结合使用，以便将需要推迟执行的代码块放入其中。这样可以在defer中传递参数或处理上下文。例如：

package main

import (
	"fmt"
)

func main() {
	x, y := 1, 2
	defer func(a, b int) {
		fmt.Printf("Deferred: %d + %d = %d\n", a, b, a+b)
	}(x, y)

	x, y = 3, 4
	fmt.Printf("Main: %d + %d = %d\n", x, y, x+y)
}

        在该例子中，defer语句使用了匿名函数，并在调用时传递了当前的x和y值（1和2）。由于defer在声明时就评估参数，因此即使x和y后来发生了变化，defer中的函数仍使用最初的参数值，因此最后输出的结果为Main: 3 + 4 = 7 Deferred: 1 + 2 = 3。

三 defer和panic

        在Go语言中，defer和panic常常一起使用，用于实现更稳健的错误处理机制。panic用于引发异常，defer则用于确保在异常发生时执行必要的清理操作或恢复（recover）程序的正常运行。

        当函数调用panic时，当前函数的正常执行流会停止，并开始执行已经注册的defer语句。defer在这种情况下可以用来进行资源清理、日志记录，或者使用recover来捕获panic并恢复程序。简单来说，在panic语句后面的defer语句不被执行，在panic语句前的defer语句会被执行。

package main

import (
	"fmt"
)

func main() {
	fmt.Println("Start of main")

	// 使用defer来捕获panic
	defer func() {
		if r := recover(); r != nil {
			fmt.Println("Recovered from panic:", r)
		}
	}()

	// 引发panic
	panic("Something went wrong!")

	fmt.Println("End of main") // 这行代码不会执行
}

        输出结果：

Start of main
Recovered from panic: Something went wrong!

四 defer的使用场景

        defer通常用于在操作完文件后关闭文件，这样即使在函数中途发生错误，文件也能保证被正确关闭。

unc ReadFile(filename string) ([]byte, error) {
    f, err := os.Open(filename)
    if err != nil {
        return nil, err
    }
    defer f.close()
    return ReadAll()
}

        在使用互斥锁（sync.Mutex）时，defer也可以确保在临界区代码执行完毕后自动释放锁。

var mu sync.Mutex
var m = make(map[string]int)
 
func lookup(key string) int {
    mu.Lock()
    defer mu.Unlock()
    return m[key]
}