深入了解 Go-Defer的機制

defer 的作用和執行時機

go 的 defer 語句是用來延遲執行函數的，而且延遲發生在調用函數 return 之後，比如

func a() int {

defer b()

return 0

}

b 的執行是發生在return 0之後，注意defer 的文法，關鍵字defer之後是函數的調用。

defer 的重要用途一：清理釋放資源

由于defer 的延遲特性，defer常用在函數調用結束之後清理相關的資源，比如

f, _ := os.Open(filename)

defer f.Close()

檔案資源的釋放會在函數調用結束之後借助 defer 自動執行，不需要時刻記住哪裡的資源需要釋放，打開和釋放必須相對應。

用一個例子深刻诠釋一下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 { //1
    return
}

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

代碼在 #1處傳回之後，src檔案沒有執行關閉操作，可能會導緻資源不能正确釋放，改用 defer實作：

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)           

src 和dst都能及時清理和釋放，無論return在什麼地方執行。

鑒于defer 的這種作用，defer常用來釋放資料庫連接配接，檔案打開句柄等釋放資源的操作。

defer的重要用途二：執行recover

被 defer的函數在return之後執行，這個時機點正好可以捕獲函數抛出的 panic，因而 defer 的另一個重要用途就是執行 recover。

recover 隻有在defer中使用才更有意義，如果在其他地方使用，由于 program已經調用結束而提前傳回而無法有效捕捉錯誤。

package main

import (

"fmt"           

)

func main() {

defer func() {
    if ok := recover(); ok != nil {
        fmt.Println("recover")
    }
}()

panic("error")
           

記住defer要放在panic執行之前。

多個 defer 的執行順序

defer 的作用就是把關鍵字之後的函數執行壓入一個棧中延遲執行，多個defer的執行順序是後進先出LIFO：

defer func() { fmt.Println("1") }()

defer func() { fmt.Println("2") }()

defer func() { fmt.Println("3") }()

輸出順序是 321。

這個特性可以對一個 array實作逆序操作。

被 deferred 函數的參數在 defer 時确定

這是 defer的特點，一個函數被 defer時，它的參數在defer 時進行計算确定，即使defer之後參數發生修改，對已經defer的函數沒有影響，什麼意思？看例子：

func a() {

i := 0
defer fmt.Println(i)
i++
return           

a 執行輸出的是 0 而不是 1，因為defer 時，i 的值是 0，此時被 defer的函數參數已經進行執行計算并确定了。

再看一個例子：

func calc(index string, a, b int) int {

ret := a + b
fmt.Println(index, a, b, ret)
return ret           

a := 1
b := 2
defer calc("1", a, calc("10", a, b))
a = 0
return           

執行代碼輸出

10 1 2 3

1 1 3 4

defer函數的參數 第三個參數在 defer時就已經計算完成并确定，第二個參數 a 也是如此，無論之後 a 變量是否修改都不影響。

被 defer的函數可以讀取和修改帶名稱的傳回值

func c() (i int) {

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

被 defer的函數是在return 之後執行，可以修改帶名稱的傳回值，上面的函數 c 傳回的是 2。

參考資料

https://blog.golang.org/defer-panic-and-recover