温馨提示×

golang怎么限制协程数量

小亿
212
2023-10-21 14:04:23
栏目: 编程语言

在Go语言中,可以使用信号量(Semaphore)来限制协程的数量。信号量是一种多线程同步的原语,用于控制对共享资源的访问。

可以使用Go语言中的sync包中的Semaphore来实现信号量。sync包中提供了一个WaitGroup类型,可以用于等待一组协程完成。

下面是一个示例代码,演示如何使用信号量限制协程数量:

package main import ( "fmt" "sync" ) func worker(id int, semaphore chan struct{}, wg *sync.WaitGroup) { // 从信号量中获取一个许可	semaphore <- struct{}{} // 执行工作	fmt.Println("Worker", id, "start") // 模拟工作 for i := 0; i < 5; i++ {	fmt.Println("Worker", id, "working", i)	}	fmt.Println("Worker", id, "done") // 释放许可,使其他协程可以获取	<-semaphore // 通知WaitGroup,当前协程已完成	wg.Done() } func main() { const numWorkers = 3	semaphore := make(chan struct{}, numWorkers) // 创建大小为numWorkers的信号量 var wg sync.WaitGroup for i := 0; i < 10; i++ {	wg.Add(1) go worker(i, semaphore, &wg)	}	wg.Wait() // 等待所有协程完成 } 

在上述代码中,我们创建了一个大小为numWorkers的信号量semaphore,用于控制协程的数量。每个协程在开始执行工作前,先从信号量获取一个许可,当工作完成后,释放许可,使其他协程可以获取。这样就可以限制同时执行的协程数量为numWorkers

需要注意的是,semaphore通道的缓冲大小决定了同时执行的协程数量。在创建信号量时,可以根据实际需求选择合适的缓冲大小。

0