关于 Go 的 for Range 上的一个坑

range作为go的一个语法糖在进行迭代的时候是很方便的, 但6是传址赋值的时候需要小心一点

出现问题

直接上代码:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
package main

import "fmt"

func main() {
slice := []int{0, 1, 2, 3}
myMap := make(map[int]*int)

for index, value := range slice {
fmt.Printf("value's memery address: %v | value's value: %v \n", &value, value)
myMap[index] = &value
}
fmt.Println("=====new map by range=====")
fmt.Printf("result: %v \n", myMap)
prtMap(myMap)

myMap2 := make(map[int]*int)
for i := 0; i < len(slice); i++ {
myMap2[i] = &slice[i]
}
fmt.Println("=====new map by for=====")
fmt.Printf("result: %v \n", myMap2)
prtMap(myMap2)

}

func prtMap(myMap map[int]*int) {
for key, value := range myMap {
fmt.Printf("map[%v]=%v\n", key, *value)
}
}

输出:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
value's memery address: 0xc0000160a8 | value's value: 0
value's memery address: 0xc0000160a8 | value's value: 1
value's memery address: 0xc0000160a8 | value's value: 2
value's memery address: 0xc0000160a8 | value's value: 3
=====new map by range=====
result: map[0:0xc0000160a8 1:0xc0000160a8 2:0xc0000160a8 3:0xc0000160a8]
map[1]=3
map[2]=3
map[3]=3
map[0]=3
i's memery address: 0xc000016120 | value: 0
i's memery address: 0xc000016120 | value: 1
i's memery address: 0xc000016120 | value: 2
i's memery address: 0xc000016120 | value: 3
=====new map by for=====
result: map[2:0xc00001c110 3:0xc00001c118 0:0xc00001c100 1:0xc00001c108]
map[2]=2
map[3]=3
map[0]=0
map[1]=1

猜测原因

rangefor 进行赋值操作只执行了一次, 因此在上面那个例子中 valuei的内存地址在循环的时候都没变过, 在进行值拷贝的时候是没问题的, 但是一旦加上 & 的时候就变得很危险了, 造成的结果就是上面那个例子中的问题, 新数组存的都是同一个内存地址, 在 for 循环的时候因为只是利用的索引, 没有直接用到切片的值而不会引发这些问题, 在 range 的时候会出现 key,value 同时出现,那么就要小心处理了, 因为这里的value并不是切片中的原值, 而是一个拷贝出来的

Go Range 内部实现
go语言坑之for range

类似的问题
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
package main

import (
"fmt"
"sync"
)

func main() {
wg := sync.WaitGroup{}
for i := 0; i < 4; i++ {
wg.Add(1)
go func() {
fmt.Printf("i is %d \n", i)
wg.Done()
}()
}
wg.Wait()
}
1
输出 4,4,4,4

这个和上一个的问题还不一样, 这个就是闭包的执行顺序以及变量作用域的问题, 在执行闭包的时候for循环已经执行完毕,此时i=4, 接下来才会执行闭包内的函数, 规避这个问题也简单, 把变量的作用域改一下就行, 还有就是函数参数的值拷贝特点

1
2
3
4
5
6
7
for i := 0; i < 4; i++ {
wg.Add(1)
go func(i int) {
fmt.Printf("i is %d \n", i)
wg.Done()
}(i)
}

闭包和变量作用域