前言

defer是golang中用的比較多的一個(gè)關(guān)鍵字，也是go面試題里經(jīng)常出現(xiàn)的問題，而在很多時(shí)候我們只知其然，而不知其所以然，今天就來整理一下關(guān)于defer的學(xué)習(xí)使用，希望對(duì)需要的朋友有所幫助。

defer是什么

defer是go中一種延遲調(diào)用機(jī)制，defer后面的函數(shù)只有在當(dāng)前函數(shù)執(zhí)行完畢后才能執(zhí)行，將延遲的語句按defer的逆序進(jìn)行執(zhí)行，也就是說先被defer的語句最后被執(zhí)行，最后被defer的語句，最先被執(zhí)行，通常用于釋放資源。

定義：

defer function([parameter_list]) // 延遲執(zhí)行函數(shù)
defer method([parameter_list]) // 延遲執(zhí)行方法

多個(gè)defer的執(zhí)行順序

多個(gè)defer出現(xiàn)的時(shí)候，它會(huì)把defer之后的函數(shù)壓入一個(gè)棧中延遲執(zhí)行，也就是先進(jìn)后出(LIFO)，寫在前面的defer會(huì)比寫在后面的defer調(diào)用的晚。下面通過一個(gè)示例看一下：

func func1(){
    fmt.Println("我是 func1")
}
func func2(){
    fmt.Println("我是 func2")
}
func func3(){
    fmt.Println("我是 func3")
}
func main(){
    defer func1()
    defer func2()
    defer func3()
    fmt.Println("main1")
    fmt.Println("main2")
}

執(zhí)行輸出如下：

main1
main2
我是 func3
我是 func2
我是 func1

通過圖示一看就很明白了

延遲函數(shù)的參數(shù)在defer聲明時(shí)就決定了

func main(){
   i:= 0
   defer func(a int) {
		fmt.Println(a)
	}(i)
    i++
}

此時(shí)輸出的值是0，而不是1，因?yàn)閐efer后面的函數(shù)在入棧的時(shí)候保存的是入棧那一刻的值，而當(dāng)時(shí)i的值是0，所以后期對(duì)i進(jìn)行修改，并不會(huì)影響棧內(nèi)函數(shù)的值。

如果我們把參數(shù)傳引用

func main(){
   i:= 0
   defer func(a *int) {
		fmt.Println(*a)
	}(&i)
    i++
}

此時(shí)輸出的值是1，因?yàn)檫@里defer后面函數(shù)入棧的時(shí)候唇乳的執(zhí)行變量i的指針，后期i值改變的時(shí)候，輸出結(jié)果也會(huì)改變。

defer和return的順序

首先看下defer和return語句的區(qū)別，如下：

可以看到 return 執(zhí)行的時(shí)候，并不是原子性操作，一般是分為兩步:將結(jié)果x賦值給了返回值，然后執(zhí)行了RET指令；而defer語句執(zhí)行的時(shí)候，是在賦值變量之后，在RET指令之前。所以這里注意一下。返回值和x的關(guān)系。如果x是一個(gè)值類型，這里是進(jìn)行了拷貝的。

示例：

package main

import "fmt"

func deferFunc() int {
	fmt.Println("defer func called")
	return 0
}

func returnFunc() int {
	fmt.Println("return func called")
	return 0
}

func returnAndDefer() int {

	defer deferFunc()

	return returnFunc()
}

func main() {
	returnAndDefer()
}

執(zhí)行結(jié)果為：

return func called
defer func called

defer和panic

當(dāng)函數(shù)遇到panic，defer仍然會(huì)被執(zhí)行。Go會(huì)先執(zhí)行所有的defer鏈表(該函數(shù)的所有defer)，當(dāng)所有defer被執(zhí)行完畢且沒有recover時(shí)，才會(huì)進(jìn)行panic。

defer 最大的功能是 panic 后依然有效,所以defer可以保證你的一些資源一定會(huì)被關(guān)閉，從而避免一些異常出現(xiàn)的問題。

package main

import "fmt"

func main() {

	deferPanic()
}

func deferPanic() {

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

	panic("出錯(cuò)啦")
}

執(zhí)行輸出如下：

defer 3
defer 2
defer 1
panic: 出錯(cuò)啦

我們可以在defer中進(jìn)行recover，如果defer中包含recover，則程序?qū)⒉粫?huì)再進(jìn)行panic，這就實(shí)現(xiàn)了Go中異常拋出/捕獲類似的機(jī)制。

package main
 
import (
    "fmt"
)
 
func main()  {
 
    defer func() {
        //捕獲異常
       if err := recover(); err != nil{
           fmt.Println(err)
       }else {
           fmt.Println("fatal")
       }
    }()
    
    //拋出異常
    panic("panic")
}

defer下的函數(shù)參數(shù)包含子函數(shù)

package main
 
import "fmt"
 
func function(index int, value int) int {
 
    fmt.Println(index)
 
    return index
}
 
func main() {
    defer function(1, function(3, 0))
    defer function(2, function(4, 0))

這個(gè)程序的執(zhí)行結(jié)果是怎么樣的的？

首先兩個(gè)defer會(huì)壓棧兩次，先進(jìn)棧1，后進(jìn)棧2，在壓棧function1的時(shí)候，需要連同函數(shù)地址、函數(shù)形參一同進(jìn)棧，那么為了得到function1的第二個(gè)參數(shù)的結(jié)果，需要先執(zhí)行function3將第二個(gè)參數(shù)算出，所以function3就被第一個(gè)執(zhí)行。同理壓入棧function2，就需要先執(zhí)行function4算出function2的第二個(gè)參數(shù)的值，然后函數(shù)結(jié)束，先出棧function2、再出棧function1。輸出結(jié)果如下：

3
4
2
1

總結(jié)

• defer是go中一種延遲調(diào)用機(jī)制，defer后面的函數(shù)只有在當(dāng)前函數(shù)執(zhí)行完畢后才能執(zhí)行。
• 多個(gè)defer出現(xiàn)的時(shí)候，它會(huì)把defer之后的函數(shù)壓入一個(gè)棧中延遲執(zhí)行，也就是先進(jìn)后出。
• defer后面的函數(shù)值在入棧的時(shí)候就決定了。
• defer 最大的功能是 panic 后依然有效,我們可以在defer中進(jìn)行recover，如果defer中包含recover，則程序?qū)⒉粫?huì)再進(jìn)行panic，實(shí)現(xiàn)try catch機(jī)制。