Go語言中defer使用的陷阱小結

更新時間：2024年01月21日 15:48:36 作者：FrankGopher

本文主要介紹了Go語言中defer使用的陷阱小結,分別是defer語句不可以在return語句之后,defer語句執(zhí)行的匿名函數(shù),匿名函數(shù)的參數(shù)會被預先處理,具有一定的參考價值,感興趣的可以了解一下

01 介紹

什么是 defer

defer 是Go語言提供的一種用于注冊延遲調用的機制，以用來保證一些資源被回收和釋放。

defer 注冊的延遲調用可以在當前函數(shù)執(zhí)行完畢后執(zhí)行（包括通過return正常結束或者panic導致的異常結束）

當defe注冊了的函數(shù)或表達式逆序執(zhí)行，先注冊的后執(zhí)行，類似于棧 ”先進后出“

下面看一個例子：

package main

import "fmt"

func main() {
   f()
}

func f() {
   defer func() {
      fmt.Println(1)
   }()
   defer func() {
      fmt.Println(2)
   }()
   defer func() {
      fmt.Println(3)
   }()
}

輸出：

3
2
1

如何使用defer

釋放資源

使用 defer 可以在一定程度上避免資源泄漏，尤其是有很多 return 語句的場景，很容易忘記或者由于邏輯上的錯誤導致資源沒有關閉。

下面的程序便是因為使用 return 后，關閉資源的語句沒有執(zhí)行，導致資源泄漏：

f, err := os.Open("test.txt")
if err != nil {
   return
}
f.process()
f.Close()

此處更好的做法如下：

f, err := os.Open("test.txt")
if err != nil {
   return
}
defer f.Close()

// 對文件進行操作
f,process()

此處當程序順利執(zhí)行后，defer 會釋放資源；defer 需要先注冊后使用，比如此處，打開文件異常時，程序執(zhí)行到 return 語句時便會退出當前函數(shù)，沒有經過 defer，所以此處defer 不會執(zhí)行

defer 捕獲異常

在 go 中沒有 try 和 catch , 當程序出現(xiàn)異常是，我們需要從異常中恢復。我們這時可以利用 defer + recover 進行異常捕獲

func f() {
    defer func() {
       if err := recover(); err != nil {
          fmt.Println(err)
       }
    }()
    // do something
    panic("panic")
}

注意，recover() 函數(shù)在在defer中用匿名函數(shù)調用才有效，以下程序不能進行異常捕獲：

func f() {
    if err := recover(); err != nil {
        fmt.Println(err)
    }
    //  do something
    panic("panic")
}

實現(xiàn)代碼追蹤

下面提供一個方法能追蹤到程序時進入或離開某個函數(shù)的信息，此處可以用來測試特定函數(shù)有沒有被執(zhí)行

func trace(msg string) { fmt.Println("entering:", msg) }
func untrace(msg string) { fmt.Println("leaving:", msg) }

記錄函數(shù)的參數(shù)與返回值

有時候程序返回結果不符合預期是，大家可能手動打印 log 調試，此時使用 defer 記錄函數(shù)的參數(shù)和返回值，避免手動多處打印調試語句

func func1(s string) (n int, err error) {
    defer func() {
        log.Printf("func1(%q) = %d, %v", s, n, err)
    }()
    return 7, nil
}

實現(xiàn)代碼追蹤和記錄函數(shù)的參數(shù)與返回值

在 Go 語言中，defer 一般用于資源釋放，或使用 defer 調用一個匿名函數(shù)，在匿名函數(shù)中使用 recover() 處理異常 panic。

在使用 defer 時，也很容易遇到陷阱，本文我們介紹使用 defer 時有哪些陷阱。

02 defer 陷阱

defer 語句不可以在 return 語句之后。

示例代碼：

func main() {
    name := GetUserName("phper")
    fmt.Printf("name:%s\n", name)
    if name != "gopher" {
        return
    }
    defer fmt.Println("this is a defer call")
}

func GetUserName(name string) string {
    return name
}

輸出結果：

name:phper

閱讀上面這段代碼，我們在 return 語句之后執(zhí)行 defer 語句，通過輸出結果可以發(fā)現(xiàn) defer 語句調用未執(zhí)行。

雖然 defer 可以在函數(shù)體中的任意位置，我們也是需要特別注意使用 defer 的位置是否可以執(zhí)行。

defer 語句執(zhí)行匿名函數(shù)，參數(shù)預處理。

示例代碼：

func main() {
    var count int64
    defer func(data int64) {
        fmt.Println("defer:", data)
    }(count + 1)
    count = 100
    fmt.Println("main:", count)
}

輸出結果：