温馨提示×

golang控制并发数量的方法是什么

小亿
99
2023-10-20 21:18:54
栏目: 编程语言

在golang中,可以使用goroutine和channel来控制并发数量。以下是一些常用的方法:

  1. 使用goroutine和WaitGroup:可以使用sync包中的WaitGroup来实现控制并发数量。创建一个WaitGroup对象,然后在每个goroutine中调用Add方法增加计数器,执行完任务后调用Done方法减少计数器。最后,可以使用Wait方法阻塞主goroutine,直到计数器归零。
import (
    "sync"
)

func main() {
    var wg sync.WaitGroup
    maxConcurrency := 10 // 并发数量

    for i := 0; i < maxConcurrency; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            // 执行任务
        }()
    }

    wg.Wait()
}
  1. 使用channel和限制缓冲区大小:可以使用带有缓冲区的channel来控制并发数量。创建一个带有指定缓冲区大小的channel,然后在每个goroutine中向channel发送任务。channel的缓冲区大小限制了并发数量。
func main() {
    maxConcurrency := 10 // 并发数量
    taskCh := make(chan int, maxConcurrency)

    for i := 0; i < maxConcurrency; i++ {
        go func() {
            for task := range taskCh {
                // 执行任务
            }
        }()
    }

    // 发送任务到channel
    for i := 0; i < 100; i++ {
        taskCh <- i
    }

    close(taskCh)
}

这些方法都可以有效地控制并发数量,根据实际情况选择合适的方法。

0