0%

Go语言中for range的"坑"

前言

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)
}
}

代码解析:

  1. 创建一个int slice,变量名为arr1并初始化 1,2,3 作为切片的值。
  2. 创建一个*int slice,变量名为arr2
  3. 通过for range遍历arr1,然后获取每一个元素的指针,赋值到对应arr2中。
  4. 逐行打印arr2中每个元素的值。

从代码上看,打印出来的结果应该是

1
2
3
1
2
3

然而真正的结果是

1
2
3
3
3
3

原因

因为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
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. 传递原始指针
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. 使用临时变量
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. 使用闭包
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甚至将其写入了文档的『常见错误』部分:文档


我是MonkeyWie,欢迎扫码👇👇关注!不定期在公众号中分享JAVAGolang前端dockerk8s等干货知识。

如果觉得本文对您有帮助,可以请我喝一杯咖啡☕