golang怎样限制协程数量
在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
通道的缓冲大小决定了同时履行的协程数量。在创建信号量时,可以根据实际需求选择适合的缓冲大小。
TOP