golang中defer的使用規則

轉自個人部落格chinazt.cc

在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的預期結果。