Go 示例: Defer

Defer 用于确保函数调用在程序执行的后期执行,通常用于清理目的。defer 通常用于其他语言中使用 ensurefinally 的地方。

package main
import (
    "fmt"
    "os"
)

假设我们想创建一个文件,写入文件,然后在完成时关闭它。以下是使用 defer 的方法。

func main() {

在使用 createFile 获取文件对象后,我们使用 closeFile 推迟关闭该文件。这将在封闭函数 (main) 结束时执行,在 writeFile 完成后执行。

    f := createFile("/tmp/defer.txt")
    defer closeFile(f)
    writeFile(f)
}
func createFile(p string) *os.File {
    fmt.Println("creating")
    f, err := os.Create(p)
    if err != nil {
        panic(err)
    }
    return f
}
func writeFile(f *os.File) {
    fmt.Println("writing")
    fmt.Fprintln(f, "data")
}

在关闭文件时检查错误非常重要,即使是在延迟函数中也是如此。

func closeFile(f *os.File) {
    fmt.Println("closing")
    err := f.Close()
    if err != nil {
        fmt.Fprintf(os.Stderr, "error: %v\n", err)
        os.Exit(1)
    }
}

运行程序确认文件在写入后已关闭。

$ go run defer.go
creating
writing
closing

下一个示例:Recover.