Go語言中使用Defer幾個場景
關于 defer 的詳細介紹請參考: Defer, Panic, and Recover .
C++ 中模擬的 defer 實現請參考: C++版的defer語句
1. 簡化資源的回收
這是最常見的 defer 用法. 比如:
- mu.Lock()
- defer mu.Unlock()
當然, defer 也有一定的開銷, 也有為了節省性能而回避使用的 defer 的:
- mu.Lock()
- count++
- mu.Unlock()
從簡化資源的釋放角度看, defer 類似一個語法糖, 好像不是必須的.
2. panic異常的捕獲
defer 除了用于簡化資源的釋放外, 還是Go語言異常框架的一個組成部分.
Go語言中, panic用于拋出異常, recover用于捕獲異常. recover只能在defer語句中使用, 直接調用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語言中函數的異常, 就離不開defer語句了.
3. 修改返回值
defer 除了用于配合 recover, 用于捕獲 panic 異常外, 還可以用于在 return 之后修改函數的返回值.
比如:
- func doubleSum(a, b int) (sum int) {
- defer func() {
- sum *= 2
- }()
- sum = a + b
- }
當然, 這個特性應該只是 defer 的副作用, 具體在什么場景使用就要由開發者自己決定了.
4. 安全的回收資源
前面第一點提到, defer 最常見的用法是簡化資源的回收. 而且, 從資源回收角度看, defer 只是一個語法糖.
其實, 也不完全是這樣, 特別是在涉及到第二點提到的panic異常等因素導致goroutine提前退出時.
比如, 有一個線程安全的slice修改函數, 為了性能沒有使用defer語句:
- func set(mu *sync.Mutex, arr []int, i, v int) {
- mu.Lock()
- arr[i] = v
- mu.Unlock()
- }
但是, 如果 i >= len(arr)的話, runtime就會拋出切片越界的異常(這里只是舉例, 實際開發中不應該出現切片越界異常). 這樣的話, mu.Unlock() 就沒有機會被執行了.
如果用defer的話, 即使出現異常也能保證mu.Unlock()被調用:
- func set(mu *sync.Mutex, arr []int, i, v int) {
- mu.Lock()
- defer mu.Unlock()
- arr[i] = v
- }
當然, Go語言約定異常不會跨越package邊界. 因此, 調用一般函數的時候不用擔心goroutine異常退出的情況.
不過對于一些比較特殊的package, 比如go test依賴的testing包, 包中的t.Fatal就是依賴了Go中類似異常的特性(準確的說是調用了runtime.Goexit()).
比如有以下的測試函數(詳情請參考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()
- }
當測試失敗的時候, wg.Done()將沒有機會執行, 最終導致wg.Wait()死鎖.
對于這個例子, 安全的做法是使用defer語句保證wg.Done()始終會被執行.
