在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可以运行。