阅读量: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可以运行。