golang defer 使用小結與注意要點

關於延時調用函數(Deferred Function Calls)

延時調用函數的語法以下:

defer func_name(param-list)

當一個函數調用前有關鍵字 defer 時, 那麼這個函數的執行會推遲到包含這個 defer 語句的函數即將返回前才執行. 例如:

func main() {
    defer fmt.Println("Fourth")
    fmt.Println("First")
    fmt.Println("Third")
}

最後打印順序以下:

First
Second
Third

須要注意的是, defer 調用的函數參數的值 defer 被定義時就肯定了.
例如:

i := 1
defer fmt.Println("Deferred print:", i)
i++
fmt.Println("Normal print:", i)

打印的內容以下:

Normal print: 2
Deferred print: 1

所以咱們知道, 在 "defer fmt.Println("Deferred print:", i)" 調用時, i 的值已經肯定了, 所以至關於 defer fmt.Println("Deferred print:", 1) 了.
須要強調的時, defer 調用的函數參數的值在 defer 定義時就肯定了, 而 defer 函數內部所使用的變量的值須要在這個函數運行時才肯定. 例如:

func f1() (r int) {
    r = 1
    defer func() {
        r++
        fmt.Println(r)
    }()
    r = 2
    return
}

func main() {
    f1()
}

上面的例子中, 最終打印的內容是 "3", 這是由於在 "r = 2" 賦值以後, 執行了 defer 函數, 所以在這個函數內, r 的值是2了, 自增後變爲3.

defer 順序

若是有多個defer 調用, 則調用的順序是先進後出的順序, 相似於入棧出棧同樣:

func main() {
    defer fmt.Println(1)
    defer fmt.Println(2)
    defer fmt.Println(3)
    defer fmt.Println(4)
}

最早執行的是 "fmt.Println(4)", 接着是 "fmt.Println(3)" 依次類推, 最後的輸出以下:

4
3
2
1

defer 注意要點

defer 函數調用的執行時機是外層函數設置返回值以後, 而且在即將返回以前.
例如:

func f1() (r int) {
    defer func() {
        r++
    }()
    return 0
}
func main() {
    fmt.Println(f1())
}

上面 fmt.Println(f1()) 打印的是什麼呢? 不少朋友可能會認爲打印的是0, 可是正確答案是 1. 這是爲何呢?
要弄明白這個問題, 咱們須要牢記兩點

• defer 函數調用的執行時機是外層函數設置返回值以後, 而且在即將返回以前

• return XXX 操做並非原子的.

咱們將上面的例子改寫一下你們就很明白了:

func f1() (r int) {
    defer func() {
        r++
    }()
    r = 0
    return
}

當進行賦值操做 "r = 0" 後, 才調用 defer 函數, 最後纔是返回語句.
所以上面的代碼等效於:

func f1() (r int) {
    r = 0
    func() {
        r++
    }()
    return
}

接下來咱們再來看一個更有意思的例子:

func double(x int) int {
    return x + x
}

func triple(x int) (r int) {
    defer func() {
        r += x
    }()
    return double(x)
}

func main() {
    fmt.Println(triple(3))
}

若是咱們已經理解了上面所說的內容的話, 那麼 triple 函數就很好理解了, 它其實是:

func triple(x int) (r int) {
    r = double(x)
    func() {
        r += x
    }()
    return
}

defer 表達式的使用場景

defer 一般用於 open/close, connect/disconnect, lock/unlock 等這些成對的操做, 來保證在任何狀況下資源都被正確釋放. 在這個角度來講, defer 操做和 Java 中的 try ... finally 語句塊有殊途同歸之處.
例如:

var mutex sync.Mutex
var count = 0

func increment() {
    mutex.Lock()
    defer mutex.Unlock()
    count++
}

在increment 函數中, 咱們爲了不競態條件的出現, 而使用了 Mutex 進行加鎖. 而在進行併發編程時, 加鎖了卻忘記(或某種狀況下 unlock 沒有被執行), 每每會形成災難性的後果. 爲了在任意狀況下, 都要保證在加鎖操做後, 都進行對應的解鎖操做, 咱們可使用 defer 調用解鎖操做.

本文由 yongshun 發表於我的博客, 採用署名-非商業性使用-相同方式共享 3.0 中國大陸許可協議.非商業轉載請註明做者及出處. 商業轉載請聯繫做者本人Email: yongshun1228@gmail.com本文標題爲: golang defer 使用小結與注意要點本文連接爲: segmentfault.com/a/1190000006823652