在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)
}
这些方法都可以有效地控制并发数量,根据实际情况选择合适的方法。