0
点赞
收藏
分享

微信扫一扫

golang 协程并发代码 demo

有时我们可能想既在外层循环中实现多协程并发,还想在内层循环中实现多协程并发,那么我们需要同时在内层和外层使用 ​​WaitGroup()​​ 来控制主协程不退出。

下面是一个 demo:

博客平台纯手敲,可能存在字符拼写错误

import (
"fmt"
"sync"
)

func handleTask() {
demoList := []string{"123", "456", "abc", "ddd"}

var wg = sync.WaitGroup{}

for idx, item := range demoList {
// 每个元素创建一个新协程去处理
wg.Add(1) //
go func(idx int){
defer wg.Done() // 协程退出前将 wg 计数器减一,否则最后计数器无法减为0,会一直卡在 wg.wait() 那一行

// 内部还想新建子协程去做不同的事,减少执行时间
var innerWg = sync.WaitGroup{} //

innerWg.Add(1)
go func(){
defer innerWg.Done() // 子协程退出前将 innerWg 的计数器减一
fmt.Printf("子协程一正在执行")
}

innerWg.Add(1)
go func(){
defer innerWg.Done() // 子协程退出前将 innerWg 的计数器减一
fmt.Printf("子协程二正在执行")
}
innerWg.Wait() // 等所有所有子协程执行完才继续往下执行

fmt.Printf("外层协程%d即将执行完毕", idx)

}(idx)
}

wg.Wait() // 等待 wg 的计数器减为0后才继续往下执行,等待所有的协程处理完毕

fmt.Println("Execution completed. will exit")
}

func main() {

handleTask()
}

  • sync.WaitGroup{}:一个内部加锁的计数器,Add(num) 函数会使计数器加上对应的 num 值;Done() 函数会使计数器减一;Wait() 函数会一直阻塞程序的继续向下运行,直到计数器减为0。
  • Add() 函数需要在子协程开辟前执行;Done() 函数最好在子协程最开始的 defer 中执行,保证子协程退出后一定会将计数器减一,否则主协程可能因为计数器不为 0 一直卡主;Wait() 函数一般放在主协程里,且是子协程执行完毕后。
  • 如果需要控制协程的数量,还得引入 channel ,参考:​​5、Go是否可以无限go? 如何限定数量?​​,文末是从此文拷贝过来的一段代码

下面是内层子协程并发的另一种写法

博客平台纯手敲,可能存在字符拼写错误

import (
"fmt"
"sync"
)

func handleTask() {
demoList := []string{"123", "456", "abc", "ddd"}

var wg = sync.WaitGroup{}

for idx, item := range demoList {
// 每个元素创建一个新协程去处理
wg.Add(1) //
go func(idx int){
defer wg.Done() // 协程退出前将 wg 计数器减一,否则最后计数器无法减为0,会一直卡在 wg.wait() 那一行

// 内部还想新建子协程去做不同的事,减少执行时间
var innerWg = sync.WaitGroup{} //
for i := 1; i < 3; i++ {
innerWg.Add(1)
go func(){
defer innerWg.Done() // 子协程退出前将 innerWg 的计数器减一
fmt.Printf("子协程 %d 正在执行\n", i)
}
}

fmt.Printf("外层协程%d即将执行完毕", idx)

}(idx)
}

wg.Wait() // 等待 wg 的计数器减为0后才继续往下执行,等待所有的协程处理完毕

fmt.Println("Execution completed. will exit")
}

func main() {

handleTask()
}

channel 与 sync 同步组合方式控制协程数量

代码来自:​​5、Go是否可以无限go? 如何限定数量?​​

package main

import (
"fmt"
"math"
"sync"
"runtime"
)

var wg = sync.WaitGroup{}

func busi(ch chan bool, i int) {

fmt.Println("go func ", i, " goroutine count = ", runtime.NumGoroutine())

<-ch

wg.Done()
}

func main() {
//模拟用户需求go业务的数量
task_cnt := math.MaxInt64

ch := make(chan bool, 3)

for i := 0; i < task_cnt; i++ {
wg.Add(1)

ch <- true

go busi(ch, i)
}

wg.Wait()
}

参考:​​5、Go是否可以无限go? 如何限定数量?​​



举报

相关推荐

0 条评论