Golang 的defer執(zhí)行規(guī)則說(shuō)明

更新時(shí)間：2021年04月28日 10:01:23 作者：dengming0922

這篇文章主要介紹了Golang 的defer執(zhí)行規(guī)則說(shuō)明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

defer介紹

defer是golang的一個(gè)特色功能，被稱為“延遲調(diào)用函數(shù)”。當(dāng)外部函數(shù)返回后執(zhí)行defer。類似于其他語(yǔ)言的 try… catch … finally… 中的finally，當(dāng)然差別還是明顯的。

在使用defer之前我們應(yīng)該多了解defer的特性，這樣才能避免使用上的誤區(qū)。

1. 最簡(jiǎn)單的defer

func test(){
    defer func(){ fmt.Println("defer") }()
    //todo
    //...
    return
    //defer執(zhí)行時(shí)機(jī)
}

我們可以稍微改動(dòng)一下上述代碼再次確認(rèn)defer的執(zhí)行時(shí)機(jī)。

func main() {
    fmt.Println(test())
}
func test() (i int) {
    defer func() { i++ }()
    defer func() { fmt.Println(i) }()
    //todo
    //...
    fmt.Println(0)
    return 1
    //defer執(zhí)行時(shí)機(jī)
}

output:

0

1

2

從上面示例可以發(fā)現(xiàn)defer執(zhí)行是在return之后，且按照defer聲明的先進(jìn)后出順序執(zhí)行。以下是真實(shí)場(chǎng)景中比較常見的用法。

釋放占用的資源

func test() error {
    file, err := os.Open("path")
    if err != nil {
        return err
    }
    //放在判斷err狀態(tài)之后
    defer file.Close()
    //todo
    //...
    return nil
    //defer執(zhí)行時(shí)機(jī)
}

捕捉處理異常

func test2() {
    defer func() {
        if err := recover(); err != nil {
            fmt.Println(err)
        }
    }()
    file, err := os.Open("path")
    if err != nil {
        panic(err)
    }
    defer file.Close()
    //todo
    //...
    return
    //defer執(zhí)行時(shí)機(jī)
}

輸出日志等收尾工作

func test3() {
    t1 := time.Now()
    defer func() {
        fmt.Printf("耗時(shí): %f s", time.Now().Sub(t1).Seconds())
    }()
    //todo
    //...
    return
    //defer執(zhí)行時(shí)機(jī)
}

2. 復(fù)雜的defer

當(dāng)我們已經(jīng)深深記住defer的執(zhí)行時(shí)機(jī)并打算翻過(guò)這一頁(yè)時(shí)，事情的發(fā)展又開始偏離初始。請(qǐng)看如下代碼

func test() {
    i := 0
    defer fmt.Println(i)                       //輸出 0
    defer func(x int) { fmt.Println(x) }(i)    //輸出 0
    defer func(x *int) { fmt.Println(*x) }(&i) //輸出 1
    defer func() { fmt.Println(i) }()          //輸出 1
    i++
    //todo
    //...
    fmt.Println(i) //輸出 1
    return   
}

output:

1

1

1

0 // 值未修改

0 // 值未修改

通常認(rèn)為defer就像真的被挪到了return之后。

但defer的本質(zhì)依然是函數(shù)調(diào)用。當(dāng)執(zhí)行到defer定義時(shí)，首先會(huì)對(duì)參數(shù)進(jìn)行求值，然后參數(shù)被壓入函數(shù)調(diào)用棧，此時(shí)不會(huì)進(jìn)入defer函數(shù)體，而是直到函數(shù)返回時(shí)才調(diào)用defer函數(shù)體。

參數(shù)被壓入函數(shù)調(diào)用棧時(shí)，如果參數(shù)是值類型，那么將復(fù)制值，如果參數(shù)是指針，那么將復(fù)制指針而不是復(fù)制指針指向的值。

defer函數(shù)體內(nèi)的變量是在return后執(zhí)行因此不受影響。

因此我們?cè)谑褂胐efer時(shí)一定要明確函數(shù)的參數(shù)類型（如果有的話），其次要明確defer函數(shù)體內(nèi)的變量引用是否正確。

以下是常見錯(cuò)誤

func test4() error {
    f, err := os.Open("A.txt")
    if err != nil {
        return err
    }
    defer func() { f.Close() }()//錯(cuò)誤: 關(guān)閉是B文件,f引用被重新賦值
    f, err = os.Open("B.txt")
    if err != nil {
        return err
    }
    defer func() { f.Close() }() //關(guān)閉是B文件
    list := []int{1, 2}
    for _, i := range list {
        defer fmt.Println(i) //輸出 2 1 //i為值類型參數(shù)被復(fù)制
        defer func() { fmt.Println(i) }() //錯(cuò)誤: 輸出 2 2 //函數(shù)體內(nèi)對(duì)i引用,留最終值
    }
    return nil
}

3. 更復(fù)雜點(diǎn)的defer

且看如下代碼

type Test struct {
    name string
}
func (this *Test) Point() { // this  為指針
    fmt.Println(this.name)
}
func (this  Test) Value() { //this  為值類型 
    fmt.Println(this.name)
}
func test5() {
    ts := []Test{{"a"}, {"b"}, {"c"}}
    for _, t := range ts {
        defer t.Point() //輸出 c c c
        defer t.Value() //輸出 c b a
    }
}

