GoChannel应用系列之控制协程数量

大家好，我是渔夫子。

创新互联从2013年创立，先为孟州等服务建站，孟州等地企业，进行企业商务咨询服务。为孟州企业网站制作PC+手机+微官网三网同步一站式服务解决您的所有建站问题。

今天给大家介绍一下如何利用go中的缓冲channel来控制协程的数量。如下示例：

package main

import (
    "fmt"
    "sync/atomic"
)

func main() {
    sem := make(chan struct{}, 10)
    var count int32
    for {
       //time.Sleep(5*time.Millisecond)
       go func() {
          sem <- struct{}{}
          atomic.AddInt32(&count, 1)
          defer func() {
             <-sem
             atomic.AddInt32(&count, -1)
         }()

          fmt.Printf("count is :%d\n", count)
      }()
   }
    
}

在这个示例中，我们首先在for外层初始化了一个缓冲channel：sem，该channel可以存放10个元素。在for循环中我们不停的启动协程。在协程中我们尝试往sem中发送元素，如果发送成功，就运行后面的逻辑，如果发送不成功，即sem空间已经满了，就利用channel的堵塞特性，该协程在这里堵塞等待。最后等某个协程运行完毕后，从通道sem中输出一个元素，这样就腾出来一个空间，让堵塞的一个协程继续运行了。

这样，我们在for循环中虽然一直在不停的启用协程，但能够运行的协程数量不会超过10个。这样利用了通道的阻塞特定，不用加锁就能实现控制协程数量的目的。

应用示例

在之前我们推荐过一个爬虫的开源项目：Geziyor。这个项目中就使用了这种方式来控制并发请求的数量。我们看下源代码：首先，在初始化的地方，先根据并发个数来初始化了一个缓冲通道，示例代码看源码第109行

图片

然后，在Start函数中，循环启动每一个采集的任务。源代码中的106行Start函数，如下：

图片

然后，在源代码的第220行，则启动任务协程，在任务协程中首先通过acquireSem函数往通道中发送一个消息，如果发送不成功就堵塞在这里，直到其他协程通过releaseSem释放通道中的一个空间出来。如下：

图片

好了，今天缓冲通道的其中应用场景就分享到这里。

本文名称：GoChannel应用系列之控制协程数量
浏览地址：http://www.shufengxianlan.com/qtweb/news31/61281.html

网站建设、网络推广公司-创新互联，是专注品牌与效果的网站制作，网络营销seo公司；服务项目有等

广告

声明：本网站发布的内容（图片、视频和文字）以用户投稿、用户转载内容为主，如果涉及侵权请尽快告知，我们将会在第一时间删除。文章观点不代表本网站立场，如需处理请联系客服。电话：028-86922220；邮箱：631063699@qq.com。内容未经允许不得转载，或转载时需注明来源： 创新互联