Goroutine并发控制

痴心易碎 提交于 2020-02-13 12:57:07

创建协程

    jobCount := 10
    // sync.WaitGroup 监控所有协程的状态,从而保证主协程结束时所有的子协程已经退出
    group := sync.WaitGroup{}
    for i:=0;i < jobCount;i++ {
        group.Add(1)
        go func(i int) {
            fmt.Println("task ",i)
            time.Sleep(time.Second) // 刻意睡 1 秒钟,模拟耗时
            group.Done()
        }(i)
        fmt.Printf("index: %d,goroutine Num: %d \n", i, runtime.NumGoroutine())
    }
    group.Wait()

运行结果:

index: 0,goroutine Num: 2 
index: 1,goroutine Num: 3 
task  0
index: 2,goroutine Num: 4 
index: 3,goroutine Num: 5 
task  3
task  2
index: 4,goroutine Num: 6 
index: 5,goroutine Num: 7 
index: 6,goroutine Num: 8 
task  4
index: 7,goroutine Num: 9 
task  6
index: 8,goroutine Num: 10 
index: 9,goroutine Num: 11 
task  7
task  8
task  1
task  9
task  5

可以看到总共有11个协程,其中一个是主协程,其他十个是子协程,为了让主协程等待所有的子协程执行完毕后再退出,使用 sync.WaitGroup 监控所有协程的状态,从而保证主协程结束时所有的子协程已经退出。

实际生产中,产生的goroutine的数量,是巨大,但是这种简单直接的方式就显得不那么高效了。CPU同一时间只能处理一个Goroutine,大量job的情况将会出现大量的Goroutine等待,以致于造成资源的浪费。

控制Gorutine的数量

// 控制 Goroutine 数量
    jobCount := 10
    group := sync.WaitGroup{}

    // 规定chan容量为3
    jobsChan := make(chan int,3)

    // 起3个worker
    workerCount := 3
    for w:=0; w <= workerCount; w++ {
        go func(w int){
            for j := range jobsChan {
                fmt.Printf("worker %d get chan msg %d \n",w,j)
                time.Sleep(time.Second)
                group.Done()
            }
        }(w)
    }

    // 发布消息到chan
    for j :=0; j < jobCount; j++ {
        jobsChan <- j
        group.Add(1)
        fmt.Printf("index: %d,goroutine Num: %d\n", j, runtime.NumGoroutine())
    }

    group.Wait()
worker 1 get chan msg 0 
index: 0,goroutine Num: 5
index: 1,goroutine Num: 6
worker 0 get chan msg 1 
worker 2 get chan msg 2 
index: 2,goroutine Num: 5
index: 3,goroutine Num: 5
index: 4,goroutine Num: 5
index: 5,goroutine Num: 5
index: 6,goroutine Num: 5
worker 3 get chan msg 3 
worker 0 get chan msg 4 
worker 1 get chan msg 7 
index: 7,goroutine Num: 5
worker 2 get chan msg 5 
index: 8,goroutine Num: 5
index: 9,goroutine Num: 5
worker 3 get chan msg 6 
worker 0 get chan msg 8 
worker 3 get chan msg 9 
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!