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 的值来控制并发数。