轉(zhuǎn)自個人博客chinazt.cc

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

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

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}

這段代碼可以運行,但存在'安全隱患'。如果調(diào)用dst, err := os.Create(dstName)失敗,則函數(shù)會執(zhí)行return退出運行。但之前創(chuàng)建的src(文件句柄)沒有被釋放。 上面這段代碼很簡單,所以我們可以一眼看出存在文件未被釋放的問題。 如果我們的邏輯復(fù)雜或者代碼調(diào)用過多時,這樣的錯誤未必會被及時發(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)
}

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

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

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

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

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

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

這是因為雖然我們在defer后面定義的是一個帶變量的函數(shù): fmt.Println(i). 但這個變量(i)在defer被聲明的時候,就已經(jīng)確定其確定的值了。 換言之,上面的代碼等同于下面的代碼:

func a() {
i := 0defer fmt.Println(0) //因為i=
        
		

網(wǎng)友評論