腳本之家服務器常用軟件

快捷導航

Golang中關于defer的盲區(qū)梳理

更新時間：2023年03月06日 10:01:48 作者：7small7

關于Go中的defer，是做什么的？執(zhí)行順序是怎么樣的？相信學過Go語言的同學，已經(jīng)不在陌生，今天就來講講其中需要掌握的幾個知識點，希望對大家有所幫助

正因為defer具備這種機制，可以用在函數(shù)返回之前，關閉一些資源。例如在某些操作中，連接了MySQL、Redis這樣的服務，在函數(shù)返回之前，就可以使用defer語句對連接進行關閉。就類似oop語言中的 finally 操作一樣，不管發(fā)生任何異常，最終都會被執(zhí)行。

其語法格式也非常的簡單。

package main
import "fmt"

func main() {
    function1()
}

func function1() {
    fmt.Printf("1")
    defer function2()
    fmt.Printf("2")
}

func function2() {
    fmt.Printf("3")
}

上述代碼執(zhí)行的結(jié)果是：

1
2
3

下面就來總結(jié)這六個小知識點：

1、defer 的執(zhí)行順序。采用棧的方式執(zhí)行，先定義后執(zhí)行。

2、defer 與 return 誰先誰后。return 之后的語句先執(zhí)行，defer 后的語句后執(zhí)行。

3、函數(shù)的返回值初始化與 defer 間接影響。defer中修改了返回值，實際返回的值是按照defer修改后的值進行返回。

4、defer 遇見 panic。按照defer的棧順序，輸出panic觸發(fā)之前定義好的defer。

5、defer 中包含 panic。按照defer的棧順序，輸出panic觸發(fā)之前的defer。并且defer中會接收到panic信息。

6、defer 下的函數(shù)參數(shù)包含子函數(shù)。會先進行子函數(shù)的結(jié)果值，然后在按照棧的順序進行輸出。

defer的執(zhí)行順序是什么樣的

關于這個問題，前面的示例代碼也提到過了，采用棧的順序執(zhí)行。在定義時，壓入棧中，執(zhí)行是從棧中獲取。

defer與return誰先誰后

先來看如下一段代碼，最終的執(zhí)行結(jié)果是怎么樣的。

func main() {
    fmt.Println(demo2())
}

func demo2() int {
    defer func() {
        fmt.Println("2")
    }()
    return func() int {
        fmt.Println("1")
        return 4
    }()
}

運行上述代碼，得到的結(jié)果是：

1
2
4

可能你會有一個疑問 ，既然都提到了defer是在函數(shù)返回之前執(zhí)行，為什么還是先輸出1，然后在輸出2呢？關于defer的定義，就是在函數(shù)返回之前執(zhí)行。這一點毋庸置疑，肯定是在return之前執(zhí)行。需要注意的是，return 是非原子性的，需要兩步，執(zhí)行前首先要得到返回值 (為返回值賦值)，return 將返回值返回調(diào)用處。defer 和 return 的執(zhí)行順序是先為返回值賦值，然后執(zhí)行 defer，然后 return 到函數(shù)調(diào)用處。

函數(shù)的返回值初始化與defer間接影響

同樣的方式，我們先看一段代碼，猜測一下最終的執(zhí)行結(jié)果是什么。

func main() {
    fmt.Println(demo3())
}

func demo3() (a int) {
    defer func() {
        a = 3
    }()
    return 1
}

上訴代碼，最終的運行結(jié)果如下：

3

跟上第2個知識點類似，函數(shù)在return之前，會進行返回值賦值，然后在執(zhí)行defer語句，最終在返回結(jié)果值。

1、在定義函數(shù)demo3()時，為函數(shù)設置了一個int類型的變量a，此時int類型初始化值默認是0。

2、定義一個defer語句，在函數(shù)return之前執(zhí)行，匿名函數(shù)中對返回變量a進行了一次賦值，設置 a=3。

3、此時執(zhí)行return語句，因為return語句是執(zhí)行兩步操作，先為返回變量a執(zhí)行一次賦值操作，將a設置為3。緊接著執(zhí)行defer語句，此時defer又將a設置為3。

