go 语言是支持函数式编程的,在刚刚开始学习go开发圣经的时候遇到过这样一个关于函数式编程(函数值)的问题。
go 语言陷阱:捕获迭代变量
go 开发圣经(5.6节)
这个问题的起因是值传递与引用传递的问题:
考虑这个样一个问题:你被要求首先创建一些目录,再将目录删除。在下面的例子中我们用函数值来完成删除操作。下面的示例代码需要引入os包。为了使代码简单,我们忽略了所有的异常处理。
var rmdirs []func()
for _, d := range tempDirs() {
dir := d // NOTE: necessary!
os.MkdirAll(dir, 0755) // creates parent directories too
rmdirs = append(rmdirs, func() {
os.RemoveAll(dir)
})
}
// ...do some work…
for _, rmdir := range rmdirs {
rmdir() // clean up
}
局部变量:dir的声明是必须的。
for循环语句引入了新的词法块,循环变量dir在这个词法块中被声明。在该循环中生成的所有函数值都共享相同的循环变量。此时往rmdirs中添加函数值。 需要注意,函数值(这里指的是append中添加的匿名函数)中记录的是循环变量的内存地址,而不是循环变量某一时刻的值。以dir为例,后续的迭代会不断更新dir的值,当删除操作执行时,for循环已完成,dir中存储的值等于最后一次迭代的值。这意味着,每次对os.RemoveAll的调用删除的都是相同的目录。
下面的循环中,也存在同样的问题:
var rmdirs []func()
dirs := tempDirs()
for i := 0; i < len(dirs); i++ {
os.MkdirAll(dirs[i], 0755) // OK
rmdirs = append(rmdirs, func() {
os.RemoveAll(dirs[i]) // NOTE: incorrect!
})
}
此外,在range过程中,使用新的go routine, 这种情况也会发生类似的错误,举例如下:
package main
import (
"fmt"
"time"
)
func main() {
p := make(chan int, 10)
go func() {
for i := 0; i < 100; i++ {
p <- i
}
}()
for i := range p {
go func() {
time.Sleep(1 * time.Second)
// 会一直打印99
fmt.Println(i)
}()
}
fmt.Println("vim-go")
}
可以采用这样的方法进行修改:
package main
import (
"fmt"
"time"
)
func main() {
p := make(chan int, 10)
go func() {
for i := 0; i < 100; i++ {
p <- i
}
}()
for i := range p {
go func(m int) {
time.Sleep(1 * time.Second)
fmt.Println(m)
}(i)
}
fmt.Println("vim-go")
}
或者
package main
import (
"fmt"
"time"
)
func main() {
p := make(chan int, 10)
go func() {
for i := 0; i < 100; i++ {
p <- i
}
}()
for i := range p {
m := i
go func() {
time.Sleep(1 * time.Second)
fmt.Println(m)
}()
}
fmt.Println("vim-go")
}