for range 遍历是go语言中常用的循环结构之一,在使用循环赋值时有时候需要注意指针的引用问题。在探讨之前,先让我们来回顾下Go的指针。

Go 指针回顾

Go 语言中有指针类型,没有指针的计算,这在一定程度上削弱了指针的功能,但也减少了指针的复杂度,给使用者带了更好的使用体验。在Go 中,类型 *T 是指向类型 T 的值的指针, & 符号会生成一个指向其作用对象的指针, * 符号表示指针指向的底层的值。如下:

var p *int      // 定义一个指针p 
i := 42         // 初始化一个整数类型 i 
p = &i          // p指针指向i,或者说将 i 的指针赋值给p  
fmt.Println(*p) // 通过指针 p 读取 i
*p = 21         // 通过指针 p 设置 i

可参考如下图理解:

有时候当变量值很大,在函数中传来传去,就会占用大量的内存空间。指针中存了指向变量值的地址,在一定程度上可以使用指针来代替变量值,这样在函数参数传递、各种逻辑计算中,就可以节省大量空间。

for range 遍历取值的问题

我们来看一段示意代码,代码中在 append 赋值的时候为了节省内存使用了指针。猜猜遍历结果集是什么?

package main

import "fmt"

type Persion struct {
    name string
}

func main() {
    arr := []Persion{
        Persion{"小明"},
        Persion{"小刚"},
    }
    var res []*Persion

    for _, v := range arr {
        res = append(res, &v)
    }
    // 遍历查看结果集
    for _, persion := range res{
        fmt.Println("name-->:", persion.name)
    }
}

看结果:

name-->: 小刚
name-->: 小刚

可以看到在遍历 arr 这个数组时,后边的值把前边的覆盖了。让我们加点打印信息:

···
    for _, v := range arr {
        fmt.Printf("v 指针 %p\n", &v)
        fmt.Println("v 的值", v)
        res = append(res, &v)
  }
  fmt.Println(res)
···

输出如下:

v 指针 0xc0001101e0
v 的值 {小明}
v 指针 0xc0001101e0
v 的值 {小刚}
[0xc0001101e0 0xc0001101e0]

可以看到在结果集中的指针是一样的,也就是说在 for range 的时候,v 只 初始化了一次 ,之后的遍历都是在原来遍历的基础上赋值,所有v的指针(地址)并没有变。该指针指向的是最后一次遍历的v的值,所以最后结果集中,也就都成了最后遍历的v的值。

这里正确的做法是使用下标。如下:

for i, _ := range arr {
        fmt.Printf("v 指针 %p\n", &arr[i])
        fmt.Println("v 的值", arr[i])
        res = append(res, &arr[i])
    }

结果:

v 指针 0xc0000a6020
v 的值 {小明}
v 指针 0xc0000a6030
v 的值 {小刚}
[0xc0000a6020 0xc0000a6030]
name-->: 小明
name-->: 小刚

总结

Go 语言中的指针在一定程度上可以节省内存的占用,但在使用时一定要注意前后语言环境中是否有变动,以免造成引用同值的锅。

转自:https://zhuanlan.zhihu.com/p/325916244