golang的最大并发数怎么控制

avatar
作者
猴君
阅读量:1

在Golang中,可以通过控制goroutine的数量来控制最大并发数。

一种常见的方法是使用带有缓冲的channel来限制并发数。通过创建一个具有固定缓冲大小的channel,可以限制同时运行的goroutine数量。当goroutine想要启动时,它需要先从channel中获取一个可用的令牌,如果没有可用的令牌,则goroutine将被阻塞,直到有令牌可用为止。

以下是一个示例代码:

package main  import "fmt"  func worker(id int, tokens chan struct{}) { 	fmt.Printf("Worker %d started\n", id) 	<-tokens // 获取一个令牌  	// 执行工作任务 	fmt.Printf("Worker %d is working\n", id)  	tokens <- struct{}{} // 释放令牌 	fmt.Printf("Worker %d finished\n", id) }  func main() { 	concurrency := 3 // 最大并发数  	tokens := make(chan struct{}, concurrency) // 创建带有固定缓冲大小的channel  	// 启动多个goroutine 	for i := 0; i < 10; i++ { 		tokens <- struct{}{} // 添加令牌到channel 		go worker(i, tokens) 	}  	// 等待所有goroutine完成 	for i := 0; i < concurrency; i++ { 		tokens <- struct{}{} 	} } 

在上述示例中,我们定义了一个带有固定缓冲大小的channel tokens。在main函数中,我们启动了10个goroutine,每个goroutine都会从tokens channel中获取一个令牌,然后执行工作任务。当任务完成后,goroutine会将令牌放回到channel中,以便其他goroutine可以获取并执行任务。

通过调整concurrency变量的值,您可以控制最大并发数。在上述示例中,我们将最大并发数设置为3,这意味着同一时间最多有3个goroutine可以运行。

广告一刻

为您即时展示最新活动产品广告消息,让您随时掌握产品活动新动态!