golang中defer如何使用

golang中defer如何使用，很多新手對(duì)此不是很清楚，為了幫助大家解決這個(gè)難題，下面小編將為大家詳細(xì)講解，有這方面需求的人可以來(lái)學(xué)習(xí)下，希望你能有所收獲。

前言

在golang當(dāng)中，defer代碼塊會(huì)在函數(shù)調(diào)用鏈表中增加一個(gè)函數(shù)調(diào)用。這個(gè)函數(shù)調(diào)用不是普通的函數(shù)調(diào)用，而是會(huì)在函數(shù)正常返回，也就是return之后添加一個(gè)函數(shù)調(diào)用。因此，defer通常用來(lái)釋放函數(shù)內(nèi)部變量。

為了更好的學(xué)習(xí)defer的行為，我們首先來(lái)看下面一段代碼:

func CopyFile(dstName, srcName string) (written int64, err error) { 
src, err := os.Open(srcName) 
if err != nil { 
return 
}

dst, err := os.Create(dstName) 
if err != nil { 
return 
}

written, err = io.Copy(dst, src) 
dst.Close() 
src.Close() 
return 
}

這段代碼可以運(yùn)行，但存在'安全隱患'。如果調(diào)用dst, err := os.Create(dstName)失敗，則函數(shù)會(huì)執(zhí)行return退出運(yùn)行。但之前創(chuàng)建的src(文件句柄)沒(méi)有被釋放。 上面這段代碼很簡(jiǎn)單，所以我們可以一眼看出存在文件未被釋放的問(wèn)題。 如果我們的邏輯復(fù)雜或者代碼調(diào)用過(guò)多時(shí)，這樣的錯(cuò)誤未必會(huì)被及時(shí)發(fā)現(xiàn)。 而使用defer則可以避免這種情況的發(fā)生，下面是使用defer的代碼:

func CopyFile(dstName, srcName string) (written int64, err error) { 
src, err := os.Open(srcName) 
if err != nil { 
return 
}
defer src.Close()

dst, err := os.Create(dstName) 
if err != nil { 
return 
}
defer dst.Close()

return io.Copy(dst, src) 
}

通過(guò)defer，我們可以在代碼中優(yōu)雅的關(guān)閉/清理代碼中所使用的變量。defer作為golang清理變量的特性，有其獨(dú)有且明確的行為。以下是defer三條使用規(guī)則。

規(guī)則一 當(dāng)defer被聲明時(shí)，其參數(shù)就會(huì)被實(shí)時(shí)解析

我們通過(guò)以下代碼來(lái)解釋這條規(guī)則:

func a() { 
i := 0 
defer fmt.Println(i) 
i++ 
return 
}

上面我們說(shuō)過(guò)，defer函數(shù)會(huì)在return之后被調(diào)用。那么這段函數(shù)執(zhí)行完之后，是不用應(yīng)該輸出1呢？

讀者自行編譯看一下，結(jié)果輸出的是0. why？

這是因?yàn)殡m然我們?cè)赿efer后面定義的是一個(gè)帶變量的函數(shù): fmt.Println(i) . 但這個(gè)變量(i)在defer被聲明的時(shí)候，就已經(jīng)確定其確定的值了。 換言之，上面的代碼等同于下面的代碼:

func a() { 
i := 0 
defer fmt.Println(0) //因?yàn)閕=0，所以此時(shí)就明確告訴golang在程序退出時(shí)，執(zhí)行輸出0的操作 
i++ 
return 
}

為了更為明確的說(shuō)明這個(gè)問(wèn)題，我們繼續(xù)定義一個(gè)defer:

func a() { 
i := 0 
defer fmt.Println(i) //輸出0，因?yàn)閕此時(shí)就是0 
i++ 
defer fmt.Println(i) //輸出1，因?yàn)閕此時(shí)就是1 
return 
}

通過(guò)運(yùn)行結(jié)果，可以看到defer輸出的值，就是定義時(shí)的值。而不是defer真正執(zhí)行時(shí)的變量值(很重要，搞不清楚的話就會(huì)產(chǎn)生于預(yù)期不一致的結(jié)果)

但為什么是先輸出1，在輸出0呢？ 看下面的規(guī)則二。

規(guī)則二 defer執(zhí)行順序?yàn)橄冗M(jìn)后出

當(dāng)同時(shí)定義了多個(gè)defer代碼塊時(shí)，golang安裝先定義后執(zhí)行的順序依次調(diào)用defer。不要為什么，golang就是這么定義的。我們用下面的代碼加深記憶和理解:

func b() { 
for i := 0; i < 4; i++ { 
defer fmt.Print(i) 
}
}

在循環(huán)中，依次定義了四個(gè)defer代碼塊。結(jié)合規(guī)則一，我們可以明確得知每個(gè)defer代碼塊應(yīng)該輸出什么值。 安裝先進(jìn)后出的原則，我們可以看到依次輸出了3210.

規(guī)則三 defer可以讀取有名返回值

先看下面的代碼:

func c() (i int) { 
defer func() { i++ }() 
return 1 
}

輸出結(jié)果是12. 在開(kāi)頭的時(shí)候，我們說(shuō)過(guò)defer是在return調(diào)用之后才執(zhí)行的。 這里需要明確的是defer代碼塊的作用域仍然在函數(shù)之內(nèi)，結(jié)合上面的函數(shù)也就是說(shuō)，defer的作用域仍然在c函數(shù)之內(nèi)。因此defer仍然可以讀取c函數(shù)內(nèi)的變量(如果無(wú)法讀取函數(shù)內(nèi)變量，那又如何進(jìn)行變量清除呢....)。

當(dāng)執(zhí)行return 1 之后，i的值就是1. 此時(shí)此刻，defer代碼塊開(kāi)始執(zhí)行，對(duì)i進(jìn)行自增操作。 因此輸出2.

掌握了defer以上三條使用規(guī)則，那么當(dāng)我們遇到defer代碼塊時(shí)，就可以明確得知defer的預(yù)期結(jié)果。

看完上述內(nèi)容是否對(duì)您有幫助呢？如果還想對(duì)相關(guān)知識(shí)有進(jìn)一步的了解或閱讀更多相關(guān)文章，請(qǐng)關(guān)注億速云行業(yè)資訊頻道，感謝您對(duì)億速云的支持。