Go语言闭包陷阱及解决方法
在Go语言中，闭包经常会遇到一个棘手的循环变量问题。让我们通过一个例子来解释：
以下代码片段演示了这个问题：
package main
import (
"fmt"
"sync"
"time"
)
var numbers = [...]int{1, 2, 3, 4, 5}
func run() {
fmt.Println("开始")
var wg sync.WaitGroup
wg.Add(len(numbers))
for _, num := range numbers {
go func() {
fmt.Printf("获取值: %d\n", num)
time.Sleep(time.Second)
wg.Done()
}()
}
wg.Wait()
fmt.Println("结束")
}
func main() {
run()
}
运行这段代码，你会发现输出结果并非预期的1, 2, 3, 4, 5，而是多次重复输出最后一个值5。
问题根源
造成这个问题的原因在于Go语言的闭包特性以及goroutine的并发执行：
-
变量作用域: 循环变量
num并非每个goroutine的局部变量，所有goroutine都共享同一个num变量。 -
并发执行:
for循环的执行速度远快于goroutine的启动和执行。当所有goroutine启动后，for循环已完成，num的值已经变成了最后一个值5。 每个goroutine最终访问到的num都是循环结束后num的最终值。
解决方案
为了解决这个问题，我们需要为每个goroutine创建一个num的副本。修改后的代码如下：
...
for _, num := range numbers {
go func(n int) {
fmt.Printf("获取值: %d\n", n)
time.Sleep(time.Second)
wg.Done()
}(num) // 传递num的副本
}
...
通过将num作为参数传递给匿名函数，每个goroutine都拥有了num的一个独立副本，从而避免了变量共享导致的错乱问题。 这样，程序就能正确地输出1, 2, 3, 4, 5。
通过这个例子，我们了解了Go语言闭包中循环变量可能出现的陷阱，以及如何通过传递参数副本的方式来有效解决这个问题。 记住，在使用闭包和goroutine时，务必注意变量作用域和并发执行的顺序，以避免类似问题的发生。
