Go语言之defer（原理、常见的坑）

defer 是 Go 语言的一个关键字，它用于延迟函数的执行。这个功能非常有用，可以用来处理资源清理、错误处理和确保一些操作的执行。

1. defer 的使用

package main
 
import "fmt"
 
func main() {
    fmt.Println("Hello, ")
    for i := 0; i < 5; i++ {
        defer fmt.Print(i, " ")
    }
    fmt.Println("World!")
}

在这个例子中，"Hello, World!" 会首先被打印。然后，defer 关键字会将后面的代码推迟执行，直到包含它的函数执行完毕。因此，在 "Hello, World!" 被打印后，defer 推迟的代码开始执行，输出的结果是："Hello, World! 4 3 2 1 0 "。

2. defer 的常见陷阱

虽然 defer 非常有用，但是也有一些常见的陷阱，需要注意。

• 如果 defer 表达式中包含匿名函数，那么这个匿名函数会在 defer 的执行时点进行调用，而不是在 defer 语句被解析时。

func main() {
    for i := 0; i < 5; i++ {
        defer func() {
            fmt.Print(i, " ")
        }()
    }
}

在这个例子中，输出的结果是 "4 4 4 4 4 "，而不是预期的 "0 1 2 3 4 "。因为在 defer 执行时，循环变量 i 已经变成了 5。

• 如果 defer 表达式中引用了包含函数的变量（非局部变量），那么这些变量在 defer 执行时会保持最终状态。

func main() {
    var i int
    for i = 0; i < 5; i++ {
        defer fmt.Print(i, " ")
    }
}

在这个例子中，输出的结果是 "4 3 2 1 0 "，而不是预期的 "0 1 2 3 4 "。因为在 defer 执行时，变量 i 的值已经变成了 5。

3. 如何避免这些陷阱

为了避免这些陷阱，可以采用以下策略：

• 如果需要在 defer 中使用循环变量，可以通过函数封装和闭包来解决。

func main() {
    for i := 0; i < 5; i++ {
        func(i int) {
            defer fmt.Print(i, " ")
        }(i)
    }
}

• 如果需要在 defer 中引用包含函数的变量，可以通过使用局部变量和每次循环创建新的 defer 来解决。

func main() {
    for i := 0; i < 5; i++ {
        i := i
        defer fmt.Print(i, " ")
    }
}

以上就是 defer 的一些基本使用和常见的陷阱。在实际编程中，合理使用 defer 可以使代码变得简洁而且易于维护，但也需要注意避免引入不必要的问题。