go语言闭包陷阱:循环变量的正确使用方法
本文解释Go语言中一个常见的闭包问题,以及如何避免它。以下代码片段展示了这个问题:
package main import "fmt" func main() { var fs = [4]func(){} var fi = [4]int{} for i := 0; i < 4; i++ { fs[i] = func() { fi[i] = i fmt.Println("closure i =", i) } } for _, f := range fs { f() } fmt.Println(fi) }
这段代码的输出结果是:
closure i = 4 closure i = 4 closure i = 4 closure i = 4 [0 0 0 4]
许多开发者对i的值始终为4感到困惑。这是因为Go语言闭包的特性:闭包引用的是变量,而不是变量的值。在循环中,所有匿名函数都引用了同一个i变量。当循环结束后,i的值为4,所以所有匿名函数执行时,访问到的i都是4。
为了解决这个问题,需要为每个迭代创建新的变量来保存i的当前值。修改后的代码如下:
package main import "fmt" func main() { var fs = [4]func(){} var fi = [4]int{} for i := 0; i < 4; i++ { j := i // 创建新的变量j fs[i] = func() { fi[i] = j fmt.Println("closure j =", j) } } for _, f := range fs { f() } fmt.Println(fi) }
通过引入新的变量j,每个匿名函数都拥有自己独立的j变量,保存了i在每次迭代时的值。修改后的代码输出结果为:
closure j = 0 closure j = 1 closure j = 2 closure j = 3 [0 1 2 3]
现在,每个匿名函数都打印出了它创建时i的正确值。 记住,在Go语言闭包中,要避免直接使用循环变量,而是为每次迭代创建一个新的变量副本。