建站
咨詢
售后
建站咨詢
新聞中心
大家好，我是漁夫子。
創(chuàng)新互聯(lián)從2013年創(chuàng)立，先為孟州等服務建站，孟州等地企業(yè)，進行企業(yè)商務咨詢服務。為孟州企業(yè)網站制作PC+手機+微官網三網同步一站式服務解決您的所有建站問題。
今天給大家介紹一下如何利用go中的緩沖channel來控制協(xié)程的數量。如下示例：
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循環(huán)中我們不停的啟動協(xié)程。在協(xié)程中我們嘗試往sem中發(fā)送元素，如果發(fā)送成功，就運行后面的邏輯，如果發(fā)送不成功，即sem空間已經滿了，就利用channel的堵塞特性，該協(xié)程在這里堵塞等待。最后等某個協(xié)程運行完畢后，從通道sem中輸出一個元素，這樣就騰出來一個空間，讓堵塞的一個協(xié)程繼續(xù)運行了。
這樣，我們在for循環(huán)中雖然一直在不停的啟用協(xié)程，但能夠運行的協(xié)程數量不會超過10個。這樣利用了通道的阻塞特定，不用加鎖就能實現控制協(xié)程數量的目的。
應用示例
在之前我們推薦過一個爬蟲的開源項目：Geziyor。這個項目中就使用了這種方式來控制并發(fā)請求的數量。我們看下源代碼：首先，在初始化的地方，先根據并發(fā)個數來初始化了一個緩沖通道，示例代碼看源碼第109行
圖片
然后，在Start函數中，循環(huán)啟動每一個采集的任務。源代碼中的106行Start函數，如下：
圖片
然后，在源代碼的第220行，則啟動任務協(xié)程，在任務協(xié)程中首先通過acquireSem函數往通道中發(fā)送一個消息，如果發(fā)送不成功就堵塞在這里，直到其他協(xié)程通過releaseSem釋放通道中的一個空間出來。如下：
圖片
好了，今天緩沖通道的其中應用場景就分享到這里。
網站題目：GoChannel應用系列之控制協(xié)程數量
標題鏈接：http://fisionsoft.com.cn/article/dpeigpc.html
