golang遍歷時修改被遍歷對象的示例詳解

前言

很多時候需要將遍歷對象中去掉某些元素，或者往遍歷對象中添加元素，這時候就需要小心操作了。
對于go語言中的一些注意事項我做了總結(jié)和示例，留下點筆記。

遍歷切片

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在迭代時已經(jīng)確定i的范圍為[0,len(arr))的左閉右開的區(qū)間。
但是當滿足arr[i] == 3時對arr進行了修改,縮短了arr的長度，此時len(arr)=3，最大下標為2，因此當執(zhí)行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的用法，當執(zhí)行for循環(huán)時就已經(jīng)確定(i,v)的遍歷元素值，及時循環(huán)過程中修改arr，也不會改變for要遍歷的(i,v)值。
可以將上面代碼修改一下，看下在循環(huán)中改變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也對應(yīng)變化就沒問題。

遍歷map

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

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

2.清空map所有元素，如下第一種是省事的寫法，第二種不會產(chǎn)生新的對象，用哪種看個人喜好吧。

m = make(map[int]int) // 可以產(chǎn)生一個新對象,舊對象等著被垃圾回收
 
for k := range m {
  delete(m,k) // 循環(huán)遍歷并刪除map所有元素,好處是map緩存還在,下次添加時可直接使用緩存
}

總結(jié)

關(guān)于切片遍歷時進行操作需要注意一些坑。
關(guān)于map遍歷時進行操作相對坑少點，不過遍歷map需要修改元素時，map的value要為指針類型，這點得謹記。

到此這篇關(guān)于golang遍歷時修改被遍歷對象的文章就介紹到這了,更多相關(guān)golang遍歷對象內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！