阅读量:0
Go语言控制并发数量的方法有以下几种:
- 使用带有缓冲的channel:可以使用带有缓冲的channel来限制并发数量。创建一个具有指定缓冲区大小的channel,然后在并发的代码中使用该channel来控制并发数量。
ch := make(chan struct{}, maxConcurrency) for i := 0; i < totalTasks; i++ { ch <- struct{}{} // 占用一个channel的缓冲区 go func() { defer func() { <-ch }() // 释放一个channel的缓冲区 // 执行并发任务 }() }
- 使用sync.WaitGroup:可以使用sync.WaitGroup来等待并发任务完成。在并发的代码中,每个任务开始前调用Add方法,任务结束后调用Done方法,然后在主goroutine中调用Wait方法等待所有任务完成。
var wg sync.WaitGroup for i := 0; i < totalTasks; i++ { wg.Add(1) go func() { defer wg.Done() // 执行并发任务 }() } wg.Wait()
- 使用有限制的goroutine池:可以创建一个有限数量的goroutine池,使用通道来控制任务的提交和执行。当有任务需要执行时,从通道中获取一个goroutine并执行任务。
pool := make(chan struct{}, maxGoroutines) for i := 0; i < maxGoroutines; i++ { pool <- struct{}{} // 占用一个goroutine go func() { defer func() { <-pool }() // 释放一个goroutine // 执行并发任务 }() }
以上方法都可以用来控制并发数量,根据实际场景选择合适的方法。