go语言range的坑

go语言中使用range遍历slice、map等给我很大的便利。

但是当我使用range 修改值，可能会发生你意想不到的结果。

下面我们就来看下一段代码：

package main

import (
	"fmt"
)

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

	for _,v :=range slice{
		if v==1 {
			v=100
		}
	}
	for k,v :=range slice{
		fmt.Println("k:",k,"v:",v)
	}
}

我们预想的结果应该是：

k: 0 v: 0
k: 1 v: 100
k: 2 v: 2
k: 3 v: 3

但是实际上

k: 0 v: 0
k: 1 v: 1
k: 2 v: 2
k: 3 v: 3

我们发现slice的值并没有改变

出现上述问题的原因是因为for range遍历的内容是对原内容的一个拷贝，所以不能用来修改原切片中内容。

上面的问题，我们可以使用 k根据索引直接修改值。

	for k,v :=range slice{
		if v==1 {
			slice[k]=100
		}
	}

 

我们再来看看range再map中的坑

 

package main

import (
	"fmt"
)

func main() {

	s :=[]int{1,2,3,4}
	m :=make(map[int]*int)

	for k,v:=range s{
		m[k]=&v
	}
	for key, value := range m {
		fmt.Printf("map[%v]=%v\n", key, *value)
	}

	fmt.Println(m)
}

我们预期打印的值应该为：

map[0]=1
map[1]=2
map[2]=3
map[3]=4

实际结果：

map[2]=4
map[3]=4
map[0]=4
map[1]=4

从上面结果我们可以猜想到，range指向的都是同一个指针。通过Println我们可以验证下我们的猜想

map[1:0xc00008a000 2:0xc00008a000 3:0xc00008a000 0:0xc00008a000]，我们可以看到我们的猜想是正确的

其实还是因为for range创建的是每个元素的拷贝，而不是直接返回每个元素的引用，如果使用该值变量的地址作为指向每个元素的指针，就会导致错误，在迭代时，返回的变量是一个迭代过程中根据切片依次赋值的新变量，所以值的地址总是相同的，导致结果不如预期。

 

修改方式也很简单，我们只需要声明一个中间变量，保存value,并且复制给map即可

package main

import (
	"fmt"
)

func main() {

	s :=[]int{1,2,3,4}
	m :=make(map[int]*int)

	for k,v:=range s{
		n:=v
		m[k]= &n
	}
	for key, value := range m {
		fmt.Printf("map[%v]=%v\n", key, *value)
	}

	fmt.Println(m)
}