简单的 for 循环也会踩的坑

简介: 最近实现某个业务时,需要读取数据然后再异步处理;在 Go 中实现起来自然就比较简单

前言


最近实现某个业务时,需要读取数据然后再异步处理;在 Go 中实现起来自然就比较简单,伪代码如下:


list := []*Demo{{"a"}, {"b"}}
  for _, v := range list {
    go func() {
      fmt.Println("name="+v.Name)
    }()
  }
  type Demo struct {
    Name string
  }


看似非常简单几行代码却和我们的预期不符,打印之后输出的是:


name=b
name=b


并不是我们预期的:


name=a
name=b


坑一


由于写 go 的资历尚浅、道行更是浅薄,这 bug 我硬是找了个把小时;刚开始还以为是数据源的问题,经历了好几轮自我怀疑。总之过程先不表,先看看如何修复这个问题。


首先第一种办法是使用临时变量:


list := []*Demo{{"a"}, {"b"}}
  for _, v := range list {
    temp:=v
    go func() {
      fmt.Println("name="+temp.Name)
    }()
  }


这样便可正确输出,其实从这种写法中也能看出问题的端倪。


在第一种没有使用临时变量时,主协程很快就运行完毕,这时候打印的子协程可能还没运行;当开始运行的时候,这里的 v 已经被最后一个赋值了。


所以这里打印的一直都是最后一个变量。


而使用临时变量会将当前遍历的值拷贝一份,自然就不会互相影响了。


当然除了临时变量也可使用闭包解决。


list := []*Demo{{"a"}, {"b"}}
  for _, v := range list {
    go func(temp *Demo) {
      fmt.Println("name="+temp.Name)
    }(v)
  }


将参数通过闭包传递时,每个 goroutine 都会在自己的栈中存放一份参数的拷贝,这样也能区分了。


坑二


与之类似的还有第二个坑:


list2 := []Demo{{"a"}, {"b"}}
  var alist []*Demo
  for _, test := range list2 {
    alist = append(alist, &test)
  }
  fmt.Println(alist[0].Name, alist[1].Name)


这段代码与我们预期不不符:


b b


但我们稍加修改就可以了:


list2 := []Demo{{"a"}, {"b"}}
  var alist []Demo
  for _, test := range list2 {
    fmt.Printf("addr=%p\n", &test)
    alist = append(alist, test)
  }
  fmt.Println(alist[0].Name, alist[1].Name)


addr=0xc000010240
addr=0xc000010240
a b


顺便打印了内存地址,其实从结果中大概就能猜到原因;每次遍历打印的内存地址都是相同,所以如果我们存放的是指针,本质上存储的都是同一块内存地址的内容,所以值相同。


而如果我们只存储值,不存指针自然也不会有这个问题。


但如果想使用指针如何处理呢?


list2 := []Demo{{"a"}, {"b"}}
  var alist []*Demo
  for _, test := range list2 {
    temp := test
    //fmt.Printf("addr=%p\n", &test)
    alist = append(alist, &temp)
  }
  fmt.Println(alist[0].Name, alist[1].Name)


也简单,同样的使用临时变量即可。


通过官方源码可以得知,for range 只是语法糖,本质上也是 for 循环;因为每次都是对同一个对象遍历赋值,所以便会出现这样的“乌龙”。


网络异常,图片无法展示
|


defer 的坑


for 循环 + defer 也是组合坑(虽然不推荐这么用),还是先来看个例子:


// demo1
func main() {
  a := []int{1, 2, 3}
  for _, v := range a {
    defer fmt.Println(v)
  }
}
// demo2
func main() {
  a := []int{1, 2, 3}
  for _, v := range a {
    defer func() {
      fmt.Println(v)
    }()
  }
}


分别输出:


//demo1
3
2
1
//demo2
3
3
3


demo1的结果很好理解,defer 可以理解为将执行语句放入到栈中,所以呈现的结果是先进后出。


demo2中,由于是闭包,闭包对变量 v 持有的是引用,所以在最终延迟执行时 v 已经被最后一个值赋值,所以打印出来都是相同的。


解决方法与上文类似,传入参数即可解决:


for _, v := range a {
    defer func(v int) {
      fmt.Println(v)
    }(v)
  }


这类细节问题日常开发大概率是碰不上的,最有可能遇到的就是面试了,所以多了解了解也没坏处。


总结


类似于第一种情况在 for 循环中 goroutine 调用,我觉得  IDE 完全是可以做到提醒的;比如 IDEA 中就把大部分认为可能发的错误包含进去,期待后续 goland 的更新。


但其实这几种错误官方博客已经提醒过了。


网络异常,图片无法展示
|


github.com/golang/go/w…


相关文章
|
3月前
|
搜索推荐 大数据 数据处理
面试官:try-catch 到底写在循环里面好,还是外面好?大部分人都会答错!
面试官:try-catch 到底写在循环里面好,还是外面好?大部分人都会答错!
46 0
循环初级一(附小练习和详细代码答案)
循环初级一(附小练习和详细代码答案)
109 0
|
算法 搜索推荐 程序员
ChatGPT,为啥写二分搜索容易死循环?
ChatGPT,为啥写二分搜索容易死循环?
99 0
|
6月前
|
算法 搜索推荐 程序员
第四十二练 检测循环括号
第四十二练 检测循环括号
36 1
C真的不难学,不信就看下我关于循环的理解
C真的不难学,不信就看下我关于循环的理解
循环初级二(附小练习及其代码答案)
循环初级二(附小练习及其代码答案)
65 0
|
JavaScript
for-in循环一看就懂
for-in循环一看就懂
56 0
|
算法
循环 — 你必须要会的十五道编程题(1)
循环 — 你必须要会的十五道编程题(1)
221 0
循环 — 你必须要会的十五道编程题(1)
循环 — 你必须要会的十五道编程题(2)
循环 — 你必须要会的十五道编程题(2)
115 0
循环 — 你必须要会的十五道编程题(2)
|
JSON 前端开发 JavaScript
HTML+CSS+JS实现【别踩鸡块】,含(源码+思路)
HTML+CSS+JS实现【别踩鸡块】,含(源码+思路)
463 0
HTML+CSS+JS实现【别踩鸡块】,含(源码+思路)