go語言中匿名函數(shù)的作用域陷阱詳解

更新時間：2022年05月06日 10:52:02 作者：Timo

GO語言的匿名函數(shù)（anonymous?function),其實就是閉包.是指不需要定義函數(shù)名的一種函數(shù)實現(xiàn)方式,下面這篇文章主要給大家介紹了關(guān)于go語言中匿名函數(shù)作用域陷阱的相關(guān)資料,需要的朋友可以參考下

眾所周知在go語言中函數(shù)也可以當作變量在程序中使用，我們可以使用函數(shù)字面量在任何表達式內(nèi)指定函數(shù)變量。但是在編寫代碼的時候請注意：如果一個函數(shù)在使用不是在該函數(shù)內(nèi)部定義的變量時，這個變量的生命周期不是由其作用域決定的！

這段話什么意思呢，借鑒go語言圣經(jīng)中的一段代碼舉個例子：

func square() func() int { //返回一個自己定義的函數(shù)類型
   var x int = 0
   return func() int {
      x++
      return x
   }
}
func main() {
   f := square()     
   fmt.Println(f()) // 1
   fmt.Println(f()) // 2
   fmt.Println(f()) // 3
   fmt.Println(f()) // 4
}

在這段程序中，函數(shù)square返回了一個類型為func() int的函數(shù)類型，但在每次調(diào)用f()的時候，匿名函數(shù)內(nèi)部都會在x原有的值的基礎上+1來更新x的值，而不是把x值賦值為0，所有這段代碼最終的執(zhí)行結(jié)果為1 2 3 4。

這表明里層的匿名函數(shù)能獲取和更新外層的square函數(shù)的局部變量，變量x在main函數(shù)中返回square函數(shù)后依然存在。我們可以將其類比為全局變量和函數(shù)的關(guān)系來理解：把square函數(shù)看作一個獨立的go程序，那么局部變量x可視為在square函數(shù)內(nèi)部的全局變量，顯然匿名函數(shù)每次操作的都是同一個變量。我們加上變量的地址輸出驗證是否正確：

func square() func() int { //返回一個自己定義的函數(shù)類型
   var x int = 0
   return func() int {
      x++
      fmt.Println(&x)
      return x
   }
}
func main() {
   f := square()
   fmt.Println(f()) 
   fmt.Println(f()) 
   fmt.Println(f()) 
   fmt.Println(f()) 
}

結(jié)果：

可見調(diào)用的變量x的確是同一個變量，那么我們在捕獲迭代變量時使用匿名函數(shù)的時候就需要小心，看下面這段程序：

func square(x int) int { 
   fmt.Println(x)
   return x
}
func main() {
   var handlers []func()
   for i := 0; i < 10; i++ {
      handlers = append(handlers, func() {
         square(i)
      })
   }
   for _, handler := range handlers {
      handler()
   }
}

在這段程序中，變量i在for循環(huán)引進的一個塊作用域進行聲明，根據(jù)前面的經(jīng)驗我們可以知道：在循環(huán)里創(chuàng)建的所有函數(shù)變量共享相同的變量，i的值在迭代中不斷更新，因此i的實際取值是循環(huán)中i的最后一個取值，并且所有的square函數(shù)都在處理同一個值，運行結(jié)果也可以證明這一點：

我們經(jīng)常引入一個內(nèi)部變量來解決這個問題，聲明一個副本，并將這個副本的值傳入而不是直接把i的值傳入：

func square(x int) int {
   fmt.Println(x)
   return x * x
}
func main() {
   var handlers []func()
   for i := 0; i < 10; i++ {
      j := i
      handlers = append(handlers, func() {
         square(j)
      })
   }
   for _, handler := range handlers {
      handler()
   }
}

這樣就得到了我們想要的運行結(jié)果：