阅读量:0
在golang中,可以使用goroutine和channel来控制并发数量。以下是一些常用的方法:
- 使用goroutine和WaitGroup:可以使用sync包中的WaitGroup来实现控制并发数量。创建一个WaitGroup对象,然后在每个goroutine中调用Add方法增加计数器,执行完任务后调用Done方法减少计数器。最后,可以使用Wait方法阻塞主goroutine,直到计数器归零。
import ( "sync" ) func main() { var wg sync.WaitGroup maxConcurrency := 10 // 并发数量 for i := 0; i < maxConcurrency; i++ { wg.Add(1) go func() { defer wg.Done() // 执行任务 }() } wg.Wait() }
- 使用channel和限制缓冲区大小:可以使用带有缓冲区的channel来控制并发数量。创建一个带有指定缓冲区大小的channel,然后在每个goroutine中向channel发送任务。channel的缓冲区大小限制了并发数量。
func main() { maxConcurrency := 10 // 并发数量 taskCh := make(chan int, maxConcurrency) for i := 0; i < maxConcurrency; i++ { go func() { for task := range taskCh { // 执行任务 } }() } // 发送任务到channel for i := 0; i < 100; i++ { taskCh <- i } close(taskCh) }
这些方法都可以有效地控制并发数量,根据实际情况选择合适的方法。