golang 限制最大并发

发布时间:2024-11-21 23:30:48

作为一个专业的golang开发者,我们在编写并发程序时需要注意到golang的一个限制,即限制最大并发数。虽然golang天生支持高并发,但是在实际使用中,我们往往需要限制最大并发数以避免资源过度占用和系统崩溃的风险。那么如何限制最大并发数呢?本文将介绍三种常见的方法。

使用有缓冲的通道

有缓冲的通道是一种我们常用的限制最大并发数的方法。通过创建一个具有指定容量的通道,我们可以控制消息的发送和接收的速度,从而限制并发数。

我们可以使用通道的缓冲区来存储一定数量的数据,当缓冲区已满时,任何尝试往通道发送数据的操作都会被阻塞直到有空间可用。这样就可以实现对最大并发数的限制,当缓冲区已满时,新的并发任务就无法添加进来了。

下面是一个简单的例子,演示了使用有缓冲的通道限制最大并发数: package main import ( "fmt" "time" ) func worker(id int, jobs <-chan int, results chan<- string) { for j := range jobs { fmt.Println("worker", id, "processing job", j) time.Sleep(time.Second) // 模拟耗时操作 results <- fmt.Sprintf("result from worker %d for job %d", id, j) } } func main() { numWorkers := 3 numJobs := 10 jobs := make(chan int, numJobs) results := make(chan string, numJobs) for w := 1; w <= numWorkers; w++ { go worker(w, jobs, results) } for j := 1; j <= numJobs; j++ { jobs <- j } close(jobs) for r := 1; r <= numJobs; r++ { fmt.Println(<-results) } }

使用带有计数器的通道

除了使用有缓冲的通道,我们还可以使用带有计数器的通道来限制最大并发数。通过在通道中存储并发任务的计数器,我们可以实时更新当前的并发数,并根据需要进行控制。

下面是一个简单的例子,演示了使用带有计数器的通道限制最大并发数: package main import ( "fmt" "sync" "time" ) func worker(id int, wg *sync.WaitGroup) { fmt.Println("worker", id, "started") time.Sleep(time.Second) // 模拟耗时操作 fmt.Println("worker", id, "finished") wg.Done() } func main() { numWorkers := 3 maxConcurrency := 2 wg := &sync.WaitGroup{} semaphore := make(chan struct{}, maxConcurrency) for w := 1; w <= numWorkers; w++ { wg.Add(1) go func(id int) { semaphore <- struct{}{} // acquire semaphore worker(id, wg) <-semaphore // release semaphore }(w) } wg.Wait() }

使用有限制的goroutine池

另一种限制最大并发数的方法是使用有限制的goroutine池。通过初始化一个指定容量的goroutine池,我们可以控制并发任务的数量,并且可以重复利用goroutine,避免创建和销毁的开销。

下面是一个简单的例子,演示了使用有限制的goroutine池限制最大并发数: package main import ( "fmt" "sync" "time" ) func worker(id int, pool chan struct{}, wg *sync.WaitGroup) { fmt.Println("worker", id, "started") time.Sleep(time.Second) // 模拟耗时操作 fmt.Println("worker", id, "finished") <-pool // release goroutine wg.Done() } func main() { numWorkers := 3 maxConcurrency := 2 wg := &sync.WaitGroup{} pool := make(chan struct{}, maxConcurrency) for w := 1; w <= numWorkers; w++ { wg.Add(1) pool <- struct{}{} // acquire goroutine go worker(w, pool, wg) } wg.Wait() }

本文介绍了三种常见的方法,包括使用有缓冲的通道、使用带有计数器的通道、使用有限制的goroutine池来限制最大并发数。在实际应用中,我们可以根据具体的需求选择合适的方法,以保证程序的稳定性和性能。

相关推荐