天天看點

[系列] Go 使用 defer 函數 要注意的幾個點

概述

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

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

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

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

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
}
           

輸出什麼?

...

[系列] Go 使用 defer 函數 要注意的幾個點

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

執行順序

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
}
           

func t4() (a int) {
	defer func(a int) {
		a++
	}(a)
	return 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, 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。

執行到 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 就清楚了,大家可以根據自己的了解去記憶。

go-gin-api 系列文章

  • 7. 路由中間件 - 簽名驗證
  • 6. 路由中間件 - Jaeger 鍊路追蹤(實戰篇)
  • 5. 路由中間件 - Jaeger 鍊路追蹤(理論篇)
  • 4. 路由中間件 - 捕獲異常
  • 3. 路由中間件 - 日志記錄
  • 2. 規劃項目目錄和參數驗證
  • 1. 使用 go modules 初始化項目
[系列] Go 使用 defer 函數 要注意的幾個點

作者:新亮筆記(關注公衆号,可申請添加微信好友)

出處:https://www.cnblogs.com/xinliangcoder

本文版權歸作者和部落格園共有,歡迎轉載,但未經作者同意必須保留此段聲明,且在文章頁面明顯位置給出原文連接配接,否則保留追究法律責任的權利。