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