深圳幻海软件技术有限公司 欢迎您!

Go 语言使用 goroutine 运行闭包的“坑”

2023-02-28

​01 介绍在Go语言中,函数支持匿名函数,闭包就是一种特殊的匿名函数,它可以用于访问函数体外部的变量。需要注意的是,在 for...range... 中,使用goroutine执行闭包时,经常会掉“坑”。因为匿名函数可以访问函数体外部的变量,而 for...

​01 介绍

在 Go 语言中,函数支持匿名函数,闭包就是一种特殊的匿名函数,它可以用于访问函数体外部的变量。

需要注意的是,在 for ... range ... 中,使用 goroutine 执行闭包时,经常会掉“坑”。

因为匿名函数可以访问函数体外部的变量,而 for ... range ... 返回的 val 的值是引用的同一个内存地址的数据,所以匿名函数访问的函数体外部的 val 值是循环中最后输出的一个值。

02 “踩坑”示例代码

func main() {
 done := make(chan bool)

 values := []string{"a", "b", "c"}
 for _, v := range values {
  go func() {
   fmt.Println(v)
   done <- true
  }()
 }

 // wait for all goroutines to complete before exiting
 for _ = range values {
  <-done
 }
}
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.

输出结果:

c
c
c
  • 1.
  • 2.
  • 3.

阅读上面这段代码,在 for ... range ... 中,使用 goroutine 执行闭包,打印切片中的元素,实际输出结果不是我们期望得到的输出结果。

这是因为循环的每次迭代都使用相同的变量 v 实例,因此每个闭包共享该单个变量。我们可以在示例代码中简单修改,同时输出变量 v 的内存地址和值。

把 fmt.Println(v)​ 修改为 fmt.Printf("val=%s pointer=%p\n", v, &v)。

修改后的输出结果:

val=c pointer=0xc000010200
val=c pointer=0xc000010200
val=c pointer=0xc000010200
  • 1.
  • 2.
  • 3.

我们可以在输出结果中发现,打印变量 v 的内存地址都是 0xc000010200。

当闭包运行时,它会在执行 fmt.Println​ 时打印变量 v 的值,但 v 的值可能在 goroutine 启动后已被修改。感兴趣的读者朋友们可以使用 go vet 检查。

怎么避免“踩坑”呢?

一种方法是将变量作为参数传递给闭包:

func main() {
 done := make(chan bool)

 values := []string{"a", "b", "c"}
 for _, v := range values {
  go func(param string) {
   // fmt.Println(v)
   fmt.Printf("val=%s pointer=%p\n", param, &param)
   done <- true
  }(v)
 }

 // wait for all goroutines to complete before exiting
 for _ = range values {
  <-done
 }
}
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.

输出结果:

val=c pointer=0xc000010200
val=a pointer=0xc00009a000
val=b pointer=0xc0000a4000
  • 1.
  • 2.
  • 3.

阅读上面这段代码,通过将变量 v 的值作为参数传递给闭包,然后,该值作为形参 param 的值,在函数体内部被访问。

另外一种方法是创建一个新变量:

func main() {
 done := make(chan bool)

 values := []string{"a", "b", "c"}
 for _, v := range values {
  param := v
  go func() {
   // fmt.Println(v)
   fmt.Printf("val=%s pointer=%p\n", param, &param)
   done <- true
  }()
 }

 // wait for all goroutines to complete before exiting
 for _ = range values {
  <-done
 }
}
  • 1.
  • 2.
  • 3.
  • 4.
  • 5.
  • 6.
  • 7.
  • 8.
  • 9.
  • 10.
  • 11.
  • 12.
  • 13.
  • 14.
  • 15.
  • 16.
  • 17.
  • 18.

输出结果:

val=c pointer=0xc000082200
val=a pointer=0xc0000821e0
val=b pointer=0xc0000821f0
  • 1.
  • 2.
  • 3.

通过输出结果可以发现,该种方式也可以达到我们期望的结果。

03 总结

本文我们介绍了在 for ... range ... 中,Go 语言在每次迭代时,没有定义一个新变量,导致使用 goroutine 运行闭包时,经常会掉“坑”。

我们给出避免“踩坑”的两种方法,其中,第二种方法更简单。