Golang關(guān)鍵字defer的用法詳解
1. defer的簡單介紹與使用場景
defer是Go里面的一個關(guān)鍵字,用在方法或函數(shù)前面,作為方法或函數(shù)的延遲調(diào)用。它主要用于以下兩個場景:
- 優(yōu)雅釋放資源,比如一些網(wǎng)絡(luò)連接、數(shù)據(jù)庫連接以及文件資源的釋放。
- 與recover配合處理panic異常
場景一:復(fù)制文件
func CopyFile(dstFile, srcFile string) (wr int64, err error) {
src, err := os.Open(srcFile)
if err != nil {
return
}
dst, err := os.Create(dstFile)
if err != nil {
return
}
wr, err = io.Copy(dst, src)
dst.Close()
src.Close()
return
}這樣的代碼是有問題的,當?shù)?行執(zhí)行失敗的時候程序返回但沒有關(guān)閉前面成功打開的src,資源沒有正確關(guān)閉,正確代碼如下:
(在成功打開資源,沒返回err的情況下,都可以使用defer進行優(yōu)雅的關(guān)閉資源)
func CopyFile(dstFile, srcFile string) (wr int64, err error) {
src, err := os.Open(srcFile)
if err != nil {
return
}
defer src.Close()
dst, err := os.Create(dstFile)
if err != nil {
return
}
defer dst.Close()
wr, err = io.Copy(dst, src)
return err
}場景二:處理異常panic
package main
import "fmt"
func main() {
defer func() {
if r := recover(); r != nil {
fmt.Println(r)
}
}()
a := 1
b := 0
fmt.Println("result:", a/b)
}運行結(jié)果:
runtime error: integer divide by zero
程序沒有輸出result, 會拋出panic, 因為不能對除數(shù)為0的數(shù)做除法,我們使用defer在程序發(fā)生panic的時候捕獲異常。go中是用panic拋異常,recover捕獲異常,異常會在下一篇go文章進行分析。
通過這兩個使用場景我們也可以看到defer后面跟著的函數(shù)被調(diào)用的時間:
- 函數(shù)return的時候
- 當前協(xié)程發(fā)生panic的時候
說到延遲函數(shù)被調(diào)用的時機,這時順便說一下多個延遲函數(shù)被調(diào)用時候的執(zhí)行順序問題。官方對defer的解釋中寫到每次defer語句執(zhí)行的時候,會把函數(shù)“壓棧”,同時函數(shù)參數(shù)會被拷貝下來
這兩點很重要:
一是說明當一個函數(shù)中有多個defer的時候,執(zhí)行順序是LIFO先進后出
二是說明延遲函數(shù)的參數(shù)在defer語句出現(xiàn)時就已經(jīng)確定下來了
這段代碼是用來補充這二點的:看一下能否看懂執(zhí)行結(jié)果,能的話就直接跳到第二部分
func deferRun1() {
var num = 1
numptr := &num
defer fmt.Println("defer run 1: \n", numptr, num, *numptr) // 0xc000022088 1 1
defer func() {
fmt.Println("defer run 2 : \n", numptr, num, *numptr) // 0xc000116028 2 2
}()
defer func(num int, numptr *int) {
fmt.Println("defer run 3: \n", numptr, num, *numptr) //0xc000116028 1 2
}(num, numptr)
num = 2
fmt.Println(numptr, num, *numptr) // 0xc000116028 2 2
return
}對于第一個defer,傳入的num和*numptr都是一個具體的值,所以程序return完之后結(jié)果會是1,1
對于第二個defer, 沒有傳入?yún)?shù),結(jié)果會和最后的num和numptr地址對應(yīng)的內(nèi)容相同
對于第三個defer, 傳入的參數(shù)num是固定的值,而numptr是固定的地址,后面地址對應(yīng)的內(nèi)容被修改了,所以結(jié)果是1,2
同時,執(zhí)行順序會是第三個defer先執(zhí)行,然后是第二個...第一個
2. defer在return執(zhí)行的時機
第一小節(jié)上面說了有一種情況延遲函數(shù)的執(zhí)行是在return的時候,再具體一點就是在return的時候,defer操作帶來了什么結(jié)果呢?
一句話解決這個問題就是:函數(shù)return不是一個原子操作,需要經(jīng)過以下三步:
- 設(shè)置返回值
- 執(zhí)行defer
- 將結(jié)果返回
一定要牢記在心中,分析的時候也要嚴格按照這三步來,否則極容易掉坑!
下面把需要注意的情況列舉出來:
情況一:延遲函數(shù)參數(shù)早已固定下來(第一小節(jié)提到的重要的一點)
func main() {
deferRun()
deferRun2()
}
func deferRun() {
var num = 1
defer fmt.Printf("num is %d\n", num)
num = 2
return
}
func deferRun2() {
var arr = [4]int{1, 2, 3, 4}
defer printArr(&arr)
arr[0] = 100
return
}
func printArr(arr *[4]int) {
for i := range arr {
fmt.Println(arr[i])
}
}運行結(jié)果為 num is 1 以及 100,2,3,4
道理很簡單:延遲函數(shù)的參數(shù)在defer出現(xiàn)的時候就固定了,對于deferRun,傳的參數(shù)是num的值,而對于deferRun,傳的參數(shù)是arr的地址,地址不變,但是地址對應(yīng)的內(nèi)容被修改,所以輸出被改變。
情況二: 嚴格把握三步走,繞開 “匿名” 小坑,分析看注釋
func main() {
res1 := deferRun()
fmt.Println(res1)
res2 := deferRun2()
fmt.Println(res2)
res3 := deferRun3()
fmt.Println(res3)
res4 := deferRun4()
fmt.Println(res4)
}
// return 2
func deferRun() (res int) {
num := 1
defer func() {
res++
}()
return num //1.返回值參數(shù)res賦值為num即1 //2.執(zhí)行defer語句,res++ //3. 結(jié)果返回2
}
// return 1
func deferRun2() int { // 注意這里返回值是匿名的,我們可以在心里為這個匿名的返回值取名為res
var num int
defer func() {
num++
}()
return 1 //1.匿名返回值參數(shù)(res)賦值為1 //2.執(zhí)行defer語句,num++,加了個寂寞 //3.結(jié)果返回1
}
//return 1
func deferRun3() int {
num := 1
defer func() {
num++
}()
return num //1.匿名(res)賦值為num即1 //2.執(zhí)行defer語句,num++,加了個寂寞,num為2,res依舊為1
//3.結(jié)果返回1
}
//return 2
func deferRun4() (res int) {
num := 1
defer func() {
res++
}()
return num //1.返回值參數(shù)res賦值為1 //2.執(zhí)行defer語句,res++ //3.結(jié)果返回2
}執(zhí)行結(jié)果為 2 1 1 2
3. 小結(jié)
defer很優(yōu)雅,可以很簡潔的釋放資源同時也可以與recover配合進行panic異常的處理
defer定義的延遲函數(shù)的參數(shù)在defer語句出現(xiàn)的時候就確定下來了,注意有時候確定的是地址
return不是原子操作,包括了三個重要的步驟,順序是:(注意匿名返回值參數(shù))
設(shè)置返回值參數(shù)——>執(zhí)行defer語句——>返回結(jié)果
到此這篇關(guān)于Golang關(guān)鍵字defer的用法詳解的文章就介紹到這了,更多相關(guān)Golang關(guān)鍵字defer內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
服務(wù)器端Go程序?qū)﹂L短鏈接的處理及運行參數(shù)的保存
這篇文章主要介紹了服務(wù)器端Go程序?qū)﹂L短鏈接的處理及運行參數(shù)的保存,這里針對使用Go語言編寫的Socket服務(wù)器進行實例說明,需要的朋友可以參考下2016-03-03
詳解Golang time包中的結(jié)構(gòu)體time.Time
在日常開發(fā)過程中,會頻繁遇到對時間進行操作的場景,使用 Golang 中的 time 包可以很方便地實現(xiàn)對時間的相關(guān)操作,本文先講解一下 time 包中的結(jié)構(gòu)體 time.Time,需要的朋友可以參考下2023-07-07

