defer的這些坑，你遇到過嗎？

1：延遲函數(shù)傳遞的參數(shù)是值

func deferTest() {
  var a = 1
  defer fmt.Println(a)
  
  a = 2
  return
}

結(jié)論：延遲函數(shù) fmt.Println(a) 的參數(shù)在 defer 語句出現(xiàn)的時(shí)候就已經(jīng)確定下來了，所以不管后面如何修改 a 變量，都不會(huì)影響延遲函數(shù)

2：延遲函數(shù)傳遞的參數(shù)是地址

func deferTest() {
  var arr = [3]int{1, 2, 3}
  defer printTest(&arr)


  arr[0] = 4
  return
}


func printTest(array *[3]int) {
  for i := range array {
    fmt.Println(array[i])
  }
}

結(jié)論：延遲函數(shù) printTest() 的參數(shù)在 defer 語句出現(xiàn)的時(shí)候就已經(jīng)確定下來了，即為數(shù)組的地址，延遲函數(shù)執(zhí)行的時(shí)機(jī)是在 return 語句之前，所以對(duì)數(shù)組的最終修改的值會(huì)被打印出來。

3：延遲函數(shù)可能會(huì)影響函數(shù)的返回值

fmt.Println(deferTest)


func deferTest() (result int) {
  i := 1
  defer func() {
    result = 2
  }()
  return i
}

結(jié)論：函數(shù)的 return 語句并不是原子級(jí)的，實(shí)際的執(zhí)行過程為為設(shè)置返回值—>ret，defer 語句是在返回前執(zhí)行，所以返回過程是：「設(shè)置返回值—>執(zhí)行defer—>ret」。所以 return 語句先把 result 設(shè)置成 i 的值（1），defer 語句中又把 result設(shè)置為 2 ，所以最終返回值為 2

4：defer需要定義在panic前

func panicBeforeDefer() {
  panic("a")
  defer func() {
    fmt.Println("b")
  }()
}


func panicAfterDefer() {
  defer func() {
    fmt.Println("b")
  }()
  panic("a")
}

結(jié)論：代碼執(zhí)行到了painc之后再執(zhí)行的defer，然后按照defer的先進(jìn)后出的順序執(zhí)行defer,最后才執(zhí)行panic。那為什么panic時(shí)會(huì)執(zhí)行defer，可以看下這段代碼就很清楚了。

func gopanic(e interface{}) {
  gp := getg() 
  ...
  
  var p _panic
  p.arg = e
  p.link = gp._panic
  gp._panic = (*_panic)(noescape(unsafe.Pointer(&p)))


  
  for {
    
    d := gp._defer
    if d == nil {
      break
    }
      ...
  }
}

5：先判斷err，再defer釋放資源

func openFile() {
  file, err := os.Open("txt")
  if err != nil {
    return
  }
  defer file.Close()
}

結(jié)論：獲取文件資源的時(shí)候會(huì)返回err，如果我們?cè)诤罄m(xù)需要進(jìn)行defer釋放文件資源時(shí)，這里需要對(duì)err進(jìn)行判斷。因?yàn)槿绻@取文件資源失敗的時(shí)候不需要進(jìn)行釋放，也避免了沒獲取到資源可能導(dǎo)致的釋放函數(shù)執(zhí)行錯(cuò)誤。

6：os.Exit時(shí)defer不會(huì)被執(zhí)行

func deferExit() {
  defer func() {
    fmt.Println("exit")
  }()
  os.Exit(1)
}

結(jié)論:上面代碼中的defer不會(huì)得到執(zhí)行，因?yàn)閛s.Exit()用于立即中止程序，不可能恢復(fù)或運(yùn)行延遲清理語句，不像panic會(huì)去找goroutine的defer鏈表。

審核編輯：劉清