golang遍历时修改被遍历对象的示例详解

时间:2022-11-23 07:44:48

前言

很多时候需要将遍历对象中去掉某些元素,或者往遍历对象中添加元素,这时候就需要小心操作了。
对于go语言中的一些注意事项我做了总结和示例,留下点笔记。

遍历切片

1.遍历切片时去掉元素,错误示例:

?
1
2
3
4
5
6
7
8
9
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.遍历切片时去掉元素,不会报错,但不建议的写法:

?
1
2
3
4
5
6
7
8
9
10
11
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.遍历切片时去掉元素,建议写法:

?
1
2
3
4
5
6
7
8
9
10
11
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时去掉元素,可参考官方示例,可看下官方描述,下面这种方案是安全的。

?
1
2
3
4
5
for key := range m {
  if key.expired() {
    delete(m, key)
  }
}

2.清空map所有元素,如下第一种是省事的写法,第二种不会产生新的对象,用哪种看个人喜好吧。

?
1
2
3
4
5
m = make(map[int]int) // 可以产生一个新对象,旧对象等着被垃圾回收
 
for k := range m {
  delete(m,k) // 循环遍历并删除map所有元素,好处是map缓存还在,下次添加时可直接使用缓存
}

总结

关于切片遍历时进行操作需要注意一些坑。
关于map遍历时进行操作相对坑少点,不过遍历map需要修改元素时,map的value要为指针类型,这点得谨记。

到此这篇关于golang遍历时修改被遍历对象的文章就介绍到这了,更多相关golang遍历对象内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家!

原文链接:https://www.cnblogs.com/janbar/archive/2021/01/15/14281434.html