温馨提示×

golang的最大并发数怎么控制

小亿
119
2023-10-20 21:02:13
栏目: 编程语言

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

0