golang - 使用有缓冲通道控制并发数
在 Go 语言中,使用带缓冲的通道(buffered channels)可以有效地控制并发数。带缓冲的通道可以让你限制同时运行的 goroutine 数量,从而避免过度并发导致的资源耗尽问题。以下是一个使用带缓冲通道控制并发数的示例:
package mainimport ("fmt""sync""time"
)func worker(id int, sem chan struct{}, wg *sync.WaitGroup) {defer wg.Done()// 请求一个资源sem <- struct{}{}// 模拟耗时操作fmt.Printf("Worker %d started\n", id)time.Sleep(2 * time.Second)fmt.Printf("Worker %d finished\n", id)// 释放一个资源<-sem
}func main() {// 定义最大并发数const maxConcurrency = 3// 创建带缓冲的通道,用于限制并发数sem := make(chan struct{}, maxConcurrency)// 使用 sync.WaitGroup 等待所有 goroutine 完成var wg sync.WaitGroup// 启动 10 个 workerfor i := 1; i <= 10; i++ {wg.Add(1)go worker(i, sem, &wg)}// 等待所有 worker 完成wg.Wait()
}
在这个示例中,我们创建了一个带缓冲的通道 sem,并将其缓冲大小设置为最大并发数。我们使用了一个 sync.WaitGroup 实例来等待所有 goroutine 完成。当 worker 开始执行时,它会向通道发送一个空结构体,从而请求一个资源。如果通道已满,worker 将会阻塞,直到有可用资源。
当 worker 完成任务后,它会从通道接收一个空结构体,从而释放一个资源。这样,我们就可以通过调整 maxConcurrency 的值来控制并发数。