Defer

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")

}

func closeFile(f *os.File) {
	fmt.Println("closing")
	err := f.Close()
	// 关闭文件时,进行错误检查是非常重要的,
	// 即使在 defer 函数中也是如此。
	if err != nil {
		fmt.Fprintf(os.Stderr, "error: %v\n", err)
		os.Exit(1)
	}
}
# 执行程序,确认写入数据后,文件已关闭。
go run defer.go
creating
writing
closing