1. 程式人生 > 程式設計 >Go使用defer函式要注意的幾個點

Go使用defer函式要注意的幾個點

概述

defer 函式大家肯定都用過,它在宣告時不會立刻去執行,而是在函式 return 後去執行的。

它的主要應用場景有異常處理、記錄日誌、清理資料、釋放資源 等等。

這篇文章不是分享 defer 的應用場景,而是分享使用 defer 需要注意的點。

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

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

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

輸出什麼?
...

接下來,先容我分享幾個小例子,再進行作答。

執行順序

func main() {

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

 fmt.Println("main")
}

輸出:

main
3
2
1

結論:defer 函式定義的順序 與 實際執的行順序是相反的,也就是最先宣告的最後才執行。

閉包

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

結論:閉包獲取變數相當於引用傳遞,而非值傳遞。

稍微再修改一下,再看看:

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

結論:傳參是值複製。

還可以理解為:defer 呼叫的函式,引數的值在 defer 定義時就確定了,看下程式碼

defer fmt.Println(a + b),在這時,引數的值已經確定了。

而 defer 函式內部所使用的變數的值需要在這個函式執行時才確定,看下程式碼

defer func() { fmt.Println(a + b) }(),a 和 b 的值在函式執行時,才能確定。

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

結論:return 不是原子操作。

os.Exit

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

輸出:main

結論:當os.Exit()方法退出程式時,defer不會被執行。

不同協程

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 捕獲不到。
結論:defer 只對當前協程有效。
這個問題怎麼解?咱們下回再說。
接下來,咱們分析下文章開頭的問題吧。

答案解析

先列出答案:

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

其實上面那道題,可以拆解為:

func calc(index string,ret)
 return ret
}

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

所以順序就是:B D C A。
執行到 tmp1 時,輸出:B 1 2 3。
執行到 tmp2 時,輸出:D 3 2 5。
根據 defer 執行順序原則,先宣告的後執行,所以下一個該執行 C 了。
又因為傳參是值賦值,所以在 A 的時候,無法用到 x = 3 和 y = 4,在 C 的時候,無法用到 y = 4。
執行到 C 時,輸出:C 3 5 8
執行到 A 時,輸出:A 1 3 4
到這,基本上 defer 就清楚了,大家可以根據自己的理解去記憶。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。