4、最終return進行返回，由于第3步的defer對a進行了重新賦值。因此a就變成了3。

5、最后main函數(shù)打印結(jié)果，打印的其實是defer修改之后的值。

如果將變量a的聲明放回到函數(shù)內(nèi)部聲明呢，其運行的結(jié)果會根據(jù)return的值進行返回。

func main() {
    fmt.Println(demo7())
}

func demo7() int {
    var a int
    defer func(a int) {
        a = 10
    }(a)

    return 2
}

上述的最終結(jié)果返回值如下：

10
2

為什么會發(fā)生兩種不同的結(jié)果呢？這是因為，這是因為發(fā)生了值拷貝現(xiàn)象。在執(zhí)行defer語句時，將參數(shù)a傳遞給匿名函數(shù)時進行了一個值拷貝的過程。由于值拷貝是不會影響原值，因此匿名函數(shù)對變量a進行了修改，不會影響函數(shù)外部的值。當然傳遞一個指針的話，結(jié)果就不一樣了。在函數(shù)定義時，聲明的變量可以理解為一個全局變量，因此defer或者return對變量a進行了修改，都會影響到該變量上。

defer遇見panic。

panic是Go語言中的一種異?，F(xiàn)象，它會中斷程序的執(zhí)行，并拋出具體的異常信息。既然會中斷程序的執(zhí)行，如果一段代碼中發(fā)生了panic，最終還會調(diào)用defer語句嗎？

func main() {
    demo4()
}

func demo4() {
    defer func() {
        fmt.Println("1")
    }()
    defer func() {
        fmt.Println("2")
    }()
    panic("panic")
    defer func() {
        fmt.Println("3")
    }()
    defer func() {
        fmt.Println("4")
    }()
}

運行上述代碼，最終得到的結(jié)果如下：

╰─ go run defer.go
2
1
panic: panic

goroutine 1 [running]:
main.demo4()

從上面的結(jié)果不難看出，雖然發(fā)生了panic異常信息，還是輸出了defer語句中的信息，這說明panic的發(fā)生，還是會執(zhí)行defer操作。那為什么后面的兩個defer沒有被執(zhí)行呢。這是因為pani的發(fā)生，會中斷程序的執(zhí)行，因此后續(xù)的代碼根本沒有拿到執(zhí)行權。

當函數(shù)中發(fā)生了panic異常，會馬上中止當前函數(shù)的執(zhí)行，panic之前定義的defer都會被執(zhí)行，所有的 defer 語句都會保證執(zhí)行并把控制權交還給接收到 panic 的函數(shù)調(diào)用者。這樣向上冒泡直到最頂層，并執(zhí)行（每層的） defer，在棧頂處程序崩潰，并在命令行中用傳給 panic 的值報告錯誤情況：這個終止過程就是 panicking。

defer中包含panic

上一個知識點提到了，程序中雖然發(fā)生了panic，但是在panic之前定義的defer語句，還是會被執(zhí)行。要想在defer中獲取到具體的panic信息，需要使用 recover() 進行獲取。

func main() {
    demo5()
}

func demo5() {
    defer func() {
        fmt.Println("1")
        if err := recover(); err != nil {
            fmt.Println(err)
        }
    }()
    defer func() { fmt.Println("2") }()
    panic("panic")
    defer func() { fmt.Println("defer: panic 之后, 永遠執(zhí)行不到") }()
}

上述代碼執(zhí)行的結(jié)果如下：

2
1
panic

這個（recover）內(nèi)建函數(shù)被用于從 panic 或錯誤場景中恢復：讓程序可以從 panicking 重新獲得控制權，停止終止過程進而恢復正常執(zhí)行。

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

對于這種場景，可能大家很少遇見，也不是很清楚實際的調(diào)用邏輯。先來看一段代碼。

func main() {
    demo6()
}

func function(index int, value int) int {
    fmt.Println(index)
    return index
}

func demo6() {
    defer function(1, function(3, 0))
    defer function(2, function(4, 0))
}

上訴代碼最終執(zhí)行的結(jié)果是：