看似相同的代碼卻輸出了完全不同的結(jié)果。要理解這種差別還是得從調(diào)用函數(shù)的本質(zhì)來(lái)說(shuō)。golang對(duì)struct 的方法調(diào)用是這樣的

defer func (this Type, para) result

struct 的方法在定義時(shí)this 采用的值類型還是指針決定defer在調(diào)用時(shí)首個(gè)參數(shù)(隱藏)的復(fù)制的不同。

上述代碼中Point方法定義時(shí)使用了指針作為this，因此輸出是for循環(huán)最終賦值的t引用。

Value方法定義時(shí)使用值類型作為this，因此輸出是for循環(huán)執(zhí)行的每步復(fù)制后的t。

最終只有理解了以上出現(xiàn)的問(wèn)題，到此defer所帶來(lái)困擾才能遠(yuǎn)離我們。

補(bǔ)充：Golang中defer的三個(gè)實(shí)戰(zhàn)要點(diǎn)

Golang中的defer是使用頻次比較高的，能創(chuàng)造出延遲生效特效的一種方式。

defer也有自己的矯情，需要注意的。

本文將從通過(guò)代碼的方式來(lái)說(shuō)明defer的三點(diǎn)矯情。

1.defer的生效順序

先說(shuō)結(jié)論：defer的執(zhí)行順序是倒序執(zhí)行（同入棧先進(jìn)后出）

func main() {
 defer func() {
  fmt.Println("我后出來(lái)")
 }()
 defer func() {
  fmt.Println("我先出來(lái)")
 }()
}

執(zhí)行后打印出：

我先出來(lái)

我后出來(lái)

2.defer與return,函數(shù)返回值之間的順序

先說(shuō)結(jié)論：return最先執(zhí)行->return負(fù)責(zé)將結(jié)果寫入返回值中->接著defer開始執(zhí)行一些收尾工作->最后函數(shù)攜帶當(dāng)前返回值退出

返回值的表達(dá)方式，我們知道根據(jù)是否提前聲明有兩種方式：一種是func test() int 另一種是 func test() (i int)，所以兩種情況都來(lái)說(shuō)說(shuō)

func test() int
func main() {
 fmt.Println("main:", test())
}
func test() int {
 var i int
 defer func() {
  i++
  fmt.Println("defer2的值:", i)
 }()
 defer func() {
  i++
  fmt.Println("defer1的值:", i)
 }()
 return i
}

輸出：

defer1的值: 1

defer2的值: 2

main: 0

詳解：return的時(shí)候已經(jīng)先將返回值給定義下來(lái)了，就是0，由于i是在函數(shù)內(nèi)部聲明所以即使在defer中進(jìn)行了++操作，也不會(huì)影響return的時(shí)候做的決定。

func test() (i int)
func main() {
 fmt.Println("main:", test())
}
func test() (i int) {
 defer func() {
  i++
  fmt.Println("defer2的值:", i)
 }()
 defer func() {
  i++
  fmt.Println("defer1的值:", i)
 }()
 return i
}

輸出：

defer1的值: 1

defer2的值: 2

main: 2

詳解：由于返回值提前聲明了，所以在return的時(shí)候決定的返回值還是0，但是后面兩個(gè)defer執(zhí)行后進(jìn)行了兩次++，將i的值變?yōu)?，待defer執(zhí)行完后，函數(shù)將i值進(jìn)行了返回。

3.defer定義和執(zhí)行兩個(gè)步驟，做的事情

先說(shuō)結(jié)論：會(huì)先將defer后函數(shù)的參數(shù)部分的值(或者地址)給先下來(lái)【你可以理解為()里頭的會(huì)先確定】，后面函數(shù)執(zhí)行完，才會(huì)執(zhí)行defer后函數(shù)的{}中的邏輯

func test(i *int) int {
 return *i
}
func main(){
 var i = 1
 // defer定義的時(shí)候test(&i)的值就已經(jīng)定了，是1，后面就不會(huì)變了
 defer fmt.Println("i1 ="  , test(&i))
 i++
 // defer定義的時(shí)候test(&i)的值就已經(jīng)定了，是2，后面就不會(huì)變了
 defer fmt.Println("i2 ="  , test(&i))
 // defer定義的時(shí)候，i就已經(jīng)確定了是一個(gè)指針類型，地址上的值變了，這里跟著變
 defer func(i *int) {
  fmt.Println("i3 ="  , *i)
 }(&i)
 // defer定義的時(shí)候i的值就已經(jīng)定了，是2，后面就不會(huì)變了
 defer func(i int) {
  //defer 在定義的時(shí)候就定了
  fmt.Println("i4 ="  , i)
 }(i)
 defer func() {
  // 地址，所以后續(xù)跟著變
  var c = &i
  fmt.Println("i5 ="  , *c)
 }()
 
 // 執(zhí)行了 i=11 后才調(diào)用，此時(shí)i值已是11
 defer func() {
  fmt.Println("i6 ="  , i)
 }()
 i = 11
}

以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章: