golang协程闭包数据陷阱

0x1

咱们在写协程程序的时候，常常会碰到一个场景就是咱们要分发执行任务给不一样的goroutine（简称gor)，而后再把各个gor的处理结果汇总起来，这个时候就要注意gor的数据污染问题，咱们能够经过闭包来防范各个gor之间的数据污染golang

0x2

下面的一个gor之间数据互相污染的范例shell

func main() {
  setMem := make(map[int]int)
  wg := sync.WaitGroup{}
  lk := sync.RWMutex{}

  for i := 0; i < 10; i++ {    // 1
    wg.Add(1)
    
    go func(wg *sync.WaitGroup) {    // 2
      fmt.Println("i ---- ", i)
      defer wg.Done()
      lk.Lock()
      setMem[i] = i * 2
      lk.Unlock()
    }(&wg)
    
  }

  wg.Wait()
  fmt.Println("setMem 长度为 ----------", len(setMem))
  fmt.Println("setMem done ... ")
}

这个的运行结果并不是如咱们所想，汇总了10个 gor的处理结果到 setMem 这个map中去安全

setMem 长度为 ---------- 1
setMem done ...

缘由分析闭包

注释1 由于main gor 和 go func 是同时在运行的， 因此main在赋值给 i 的同时， 2 接收到 i 的值并无锁定 i ， 而是接收到最新的for循环的值code

注释2 没有一个安全的内存区域储存传给当前 gor 的 i 值，因此形成数据污染协程

0x3

用闭包的原理去解决这个问题内存

func setM(wg *sync.WaitGroup, lk *sync.RWMutex, setMemx *map[int]int, i int) {
  defer wg.Done()
  fmt.Println("i ---- ", i)
  (*setMemx)[i] = i * 2
}

func main() {
  setMemx := make(map[int]int)
  wg := sync.WaitGroup{}
  lk := sync.RWMutex{}
   
  for i := 0; i < 10; i++ {  		 // 1
    wg.Add(1)
    setM(&wg, &lk, &setMemx, i)      // 2
  }

  wg.Wait()
  fmt.Println("setMem 长度为 ----------", len(setMemx))
  fmt.Println("setMem done ... ")
}

运行结果作用域

setMen 长度为 ---------- 10
setMem done ...

注释1 1 处依然是经过循环赋予 i 值 注释2 2经过声明独立的func，来进行内存数据做用域锁定，闭包的原理，防止数据污染it

因此运行的结果如咱们所想for循环