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要為指針類型，這點得謹記。