defer in Go

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