Defer 用于确保函数调用在程序执行的后期执行,通常用于清理目的。defer 通常用于其他语言中使用 ensure 和 finally 的地方。
|
|
|
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)
}
}
|