go语言控制并发数量的方法是什么 - 问答

Go语言控制并发数量的方法有以下几种：

使用带有缓冲的channel：可以使用带有缓冲的channel来限制并发数量。创建一个具有指定缓冲区大小的channel，然后在并发的代码中使用该channel来控制并发数量。

ch := make(chan struct{}, maxConcurrency)
for i := 0; i < totalTasks; i++ {
    ch <- struct{}{} // 占用一个channel的缓冲区
    go func() {
        defer func() { <-ch }() // 释放一个channel的缓冲区
        // 执行并发任务
    }()
}

使用sync.WaitGroup：可以使用sync.WaitGroup来等待并发任务完成。在并发的代码中，每个任务开始前调用Add方法，任务结束后调用Done方法，然后在主goroutine中调用Wait方法等待所有任务完成。

var wg sync.WaitGroup
for i := 0; i < totalTasks; i++ {
    wg.Add(1)
    go func() {
        defer wg.Done()
        // 执行并发任务
    }()
}
wg.Wait()

使用有限制的goroutine池：可以创建一个有限数量的goroutine池，使用通道来控制任务的提交和执行。当有任务需要执行时，从通道中获取一个goroutine并执行任务。

pool := make(chan struct{}, maxGoroutines)
for i := 0; i < maxGoroutines; i++ {
    pool <- struct{}{} // 占用一个goroutine
    go func() {
        defer func() { <-pool }() // 释放一个goroutine
        // 执行并发任务
    }()
}

以上方法都可以用来控制并发数量，根据实际场景选择合适的方法。

0 赞

0 踩