Go语言中for range的"坑"

前言

Go 中的for range组合能够和方便的实现对一个数组或切片进行遍历，可是在某些状况下使用for range时极可能就会被"坑"，下面用一段代码来模拟下：

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

然而真正的结果是

3
3
3

缘由

由于for range在遍历值类型时，其中的v变量是一个值的拷贝，当使用&获取指针时，其实是获取到v这个临时变量的指针，而v变量在for range中只会建立一次，以后循环中会被一直重复使用，因此在arr2赋值的时候其实都是v变量的指针，而&v最终会指向arr1最后一个元素的值拷贝。

来看看下面这个代码，用for i来模拟for range，这样更易于理解:

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. 传递原始指针

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. 使用临时变量

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. 使用闭包

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，欢迎扫码👇👇关注！不按期在公众号中分享 JAVA、 Golang、 前端、 docker、 k8s等干货知识。