golang遍歷時修改被遍歷對象

目錄

• 前言
• 遍歷切片
• 遍歷map
• 總結

前言

不少時候須要將遍歷對象中去掉某些元素，或者往遍歷對象中添加元素，這時候就須要當心操做了。
對於go語言中的一些注意事項我作了總結和示例，留下點筆記。

遍歷切片

1. 遍歷切片時去掉元素，錯誤示例：

func main() {
    arr := []int{1, 2, 3, 4}
    for i := range arr {
        if arr[i] == 3 {
            arr = append(arr[:i], arr[i+1:]...)
        }
    }
    fmt.Println(arr)
}

最終報錯panic: runtime error: index out of range [3] with length 3，由於range在迭代時已經肯定i的範圍爲[0,len(arr))的左閉右開的區間。
可是當知足arr[i] == 3時對arr進行了修改,縮短了arr的長度，此時len(arr)=3，最大下標爲2，所以當執行arr[3]時會報錯。

2. 遍歷切片時去掉元素，不會報錯，但不建議的寫法：

func main() {
    arr := []int{1, 2, 3, 4}
    for i, v := range arr {
        fmt.Println(i, v)
        if v == 3 {
            arr = append(arr[:i], arr[i+1:]...)
            // arr = []int{4, 5, 6, 7} // 能夠將上一行代碼替換,看最終遍歷的i,v狀況
        }
    }
    fmt.Println(arr)
}

仍是回到range的用法，當執行for循環時就已經肯定(i,v)的遍歷元素值，及時循環過程當中修改arr，也不會改變for要遍歷的(i,v)值。
能夠將上面代碼修改一下，看下在循環中改變arr值時，後面遍歷的(i,v)是不會隨着arr的改變而改變的。

3. 遍歷切片時去掉元素，建議寫法：

func main() {
    arr := []int{1, 2, 3, 4}
    for i := 0; i < len(arr); i++ {
        fmt.Println(i, arr[i])
        if arr[i] == 3 {
            arr = append(arr[:i], arr[i+1:]...)
            i--
        }
    }
    fmt.Println(arr)
}

該方案只修改i的值，在刪除元素時進行i--，能夠確保遍歷arr沒有問題，並且每次經過arr[i]獲取切片值不存在問題。
固然用該方式也能夠在遍歷時添加元素，只要i也對應變化就沒問題。

遍歷map

1. 遍歷map時去掉元素，可參考官方示例，可看下官方描述，下面這種方案是安全的。

for key := range m {
    if key.expired() {
        delete(m, key)
    }
}

2. 清空map全部元素，以下第一種是省事的寫法，第二種不會產生新的對象，用哪一種看我的喜愛吧。

m = make(map[int]int) // 能夠產生一個新對象,舊對象等着被垃圾回收

for k := range m {
    delete(m,k) // 循環遍歷並刪除map全部元素,好處是map緩存還在,下次添加時可直接使用緩存
}

總結

關於切片遍歷時進行操做須要注意一些坑。
關於map遍歷時進行操做相對坑少點，不過遍歷map須要修改元素時，map的value要爲指針類型，這點得謹記。