快捷導(dǎo)航

golang遍歷時(shí)修改被遍歷對(duì)象的示例詳解

更新時(shí)間：2021年01月15日 14:15:27 作者：janbar

這篇文章主要介紹了golang遍歷時(shí)修改被遍歷對(duì)象的示例代碼，本文給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下

前言

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

遍歷切片

1.遍歷切片時(shí)去掉元素，錯(cuò)誤示例：

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

最終報(bào)錯(cuò)panic: runtime error: index out of range [3] with length 3，因?yàn)閞ange在迭代時(shí)已經(jīng)確定i的范圍為[0,len(arr))的左閉右開(kāi)的區(qū)間。
但是當(dāng)滿(mǎn)足arr[i] == 3時(shí)對(duì)arr進(jìn)行了修改,縮短了arr的長(zhǎng)度，此時(shí)len(arr)=3，最大下標(biāo)為2，因此當(dāng)執(zhí)行arr[3]時(shí)會(huì)報(bào)錯(cuò)。

2.遍歷切片時(shí)去掉元素，不會(huì)報(bào)錯(cuò)，但不建議的寫(xiě)法：

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的用法，當(dāng)執(zhí)行for循環(huán)時(shí)就已經(jīng)確定(i,v)的遍歷元素值，及時(shí)循環(huán)過(guò)程中修改arr，也不會(huì)改變for要遍歷的(i,v)值。
可以將上面代碼修改一下，看下在循環(huán)中改變arr值時(shí)，后面遍歷的(i,v)是不會(huì)隨著arr的改變而改變的。

3.遍歷切片時(shí)去掉元素，建議寫(xiě)法：

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的值，在刪除元素時(shí)進(jìn)行i--，可以確保遍歷arr沒(méi)有問(wèn)題，而且每次通過(guò)arr[i]獲取切片值不存在問(wèn)題。
當(dāng)然用該方式也可以在遍歷時(shí)添加元素，只要i也對(duì)應(yīng)變化就沒(méi)問(wèn)題。

遍歷map

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

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

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

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