go語言range的坑
阿新 • • 發佈:2018-12-10
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)
}