前言
Go 中的for range
组合可以和方便的实现对一个数组或切片进行遍历,但是在某些情况下使用for range
时很可能就会被"坑"
,下面用一段代码来模拟下:
1 2 3 4 5 6 7 8 9 10 11 12
| func main() { arr1 := []int{1, 2, 3} arr2 := make([]*int, len(arr1))
for i, v := range arr1 { arr2[i] = &v }
for _, v := range arr2 { fmt.Println(*v) } }
|
代码解析:
- 创建一个
int slice
,变量名为arr1
并初始化 1,2,3 作为切片的值。 - 创建一个
*int slice
,变量名为arr2
。 - 通过
for range
遍历arr1
,然后获取每一个元素的指针,赋值到对应arr2
中。 - 逐行打印
arr2
中每个元素的值。
从代码上看,打印出来的结果应该是
然而真正的结果是
原因
因为for range
在遍历值类型
时,其中的v
变量是一个值
的拷贝,当使用&
获取指针时,实际上是获取到v
这个临时变量的指针,而v
变量在for range
中只会创建一次,之后循环中会被一直重复使用,所以在arr2
赋值的时候其实都是v
变量的指针,而&v
最终会指向arr1
最后一个元素的值拷贝。
来看看下面这个代码,用for i
来模拟for range
,这样更易于理解:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
| func main() { arr1 := []int{1, 2, 3} arr2 := make([]*int, len(arr1))
var v int for i:=0;i<len(arr1);i++ { v = arr1[i] arr2[i] = &v }
for _, v := range arr2 { fmt.Println(*v) } }
|
解决方案
- 传递原始指针
1 2 3 4 5 6 7 8 9 10 11 12
| func main() { arr1 := []int{1, 2, 3} arr2 := make([]*int, len(arr1))
for i := range arr1 { arr2[i] = &arr1[i] }
for _, v := range arr2 { fmt.Println(*v) } }
|
- 使用临时变量
1 2 3 4 5 6 7 8 9 10 11 12 13
| func main() { arr1 := []int{1, 2, 3} arr2 := make([]*int, len(arr1))
for i, v := range arr1 { t := v arr2[i] = &t }
for _, v := range arr2 { fmt.Println(*v) } }
|
- 使用闭包
1 2 3 4 5 6 7 8 9 10 11 12 13 14
| func main() { arr1 := []int{1, 2, 3} arr2 := make([]*int, len(arr1))
for i, v := range arr1 { func(v int){ arr2[i] = &v }(v) }
for _, v := range arr2 { fmt.Println(*v) } }
|
官方提示
由于这一问题过于普遍,Golang甚至将其写入了文档的『常见错误』部分:文档