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 的值来控制并发数。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • Channel 是什么? channel,通道,本质上是一个通信对象,goroutine 之间可以使用它来通信。从...
    癞痢头阅读 864评论 0 0
  • Golang:通道,同步等待组 并发爬虫 在Go的并发编程中有一句很经典的话:不要以共享内存的方式去通信,而要以通...
    鬼子口音阅读 670评论 0 0
  • Go语言中的并发编程 并发是编程里面一个非常重要的概念,Go语言在语言层面天生支持并发,这也是Go语言流行的一个很...
    吴佳浩阅读 371评论 0 1
  • 并发是编程里面一个非常重要的概念,Go语言在语言层面天生支持并发,这也是Go语言流行的一个很重要的原因。 Go语言...
    雪上霜阅读 251评论 0 0
  • 导读 Channel是Golang实现并发编程非常重要的组成部分,Channel是一种内建的核心数据类型,需要使用...
    MR_Zzzzzz_阅读 801评论 0 1