Go程序員踩過的defer坑錯誤處理
前言
先聲明:我被坑過。
之前寫 Go 專欄時,寫過一篇文章:Go 專欄|錯誤處理:defer,panic 和 recover。有小伙伴留言說:道理都懂,但還是不知道怎么用,而且還總出現莫名奇妙的問題。
出問題就對了,這個小東西壞的很,一不留神就出錯。
所以,面對這種情況,我們今天就不講道理了。直接把我珍藏多年的代碼一把梭,憑借多年踩坑經歷和寫 BUG 經驗,我要站著把這個坑邁過去。
一、簡單的例子
先來一個簡單的例子熱熱身:
package?main import?( ????"fmt" ) func?main()?{ ????defer?func()?{ ????????fmt.Println("first") ????}() ????defer?func()?{ ????????fmt.Println("second") ????}() ????fmt.Println("done") }
輸出:
done
second
first
這個比較簡單,defer 語句的執(zhí)行順序是按調用 defer 語句的倒序執(zhí)行。
二、一定不要在 for 循環(huán)中使用 defer 語句
看看這段代碼有什么問題?
for?_,?filename?:=?range?filenames?{ ????f,?err?:=?os.Open(filename) ????if?err?!=?nil?{ ????????return?err ????} ????defer?f.Close() }
這段代碼其實很危險,很可能會用盡所有文件描述符。因為 defer 語句不到函數的最后一刻是不會執(zhí)行的,也就是說文件始終得不到關閉。所以切記,一定不要在 for 循環(huán)中使用 defer 語句。
那怎么優(yōu)化呢?可以將循環(huán)體單獨寫一個函數,這樣每次循環(huán)的時候都會調用關閉函數。
如下:
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() }
三、定義函數時就不要使用命名返回值
看看這三個函數的輸出結果是什么?
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
你答對了嗎?
說實話剛開始看到這個結果時,我是相當費解,完全不知道怎么回事。
但可以看到,這三個函數都有一個共同特點,它們都有一個命名返回值,并且都在函數中引用了這個返回值。
引用的方式分兩種:分別是閉包和函數參數。
先看 a() 函數:
閉包通過 r++ 修改了外部變量,返回值變成了 1。
相當于:
func?aa()?(r?int)?{ ????r?=?0 ????//?在?return?之前,執(zhí)行?defer?函數 ????func()?{ ????????r++ ????}() ????return }
再看 b() 函數:
閉包內修改的只是局部變量 t,而外部變量 t 不受影響,所以還是返回 5。
相當于:
func?bb()?(r?int)?{ ????t?:=?5 ????//?賦值 ????r?=?t ????//?在?return?之前,執(zhí)行?defer?函數 ????//?defer?函數沒有對返回值?r?進行修改,只是修改了變量?t ????func()?{ ????????t?=?t?+?5 ????}() ????return }
最后是 c 函數:
參數傳遞是值拷貝,實參不受影響,所以還是返回 1。
相當于:
func?cc()?(r?int)?{ ????//?賦值 ????r?=?1 ????//?這里修改的?r?是函數形參的值 ????//?值拷貝,不影響實參值 ????func(r?int)?{ ????????r?=?r?+?5 ????}(r) ????return }
那么,為了避免寫出這么令人意外的代碼,最好在定義函數時就不要使用命名返回值。或者如果使用了,就不要在 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
返回值符合預期,再也不用絞盡腦汁猜了。
四、defer 表達式的函數如果在 panic 后面,則這個函數無法被執(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í)行順序
看看下面這段代碼的執(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() }
順序如下:
- 調用 G() 函數;
- 調用 F() 函數;
- F() 中遇到 panic,立刻終止,不執(zhí)行 panic 之后的代碼;
- 執(zhí)行 F() 中 defer 函數,遇到 recover 捕獲錯誤,繼續(xù)執(zhí)行 defer 中代碼,然后返回;
- 執(zhí)行 G() 函數后續(xù)代碼,最后執(zhí)行 G() 中 defer 函數。
輸出:
捕獲異常: a
b
繼續(xù)執(zhí)行
c
五、捕獲異常執(zhí)行順序
看看下面這段代碼的執(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() }
順序如下:
- 調用 G() 函數;
- 調用 F() 函數;
- F() 中遇到 panic,立刻終止,不執(zhí)行 panic 之后的代碼;
- 執(zhí)行 F() 中 defer 函數,由于沒有 recover,則將 panic 拋到 G() 中;
- G() 收到 panic 則不會執(zhí)行后續(xù)代碼,直接執(zhí)行 defer 函數;
- defer 中捕獲 F() 拋出的異常 a,然后繼續(xù)執(zhí)行,最后退出。
輸出:
b
捕獲異常: a
c
六、函數執(zhí)行順序
看看下面這段代碼的執(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() }
順序如下:
- 調用 G() 函數;
- 調用 F() 函數;
- F() 中遇到 panic,立刻終止,不執(zhí)行 panic 之后的代碼;
- 執(zhí)行 F() 中 defer 函數,由于沒有 recover,則將 panic 拋到 G() 中;
- G() 收到 panic 則不會執(zhí)行后續(xù)代碼,直接執(zhí)行 defer 函數;
- 由于沒有 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í)行順序
看看下面這段代碼的執(zhí)行順序:
func?G()?{ ????defer?func()?{ ????????//?goroutine?外進行?recover ????????if?err?:=?recover();?err?!=?nil?{ ????????????fmt.Println("捕獲異常:",?err) ????????} ????????fmt.Println("c") ????}() ????//?創(chuàng)建?goroutine?調用?F?函數 ????go?F() ????time.Sleep(time.Second) } func?F()?{ ????defer?func()?{ ????????fmt.Println("b") ????}() ????//?goroutine?內部拋出panic ????panic("a") } func?main()?{ ????G() }
順序如下:
- 調用 G() 函數;
- 通過 goroutine 調用 F() 函數;
- F() 中遇到 panic,立刻終止,不執(zhí)行 panic 之后的代碼;
- 執(zhí)行 F() 中 defer 函數,由于沒有 recover,則將 panic 拋到 G() 中;
- 由于 goroutine 內部沒有進行 recover,則 goroutine 外部函數,也就是 G() 函數是沒辦法捕獲的,程序直接崩潰退出。
輸出:
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 類型,所以這樣使用的話會報錯:
err.Error?undefined?(type?interface?{}?is?interface?with?no?methods)
可以這樣來轉換一下:
defer?func()?{ ????if?err?:=?recover();?err?!=?nil?{ ????????fmt.Println("捕獲異常:",?fmt.Errorf("%v",?err).Error()) ????} }() panic("a")
或者直接打印結果:
defer?func()?{ ????if?err?:=?recover();?err?!=?nil?{ ????????fmt.Println("捕獲異常:",?err) ????} }() panic("a")
輸出:
捕獲異常: a
以上就是本文的全部內容,其實寫過其他的語言的同學都知道,關閉文件句柄,釋放鎖等操作是很容易忘的。而 Go 語言通過 defer 很好地解決了這個問題,但在使用過程中還是要小心。
源碼地址:https://github.com/yongxinz/gopher/tree/main/sc
以上就是Go程序員踩過的defer坑錯誤處理的詳細內容,更多關于Go defer錯誤處理的資料請關注腳本之家其它相關文章!
相關文章
在ubuntu下構建go語言開發(fā)環(huán)境的方法
這篇文章主要介紹了在ubuntu下構建go語言開發(fā)環(huán)境的方法,需要的朋友可以參考下2014-10-10