Defer的声明会把函数的调用放到一个执行的列表上。当调用这个defer函数的函数返回时,这个执行列表上的函数会被调用。Defer一般被用来做一些函数调用后的一些清理工作。
下面两段代码我们来看看区别:
func CopyFile(dstName, srcName string) (written int64, err error) {
src, err := os.Open(srcName)
if err != nil {
return //这里退出时src并没有关闭,会导致问题
}
dst, err := os.Create(dstName)
if err != nil {
return //同上
}
written, err = io.Copy(dst, src)
dst.Close()
src.Close()
return
}
修改如下:
func CopyFile(dstName, srcName string) (written int64, err error) { src, err := os.Open(srcName) if err != nil { return } defer src.Close() //当return时会调用src.Close(),就会把打开的文件关闭,从而避免文件打开而没有被关闭的问题 dst, err := os.Create(dstName) if err != nil { return } defer dst.Close() //同上 return io.Copy(dst, src) }
使用defer的三个规则:
1. 一个延迟执行的函数的变量的值在声明延迟时别赋值
func a() {
i := 0
defer fmt.Println(i) // return后会打印“0”
i++
return
}
2. 被延迟的函数会依据后进先出(Last In First Out)的方式执行
func b() {
for i := 0; i < 4; i++ {
defer fmt.Print(i) // 打印 “3210”
}
}
3. 延迟的函数可以读取和赋值返回函数的值
func c() (i int) {
defer func() { i++ }() // 当return 1后, i会读到1并加1, 因此延迟函数的值为2
return 1
}
我在做Go Tour#69的时候,一直纠结于如何确定某个chan是否已经结束/关闭,而不会导致整个程序无限循环。有了defer就可以很优雅的解决这个问题
package main
import (
"fmt"
)
type Tree struct {
Left *Tree
Value int
Right *Tree
}
func Walk(t *Tree, ch chan int) {
defer close(ch) // 当这个嵌套函数返回后执行
var walk func(t *Tree)
walk = func(t *Tree) {
if t == nil {
return
}
walk(t.Left)
ch <- t.Value
walk(t.Right)
}
walk(t)
}
func Same(t1, t2 *Tree) bool {
return false
}
func main() {
t := &Tree{nil, 1, &Tree{nil, 2, nil}}
ch := make(chan int)
go Walk(t, ch)
for {
v, ok := <-ch
if !ok {
return // ch关闭,从而推出循环
}
fmt.Println(v)
}
}
参考:
1. http://blog.golang.org/2010/08/defer-panic-and-recover.html