新闻资讯

质量为本、客户为根、勇于拼搏、务实创新

< 返回新闻资讯列表

golang的最大并发数怎么控制,golang 并行

发布时间:2023-10-20 21:02:13

golang的最大并发数怎样控制

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

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

以下是一个示例代码:

package main

import "fmt"

func worker(id int, tokens chan struct{}) {
	fmt.Printf("Worker %d started
", id)
	<-tokens // 获得一个令牌

	// 履行工作任务
	fmt.Printf("Worker %d is working
", id)

	tokens <- struct{}{} // 释放令牌
	fmt.Printf("Worker %d finished
", 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可以运行。