golang for range遍历并赋值给字典后出现所有值相同的问题

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

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

参考

1. https://studygolang.com/articles/9701
2. https://draveness.me/golang/docs/part2-foundation/ch05-keyword/golang-for-range/