GO中DEFER的理解--DEFER執行的原理
在golang當中,defer代碼塊會在函數調用鏈表中增加一個函數調用。這個函數調用不是普通的函數調用,而是會在函數正常返回,也就是return之後添加一個函數調用。因此,defer通常用來釋放函數內部變量。
為了更好的學習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 }
這段代碼可以運行,但存在‘安全隱患‘。如果調用dst, err := os.Create(dstName)失敗,則函數會執行return退出運行。但之前創建的src(文件句柄)沒有被釋放。 上面這段代碼很簡單,所以我們可以一眼看出存在文件未被釋放的問題。 如果我們的邏輯復雜或者代碼調用過多時,這樣的錯誤未必會被及時發現。 而使用defer則可以避免這種情況的發生,下面是使用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,我們可以在代碼中優雅的關閉/清理代碼中所使用的變量。defer作為golang清理變量的特性,有其獨有且明確的行為。以下是defer三條使用規則。
規則一 當defer被聲明時,其參數就會被實時解析
我們通過以下代碼來解釋這條規則:
func a() { i := 0 defer fmt.Println(i) i++ return }
上面我們說過,defer函數會在return之後被調用。那麽這段函數執行完之後,是不用應該輸出1呢?
讀者自行編譯看一下,結果輸出的是0. why?
這是因為雖然我們在defer後面定義的是一個帶變量的函數: fmt.Println(i). 但這個變量(i)在defer被聲明的時候,就已經確定其確定的值了。 換言之,上面的代碼等同於下面的代碼:
func a() { i := 0 defer fmt.Println(0) //因為i=0,所以此時就明確告訴golang在程序退出時,執行輸出0的操作 i++ return }
為了更為明確的說明這個問題,我們繼續定義一個defer:
func a() { i := 0 defer fmt.Println(i) //輸出0,因為i此時就是0 i++ defer fmt.Println(i) //輸出1,因為i此時就是1 return }
通過運行結果,可以看到defer輸出的值,就是定義時的值。而不是defer真正執行時的變量值(很重要,搞不清楚的話就會產生於預期不一致的結果)
但為什麽是先輸出1,在輸出0呢? 看下面的規則二。
規則二 defer執行順序為先進後出
當同時定義了多個defer代碼塊時,golang安裝先定義後執行的順序依次調用defer。不要為什麽,golang就是這麽定義的。我們用下面的代碼加深記憶和理解:
func b() { for i := 0; i < 4; i++ { defer fmt.Print(i) } }
在循環中,依次定義了四個defer代碼塊。結合規則一,我們可以明確得知每個defer代碼塊應該輸出什麽值。 安裝先進後出的原則,我們可以看到依次輸出了3210.
規則三 defer可以讀取有名返回值
先看下面的代碼:
func c() (i int) { defer func() { i++ }() return 1 }
輸出結果是12. 在開頭的時候,我們說過defer是在return調用之後才執行的。 這裏需要明確的是defer代碼塊的作用域仍然在函數之內,結合上面的函數也就是說,defer的作用域仍然在c函數之內。因此defer仍然可以讀取c函數內的變量(如果無法讀取函數內變量,那又如何進行變量清除呢....)。
當執行return 1 之後,i的值就是1. 此時此刻,defer代碼塊開始執行,對i進行自增操作。 因此輸出2.
掌握了defer以上三條使用規則,那麽當我們遇到defer代碼塊時,就可以明確得知defer的預期結果。
GO中DEFER的理解--DEFER執行的原理