1. 程式人生 > >go語言range的坑

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)
}