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