golang怎么控制协程数量

如何控制 Go 协程数量？使用 runtime.GOMAXPROCS 设置最大并发度。使用通道进行并发限制。使用上下文取消优雅地终止协程。

如何控制 Go 协程数量

协程是一种轻量级的并发执行单元，可以极大地提高 Go 程序的性能。然而，如果没有适当的控制，协程的数量可能会过大，从而导致内存消耗和性能问题。

控制协程数量的方法

有几种方法可以控制 Go 协程的数量：

立即学习“go语言免费学习笔记（深入）”；

1. 使用 runtime.GOMAXPROCS 设置最大并发度

runtime.GOMAXPROCS 设置了 Go 程序可以同时运行的最大并发协程数。默认值为处理器内核的数量。通过将 runtime.GOMAXPROCS 设置为较低的值，可以限制程序创建的协程数量。

Cutout.Pro

AI驱动的视觉设计平台

下载

<code class="go">import "runtime"

func main() {
    runtime.GOMAXPROCS(2) // 限制最大协程数为 2
}</code>

2. 使用通道进行并发限制

通道可以用来限制同时执行的协程数量。可以通过创建一个具有有限容量的通道，并在协程执行前将其发送到该通道，来限制并发度。

<code class="go">import "sync"

func main() {
    var wg sync.WaitGroup

    // 创建一个大小为 2 的通道
    c := make(chan struct{}, 2)

    for i := 0; i < 5; i++ {
        wg.Add(1)

        // 将通道发送到协程中
        go func(i int) {
            c <- struct{}{} // 等待通道可用
            defer wg.Done()
            defer <-c      // 完成后释放通道

            // 协程执行逻辑
            println("协程", i)
        }(i)
    }

    wg.Wait()
}</code>

3. 使用上下文取消

上下文取消可以用于优雅地终止协程。通过创建一个 context.Context 并将它传递给协程，可以控制协程的生命周期。当上下文被取消时，协程将被终止。

<code class="go">import (
    "context"
    "fmt"
    "sync"
    "time"
)

func main() {
    var wg sync.WaitGroup

    // 创建一个上下文
    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()

    for i := 0; i < 5; i++ {
        wg.Add(1)

        // 将上下文传递给协程
        go func(i int) {
            defer wg.Done()

            // 协程执行逻辑
            select {
            case <-ctx.Done():
                fmt.Println("协程", i, "被终止")
            }
        }(i)
    }

    wg.Wait()
}</code>

选择合适的方法

选择哪种方法来控制协程数量取决于应用程序的具体需求和并发模式。一般来说，使用 runtime.GOMAXPROCS 是一种简单且有效的方法，可以在大多数情况下控制协程数量。而通道和上下文取消则提供了更精细的控制，适用于需要更复杂的并发模式的应用程序。