golang怎么控制并发数

   2025-02-15 4740
核心提示:在Go中,可以使用goroutine和channel来控制并发数。首先,可以使用make函数创建一个带有指定并发数的channel。例如,如果要控制

在Go中,可以使用goroutine和channel来控制并发数。

首先,可以使用make函数创建一个带有指定并发数的channel。例如,如果要控制并发数为10,可以创建一个大小为10的无缓冲channel:

concurrency := 10ch := make(chan struct{}, concurrency)

然后,在需要控制并发的地方,在执行每个任务之前,可以向channel发送一个空的struct{}值来占用一个并发槽,表示一个并发任务的启动。当任务完成后,从channel接收一个值来释放一个并发槽。

ch <- struct{}{} // 占用一个并发槽go func() {    defer func() { <-ch }() // 释放一个并发槽    // 执行并发任务的代码}()

这样就能够通过控制channel的发送和接收来控制并发数了。

完整示例代码如下:

package mainimport (    "fmt"    "time")func main() {    concurrency := 10    ch := make(chan struct{}, concurrency)    tasks := []string{"task1", "task2", "task3", "task4", "task5", "task6", "task7", "task8", "task9", "task10"}    for _, task := range tasks {        ch <- struct{}{} // 占用一个并发槽        go func(t string) {            defer func() { <-ch }() // 释放一个并发槽            fmt.Printf("开始执行任务:%s\n", t)            time.Sleep(time.Second) // 模拟任务的耗时            fmt.Printf("任务:%s 执行完成\n", t)        }(task)    }    // 等待所有任务完成    for i := 0; i < concurrency; i++ {        ch <- struct{}{}    }}

在上面的示例中,我们创建了一个包含10个并发槽的channel,然后使用goroutine来执行任务。在每个任务开始和结束时,我们输出相应的日志。最后,我们使用一个for循环来等待所有任务完成。

注意,为了确保所有的goroutine都能够执行完毕,我们在最后的for循环中再次向channel发送了10个空的struct{}值,以确保释放所有的并发槽。

这样就能够通过channel来控制并发数了。

 
 
更多>同类维修知识
推荐图文
推荐维修知识
点击排行
网站首页  |  关于我们  |  联系方式  |  用户协议  |  隐私政策  |  网站留言