Go使用defer函數(shù)要注意的幾個(gè)點(diǎn)

概述

defer 函數(shù)大家肯定都用過，它在聲明時(shí)不會(huì)立刻去執(zhí)行，而是在函數(shù) return 后去執(zhí)行的。

它的主要應(yīng)用場景有異常處理、記錄日志、清理數(shù)據(jù)、釋放資源 等等。

這篇文章不是分享 defer 的應(yīng)用場景，而是分享使用 defer 需要注意的點(diǎn)。

咱們先從一道題開始，一起來感受下 ...

func calc(index string, a, b int) int {
 ret := a + b
 fmt.Println(index, a, b, ret)
 return ret
}

func main() {
 x := 1
 y := 2
 defer calc("A", x, calc("B", x, y))
 x = 3
 defer calc("C", x, calc("D", x, y))
 y = 4
}

輸出什么？
...

接下來，先容我分享幾個(gè)小例子，再進(jìn)行作答。

執(zhí)行順序

func main() {

 defer fmt.Println("1")
 defer fmt.Println("2")
 defer fmt.Println("3")

 fmt.Println("main")
}

輸出：

main
3
2
1

結(jié)論：defer 函數(shù)定義的順序 與 實(shí)際執(zhí)的行順序是相反的，也就是最先聲明的最后才執(zhí)行。

閉包

func main() {

 var a = 1
 var b = 2

 defer fmt.Println(a + b)

 a = 2

 fmt.Println("main")
}

輸出：

main
3

稍微修改一下，再看看：

func main() {
 var a = 1
 var b = 2

 defer func() {
 fmt.Println(a + b)
 }()

 a = 2

 fmt.Println("main")
}

輸出：

main
4

結(jié)論：閉包獲取變量相當(dāng)于引用傳遞，而非值傳遞。

稍微再修改一下，再看看：

func main() {
 var a = 1
 var b = 2

 defer func(a int, b int) {
 fmt.Println(a + b)
 }(a, b)

 a = 2

 fmt.Println("main")
}

輸出：

main
3

結(jié)論：傳參是值復(fù)制。

還可以理解為：defer 調(diào)用的函數(shù)，參數(shù)的值在 defer 定義時(shí)就確定了，看下代碼

defer fmt.Println(a + b)，在這時(shí)，參數(shù)的值已經(jīng)確定了。

而 defer 函數(shù)內(nèi)部所使用的變量的值需要在這個(gè)函數(shù)運(yùn)行時(shí)才確定，看下代碼

defer func() { fmt.Println(a + b) }()，a 和 b 的值在函數(shù)運(yùn)行時(shí)，才能確定。

Return

一

func t1() int {
 a := 1
 defer func() {
 a++
 }()
 return a
}

輸出：1

二

func t2() (a int) {
 defer func() {
 a++
 }()
 return 1
}

輸出：2

三

func t3() (b int) {
 a := 1
 defer func() {
 a++
 }()
 return 1
}

輸出：1

四

func t4() (a int) {
 defer func(a int) {
 a++
 }(a)
 return 1
}

輸出：1

結(jié)論：return 不是原子操作。

os.Exit

func main() {
 defer fmt.Println("1")
 fmt.Println("main")
 os.Exit(0)
}

輸出：main

結(jié)論：當(dāng)os.Exit()方法退出程序時(shí)，defer不會(huì)被執(zhí)行。

不同協(xié)程

func main() {
 GoA()
 time.Sleep(1 * time.Second)
 fmt.Println("main")
}

func GoA() {
 defer (func(){
 if err := recover(); err != nil {
  fmt.Println("panic:" + fmt.Sprintf("%s", err))
 }
 })()

 go GoB()
}

func GoB() {
 panic("error")
}

GoB() panic 捕獲不到。
結(jié)論：defer 只對(duì)當(dāng)前協(xié)程有效。
這個(gè)問題怎么解？咱們下回再說。
接下來，咱們分析下文章開頭的問題吧。

答案解析

先列出答案：

B 1 2 3
D 3 2 5
C 3 5 8
A 1 3 4

其實(shí)上面那道題，可以拆解為：

func calc(index string, a, b int) int {
 ret := a + b
 fmt.Println(index, a, b, ret)
 return ret
}

func main() {
 x := 1
 y := 2
 tmp1 := calc("B", x, y)
 defer calc("A", x, tmp1)
 x = 3
 tmp2 := calc("D", x, y)
 defer calc("C", x, tmp2)
 y = 4
}

所以順序就是：B D C A。
執(zhí)行到 tmp1 時(shí)，輸出：B 1 2 3。
執(zhí)行到 tmp2 時(shí)，輸出：D 3 2 5。
根據(jù) defer 執(zhí)行順序原則，先聲明的后執(zhí)行，所以下一個(gè)該執(zhí)行 C 了。
又因?yàn)閭鲄⑹侵蒂x值，所以在 A 的時(shí)候，無法用到 x = 3 和 y = 4，在 C 的時(shí)候，無法用到 y = 4。
執(zhí)行到 C 時(shí)，輸出：C 3 5 8
執(zhí)行到 A 時(shí)，輸出：A 1 3 4
到這，基本上 defer 就清楚了，大家可以根據(jù)自己的理解去記憶。

以上就是本文的全部內(nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。