Go语言之城 - 程序示例: Defer

Defer 用于确保程序在执行完成后，会调用某个函数，一般是执行清理工作。 Defer 的用途跟其他语言的 ensure 或 finally 类似。

 
package main

import (
    "fmt"
    "os"
)

假设我们想要创建一个文件、然后写入数据、最后在程序结束时关闭该文件。 这里展示了如何通过 defer 来做到这一切。

 
func main() {

在 createFile 后立即得到一个文件对象， 我们使用 defer 通过 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")

}

关闭文件时，进行错误检查是非常重要的， 即使在 defer 函数中也是如此。

 
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

Go语言之城 | 闽ICP备20003902号-3