關(guān)于 defer 的詳細(xì)介紹請參考: Defer, Panic, and Recover .

C++ 中模擬的 defer 實現(xiàn)請參考: C++版的defer語句

1. 簡化資源的回收

這是最常見的 defer 用法. 比如:

mu.Lock()  
defer mu.Unlock() 

當(dāng)然, defer 也有一定的開銷, 也有為了節(jié)省性能而回避使用的 defer 的:

mu.Lock()  
count++  
mu.Unlock() 

從簡化資源的釋放角度看, defer 類似一個語法糖, 好像不是必須的.

2. panic異常的捕獲

defer 除了用于簡化資源的釋放外, 還是Go語言異?？蚣艿囊粋€組成部分.

Go語言中, panic用于拋出異常, recover用于捕獲異常. recover只能在defer語句中使用, 直接調(diào)用recover是無效的.

比如:

func main() {  
    f()  
    fmt.Println("Returned normally from f.")  
}  
 
func f() {  
    defer func() {  
        if r := recover(); r != nil {  
            fmt.Println("Recovered in f", r)  
        }  
    }()  
    fmt.Println("Calling g.")  
    g()  
    fmt.Println("Returned normally from g.")  
}  
 
func g() {  
    panic("ERROR")  
} 

因此, 如果要捕獲Go語言中函數(shù)的異常, 就離不開defer語句了.

3. 修改返回值

defer 除了用于配合 recover, 用于捕獲 panic 異常外, 還可以用于在 return 之后修改函數(shù)的返回值.

比如:

func doubleSum(a, b int) (sum int) {  
    defer func() {  
        sum *= 2 
    }()  
    sum = a + b  
} 

當(dāng)然, 這個特性應(yīng)該只是 defer 的副作用, 具體在什么場景使用就要由開發(fā)者自己決定了.

4. 安全的回收資源

前面第一點提到, defer 最常見的用法是簡化資源的回收. 而且, 從資源回收角度看, defer 只是一個語法糖.

其實, 也不完全是這樣, 特別是在涉及到第二點提到的panic異常等因素導(dǎo)致goroutine提前退出時.

比如, 有一個線程安全的slice修改函數(shù), 為了性能沒有使用defer語句:

func set(mu *sync.Mutex, arr []int, i, v int) {  
    mu.Lock()  
    arr[i] = v  
    mu.Unlock()  
} 

但是, 如果 i >= len(arr)的話, runtime就會拋出切片越界的異常(這里只是舉例, 實際開發(fā)中不應(yīng)該出現(xiàn)切片越界異常). 這樣的話, mu.Unlock() 就沒有機會被執(zhí)行了.

如果用defer的話, 即使出現(xiàn)異常也能保證mu.Unlock()被調(diào)用:

func set(mu *sync.Mutex, arr []int, i, v int) {  
    mu.Lock()  
    defer mu.Unlock()  
    arr[i] = v  
} 

當(dāng)然, Go語言約定異常不會跨越package邊界. 因此, 調(diào)用一般函數(shù)的時候不用擔(dān)心goroutine異常退出的情況.

不過對于一些比較特殊的package, 比如go test依賴的testing包, 包中的t.Fatal就是依賴了Go中類似異常的特性(準(zhǔn)確的說是調(diào)用了runtime.Goexit()).

比如有以下的測試函數(shù)(詳情請參考Issue5746):

func TestFailed(t *testing.T) {  
    var wg sync.WaitGroup  
    for i := 0; i < 2; i++ {  
        wg.Add(1)  
        go func(id int) {  
            // defer wg.Done()  
            t.Fatalf("TestFailed: id = %v\n", id)  
            wg.Done()  
        }(i)  
    }  
    wg.Wait()  
} 

當(dāng)測試失敗的時候, wg.Done()將沒有機會執(zhí)行, 最終導(dǎo)致wg.Wait()死鎖.

對于這個例子, 安全的做法是使用defer語句保證wg.Done()始終會被執(zhí)行.

原文鏈接：http://my.oschina.net/chai2010/blog/140065