在 Go 语言中,使用带缓冲的通道(buffered channels)可以有效地控制并发数。带缓冲的通道可以让你限制同时运行的 goroutine 数量,从而避免过度并发导致的资源耗尽问题。以下是一个使用带缓冲通道控制并发数的示例:
package main
import (
"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 个 worker
for i := 1; i <= 10; i++ {
wg.Add(1)
go worker(i, sem, &wg)
}
// 等待所有 worker 完成
wg.Wait()
}
在这个示例中,我们创建了一个带缓冲的通道 sem,并将其缓冲大小设置为最大并发数。我们使用了一个 sync.WaitGroup 实例来等待所有 goroutine 完成。当 worker 开始执行时,它会向通道发送一个空结构体,从而请求一个资源。如果通道已满,worker 将会阻塞,直到有可用资源。
当 worker 完成任务后,它会从通道接收一个空结构体,从而释放一个资源。这样,我们就可以通过调整 maxConcurrency 的值来控制并发数。