聽說,99% 的 Go 程序員都被 defer 坑過
先聲明:我被坑過。
之前寫 Go 專欄時,寫過一篇文章:Go 專欄|錯誤處理:defer,panic 和 recover。有小伙伴留言說:道理都懂,但還是不知道怎么用,而且還總出現(xiàn)莫名奇妙的問題。
出問題就對了,這個小東西壞的很,一不留神就出錯。
所以,面對這種情況,我們今天就不講道理了。直接把我珍藏多年的代碼一把梭,憑借多年踩坑經(jīng)歷和寫 BUG 經(jīng)驗(yàn),我要站著把這個坑邁過去。
一、
先來一個簡單的例子熱熱身:
package?main
import?(
????"fmt"
)
func?main()?{
????defer?func()?{
????????fmt.Println("first")
????}()
????defer?func()?{
????????fmt.Println("second")
????}()
????fmt.Println("done")
}
輸出:
done
second
first
這個比較簡單,defer 語句的執(zhí)行順序是按調(diào)用 defer 語句的倒序執(zhí)行。
二、
看看這段代碼有什么問題?
for?_,?filename?:=?range?filenames?{
????f,?err?:=?os.Open(filename)
????if?err?!=?nil?{
????????return?err
????}
????defer?f.Close()
}
這段代碼其實(shí)很危險,很可能會用盡所有文件描述符。因?yàn)?defer 語句不到函數(shù)的最后一刻是不會執(zhí)行的,也就是說文件始終得不到關(guān)閉。所以切記,一定不要在 for 循環(huán)中使用 defer 語句。
那怎么優(yōu)化呢?可以將循環(huán)體單獨(dú)寫一個函數(shù),這樣每次循環(huán)的時候都會調(diào)用關(guān)閉函數(shù)。
如下:
for?_,?filename?:=?range?filenames?{
????if?err?:=?doFile(filename);?err?!=?nil?{
????????return?err
????}
}
func?doFile(filename?string)?error?{
????f,?err?:=?os.Open(filename)
????if?err?!=?nil?{
????????return?err
????}
????defer?f.Close()
}
三、
看看這三個函數(shù)的輸出結(jié)果是什么?
package?main
import?(
????"fmt"
)
func?a()?(r?int)?{
????defer?func()?{
????????r++
????}()
????return?0
}
func?b()?(r?int)?{
????t?:=?5
????defer?func()?{
????????t?=?t?+?5
????}()
????return?t
}
func?c()?(r?int)?{
????defer?func(r?int)?{
????????r?=?r?+?5
????}(r)
????return?1
}
func?main()?{
????fmt.Println("a?=?",?a())
????fmt.Println("b?=?",?b())
????fmt.Println("c?=?",?c())
}
公布答案:
a?=??1
b?=??5
c?=??1
你答對了嗎?
說實(shí)話剛開始看到這個結(jié)果時,我是相當(dāng)費(fèi)解,完全不知道怎么回事。
但可以看到,這三個函數(shù)都有一個共同特點(diǎn),它們都有一個命名返回值,并且都在函數(shù)中引用了這個返回值。
引用的方式分兩種:分別是閉包和函數(shù)參數(shù)。
先看 a() 函數(shù):
閉包通過 r++ 修改了外部變量,返回值變成了 1。
相當(dāng)于:
func?aa()?(r?int)?{
????r?=?0
????//?在?return?之前,執(zhí)行?defer?函數(shù)
????func()?{
????????r++
????}()
????return
}
再看 b() 函數(shù):
閉包內(nèi)修改的只是局部變量 t,而外部變量 t 不受影響,所以還是返回 5。
相當(dāng)于:
func?bb()?(r?int)?{
????t?:=?5
????//?賦值
????r?=?t
????//?在?return?之前,執(zhí)行?defer?函數(shù)
????//?defer?函數(shù)沒有對返回值?r?進(jìn)行修改,只是修改了變量?t
????func()?{
????????t?=?t?+?5
????}()
????return
}
最后是 c 函數(shù):
參數(shù)傳遞是值拷貝,實(shí)參不受影響,所以還是返回 1。
相當(dāng)于:
func?cc()?(r?int)?{
????//?賦值
????r?=?1
????//?這里修改的?r?是函數(shù)形參的值
????//?值拷貝,不影響實(shí)參值
????func(r?int)?{
????????r?=?r?+?5
????}(r)
????return
}
那么,為了避免寫出這么令人意外的代碼,最好在定義函數(shù)時就不要使用命名返回值?;蛘呷绻褂昧?,就不要在 defer 中引用。
再看下面兩個例子:
func?d()?int?{
????r?:=?0
????defer?func()?{
????????r++
????}()
????return?r
}
func?e()?int?{
????r?:=?0
????defer?func(i?int)?{
????????i++
????}(r)
????return?0
}
d?=??0
e?=??0
返回值符合預(yù)期,再也不用絞盡腦汁猜了。
四、
defer 表達(dá)式的函數(shù)如果在 panic 后面,則這個函數(shù)無法被執(zhí)行。
func?main()?{
????panic("a")
????defer?func()?{
????????fmt.Println("b")
????}()
}
輸出如下,b 沒有打印出來。
panic:?a
goroutine?1?[running]:
main.main()
????xxx.go:87?+0x4ce
exit?status?2
而如果 defer 在前,則可以執(zhí)行。
func?main()?{
????defer?func()?{
????????fmt.Println("b")
????}()
????panic("a")
}
輸出:
b
panic:?a
goroutine?1?[running]:
main.main()
????xxx.go:90?+0x4e7
exit?status?2
五、
看看下面這段代碼的執(zhí)行順序:
func?G()?{
????defer?func()?{
????????fmt.Println("c")
????}()
????F()
????fmt.Println("繼續(xù)執(zhí)行")
}
func?F()?{
????defer?func()?{
????????if?err?:=?recover();?err?!=?nil?{
????????????fmt.Println("捕獲異常:",?err)
????????}
????????fmt.Println("b")
????}()
????panic("a")
}
func?main()?{
????G()
}
順序如下:
調(diào)用
G()函數(shù);調(diào)用
F()函數(shù);F()中遇到panic,立刻終止,不執(zhí)行panic之后的代碼;執(zhí)行
F()中defer函數(shù),遇到recover捕獲錯誤,繼續(xù)執(zhí)行defer中代碼,然后返回;執(zhí)行
G()函數(shù)后續(xù)代碼,最后執(zhí)行G()中defer函數(shù)。
輸出:
捕獲異常:?a
b
繼續(xù)執(zhí)行
c
五、
看看下面這段代碼的執(zhí)行順序:
func?G()?{
????defer?func()?{
????????if?err?:=?recover();?err?!=?nil?{
????????????fmt.Println("捕獲異常:",?err)
????????}
????????fmt.Println("c")
????}()
????F()
????fmt.Println("繼續(xù)執(zhí)行")
}
func?F()?{
????defer?func()?{
????????fmt.Println("b")
????}()
????panic("a")
}
func?main()?{
????G()
}
順序如下:
調(diào)用
G()函數(shù);調(diào)用
F()函數(shù);F()中遇到panic,立刻終止,不執(zhí)行panic之后的代碼;執(zhí)行
F()中defer函數(shù),由于沒有recover,則將panic拋到G()中;G()收到panic則不會執(zhí)行后續(xù)代碼,直接執(zhí)行defer函數(shù);defer中捕獲F()拋出的異常a,然后繼續(xù)執(zhí)行,最后退出。
輸出:
b
捕獲異常:?a
c
六、
看看下面這段代碼的執(zhí)行順序:
func?G()?{
????defer?func()?{
????????fmt.Println("c")
????}()
????F()
????fmt.Println("繼續(xù)執(zhí)行")
}
func?F()?{
????defer?func()?{
????????fmt.Println("b")
????}()
????panic("a")
}
func?main()?{
????G()
}
順序如下:
調(diào)用
G()函數(shù);調(diào)用
F()函數(shù);F()中遇到panic,立刻終止,不執(zhí)行panic之后的代碼;執(zhí)行
F()中defer函數(shù),由于沒有recover,則將panic拋到G()中;G()收到panic則不會執(zhí)行后續(xù)代碼,直接執(zhí)行defer函數(shù);由于沒有
recover,直接拋出F()拋過來的異常a,然后退出。
輸出:
b
c
panic:?a
goroutine?1?[running]:
main.F()
????xxx.go:90?+0x5b
main.G()
????xxx.go:82?+0x48
main.main()
????xxx.go:107?+0x4a5
exit?status?2
七、
看看下面這段代碼的執(zhí)行順序:
func?G()?{
????defer?func()?{
????????//?goroutine?外進(jìn)行?recover
????????if?err?:=?recover();?err?!=?nil?{
????????????fmt.Println("捕獲異常:",?err)
????????}
????????fmt.Println("c")
????}()
????//?創(chuàng)建?goroutine?調(diào)用?F?函數(shù)
????go?F()
????time.Sleep(time.Second)
}
func?F()?{
????defer?func()?{
????????fmt.Println("b")
????}()
????//?goroutine?內(nèi)部拋出panic
????panic("a")
}
func?main()?{
????G()
}
順序如下:
調(diào)用
G()函數(shù);通過 goroutine 調(diào)用
F()函數(shù);F()中遇到panic,立刻終止,不執(zhí)行panic之后的代碼;執(zhí)行
F()中defer函數(shù),由于沒有recover,則將panic拋到G()中;由于 goroutine 內(nèi)部沒有進(jìn)行
recover,則 goroutine 外部函數(shù),也就是G()函數(shù)是沒辦法捕獲的,程序直接崩潰退出。
輸出:
b
panic:?a
goroutine?6?[running]:
main.F()
????xxx.go:96?+0x5b
created?by?main.G
????xxx.go:87?+0x57
exit?status?2
八、
最后再說一個 recover 的返回值問題:
defer?func()?{
????if?err?:=?recover();?err?!=?nil?{
????????fmt.Println("捕獲異常:",?err.Error())
????}
}()
panic("a")
recover 返回的是 interface {} 類型,而不是 error 類型,所以這樣使用的話會報(bào)錯:
err.Error?undefined?(type?interface?{}?is?interface?with?no?methods)
可以這樣來轉(zhuǎn)換一下:
defer?func()?{
????if?err?:=?recover();?err?!=?nil?{
????????fmt.Println("捕獲異常:",?fmt.Errorf("%v",?err).Error())
????}
}()
panic("a")
或者直接打印結(jié)果:
defer?func()?{
????if?err?:=?recover();?err?!=?nil?{
????????fmt.Println("捕獲異常:",?err)
????}
}()
panic("a")
輸出:
捕獲異常:?a
以上就是本文的全部內(nèi)容,其實(shí)寫過其他的語言的同學(xué)都知道,關(guān)閉文件句柄,釋放鎖等操作是很容易忘的。而 Go 語言通過 defer 很好地解決了這個問題,但在使用過程中還是要小心。
本文總結(jié)了一些容踩坑的點(diǎn),希望能夠幫助大家少寫 BUG,如果大家覺得有用的話,歡迎點(diǎn)贊和轉(zhuǎn)發(fā)。
文章中的腦圖和源碼都上傳到了 GitHub,有需要的同學(xué)可自行下載。
源碼地址:
https://github.com/yongxinz/gopher/tree/main/sc
關(guān)注公眾號 AlwaysBeta,回復(fù)「goebook」領(lǐng)取 Go 編程經(jīng)典書籍。
推薦閱讀:
參考:
《Go 語言核心編程》
https://www.jianshu.com/p/63e3d57f285f
題圖: 該圖片由 Raheel Shakeel 在 Pixabay 上發(fā)布
