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

C++ 中模擬的 defer 實(shí)現(xiàn)請(qǐng)參考: C++版的defer語(yǔ)句

1. 簡(jiǎn)化資源的回收

這是最常見(jiàn)的 defer 用法. 比如:

mu.Lock()  
defer mu.Unlock() 

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

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

從簡(jiǎn)化資源的釋放角度看, defer 類(lèi)似一個(gè)語(yǔ)法糖, 好像不是必須的.

2. panic異常的捕獲

defer 除了用于簡(jiǎn)化資源的釋放外, 還是Go語(yǔ)言異常框架的一個(gè)組成部分.

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

比如:

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語(yǔ)言中函數(shù)的異常, 就離不開(kāi)defer語(yǔ)句了.

3. 修改返回值

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

比如:

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

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

4. 安全的回收資源

前面第一點(diǎn)提到, defer 最常見(jiàn)的用法是簡(jiǎn)化資源的回收. 而且, 從資源回收角度看, defer 只是一個(gè)語(yǔ)法糖.

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

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

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

但是, 如果 i >= len(arr)的話, runtime就會(huì)拋出切片越界的異常(這里只是舉例, 實(shí)際開(kāi)發(fā)中不應(yīng)該出現(xiàn)切片越界異常). 這樣的話, mu.Unlock() 就沒(méi)有機(jī)會(huì)被執(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語(yǔ)言約定異常不會(huì)跨越package邊界. 因此, 調(diào)用一般函數(shù)的時(shí)候不用擔(dān)心goroutine異常退出的情況.

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

比如有以下的測(cè)試函數(shù)(詳情請(qǐng)參考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)測(cè)試失敗的時(shí)候, wg.Done()將沒(méi)有機(jī)會(huì)執(zhí)行, 最終導(dǎo)致wg.Wait()死鎖.

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

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