golang - 使用有缓冲通道控制并发数

在 Go 语言中，使用带缓冲的通道（buffered channels）可以有效地控制并发数。带缓冲的通道可以让你限制同时运行的 goroutine 数量，从而避免过度并发导致的资源耗尽问题。以下是一个使用带缓冲通道控制并发数的示例：

package main

import (
    "fmt"
    "sync"
    "time"
)

func worker(id int, sem chan struct{}, wg *sync.WaitGroup) {
    defer wg.Done()

    // 请求一个资源
    sem <- struct{}{}

    // 模拟耗时操作
    fmt.Printf("Worker %d started\n", id)
    time.Sleep(2 * time.Second)
    fmt.Printf("Worker %d finished\n", id)

    // 释放一个资源
    <-sem
}

func main() {
    // 定义最大并发数
    const maxConcurrency = 3

    // 创建带缓冲的通道，用于限制并发数
    sem := make(chan struct{}, maxConcurrency)

    // 使用 sync.WaitGroup 等待所有 goroutine 完成
    var wg sync.WaitGroup

    // 启动 10 个 worker
    for i := 1; i <= 10; i++ {
        wg.Add(1)
        go worker(i, sem, &wg)
    }

    // 等待所有 worker 完成
    wg.Wait()
}

在这个示例中，我们创建了一个带缓冲的通道 sem，并将其缓冲大小设置为最大并发数。我们使用了一个 sync.WaitGroup 实例来等待所有 goroutine 完成。当 worker 开始执行时，它会向通道发送一个空结构体，从而请求一个资源。如果通道已满，worker 将会阻塞，直到有可用资源。

当 worker 完成任务后，它会从通道接收一个空结构体，从而释放一个资源。这样，我们就可以通过调整 maxConcurrency 的值来控制并发数。