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