for range遍历并赋值给字典后出现所有值相同的问题
先看这段(非常经典的)错误示例:(为了方便用json包打印输出结构体,我使用了大写字母开头的导出字段)
m := make(map[string]*student)
stus := []student{
{Name: "名字1", Age: 1},
{Name: "名字2", Age: 2},
{Name: "名字3", Age: 3},
}
for _, stu := range stus {
m[stu.Name] = &stu
}
for k, v := range m {
fmt.Println(k, ":", v.Name)
}
输出结果:
名字3 : 名字3
名字1 : 名字3
名字2 : 名字3
- 如果在循环中打印
stu
,会发现每次遍历时stu的输出都是正常的,也就是确实遍历到了数组相应位置的元素 - 但是遍历到下一个元素时,上一个元素在字典中对应的值也将变为本次遍历元素的值
- 最后所有键mapping的值都为同一个——也即最后遍历到的元素
for range 是创建了每个元素的副本然后映射该副本,并不是返回对元素的引用,而且使用的一直是同一个遍历指针,这在循环读取中是没有问题的。但如果循环建立哈希表,使用遍历的指针作为每个键的映射,那么循环结束后会指向最后一个元素,造成所有的值都是同一个。如果我们在其中加入打印
for _, stu := range stus {
m[stu.Name] = &stu
fmt.Println(m)
}
输出
map[名字1:0xc00000c060 名字2:0xc00000c060 名字3:0xc00000c060]
也表明了这几个key所对应的是同一个内存地址。那么自然字典中的值都是同一个了。
解决方案有两种:
- 使用数组索引来代替遍历指针
for i, v := range stus {
m[v.Name] = &stus[i]
}
输出
名字1 : 名字1
名字2 : 名字2
名字3 : 名字3
- 创建新的变量
for _, stu := range stus {
st := stu
m[stu.Name] = &st
}
输出也是
名字1 : 名字1
名字2 : 名字2
名字3 : 名字3
参考
- https://studygolang.com/articles/9701
- https://draveness.me/golang/docs/part2-foundation/ch05-keyword/golang-for-range/