文章详情

短信预约-IT技能 免费直播动态提醒

请输入下面的图形验证码

提交验证

短信预约提醒成功

golang限制最大并发数的方法是什么

2023-10-26 05:13

关注

在Go语言中,可以使用`sync.WaitGroup`和`chan`来限制最大并发数。

  1. 使用`sync.WaitGroup`:`sync.WaitGroup`是一个计数器,用来等待一组goroutine完成任务。可以在每个goroutine开始执行前调用`Add()`方法增加计数器,然后在goroutine执行结束时调用`Done()`方法减少计数器,最后可以调用`Wait()`方法等待所有的goroutine完成。

以下是一个使用`sync.WaitGroup`限制最大并发数的示例代码:

package main

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

func main() {
    // 设置最大并发数为3
    maxConcurrency := 3

    // 创建一个等待组
    var wg sync.WaitGroup

    // 创建一个通道用于控制最大并发数
    sem := make(chan struct{}, maxConcurrency)

    // 创建10个任务
    for i := 0; i < 10; i++ {
        wg.Add(1)

        // 将任务添加到通道中
        sem <- struct{}{}

        go func(i int) {
            // 执行任务
            fmt.Println("Task", i, "start")
            time.Sleep(time.Second)
            fmt.Println("Task", i, "done")

            // 任务完成后从通道中移除
            <-sem

            // 完成任务
            wg.Done()
        }(i)
    }

    // 等待所有任务完成
    wg.Wait()
}
  1. 使用`chan`:`chan`是Go语言中的通道,可以用来进行goroutine之间的通信。通过创建一个指定缓冲大小的通道,可以限制最大并发数。

以下是一个使用`chan`限制最大并发数的示例代码:

package main

import (
    "fmt"
    "time"
)

func main() {
    // 设置最大并发数为3
    maxConcurrency := 3

    // 创建一个通道用于控制最大并发数
    sem := make(chan struct{}, maxConcurrency)

    // 创建10个任务
    for i := 0; i < 10; i++ {
        // 将任务添加到通道中
        sem <- struct{}{}

        go func(i int) {
            // 执行任务
            fmt.Println("Task", i, "start")
            time.Sleep(time.Second)
            fmt.Println("Task", i, "done")

            // 任务完成后从通道中移除
            <-sem
        }(i)
    }

    // 等待所有任务完成
    for i := 0; i < maxConcurrency; i++ {
        sem <- struct{}{}
    }
}

这样可以确保同一时间最多同时执行maxConcurrency个goroutine。

阅读原文内容投诉

免责声明:

① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的,并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据,供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。

② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341

软考中级精品资料免费领

  • 历年真题答案解析
  • 备考技巧名师总结
  • 高频考点精准押题
  • 2024年上半年信息系统项目管理师第二批次真题及答案解析(完整版)

    难度     801人已做
    查看
  • 【考后总结】2024年5月26日信息系统项目管理师第2批次考情分析

    难度     348人已做
    查看
  • 【考后总结】2024年5月25日信息系统项目管理师第1批次考情分析

    难度     311人已做
    查看
  • 2024年上半年软考高项第一、二批次真题考点汇总(完整版)

    难度     432人已做
    查看
  • 2024年上半年系统架构设计师考试综合知识真题

    难度     220人已做
    查看

相关文章

发现更多好内容

猜你喜欢

AI推送时光机
位置:首页-资讯-后端开发
咦!没有更多了?去看看其它编程学习网 内容吧
首页课程
资料下载
问答资讯