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

60次阅读

共计 756 个字符，预计需要花费 2 分钟才能阅读完成。

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
        // 执行并发任务 
    }()}