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。内容未经允许不得转载,或转载时需注明来源: 创新互联