Go中g(shù)routine通信與context控制實(shí)例詳解
需求背景:
項(xiàng)目中需要定期執(zhí)行任務(wù)A來做一些輔助的工作,A的執(zhí)行需要在超時(shí)時(shí)間內(nèi)完成,如果本次執(zhí)行超時(shí)了,那就不對(duì)本次的執(zhí)行結(jié)果進(jìn)行處理(即放棄這次執(zhí)行)。同時(shí)A又依賴B,C兩個(gè)子任務(wù)的執(zhí)行結(jié)果。B, C之間相互獨(dú)立,可以并行的執(zhí)行。但無論B,C哪一個(gè)執(zhí)行失敗或超時(shí)都會(huì)導(dǎo)致本次任務(wù)執(zhí)行失敗。
Groutine的并發(fā)控制:
go中對(duì)于groutine的并發(fā)控制有三種解決方案:
- 通過channel控制。
父groutine中聲明無buffer的chan切片,向要開啟的子groutine中傳入切片中的一個(gè)chan
子groutine執(zhí)行完成后向這個(gè)chan中寫入數(shù)據(jù)(可以是和父groutine通信的也可以不是)
父groutine遍歷所有chan并執(zhí)行 <-chan 操作, 利用無buffer的channel只有讀寫同時(shí)準(zhǔn)備好才能執(zhí)行的特性進(jìn)行控
- WaitGroup控制。
通過sync.Waitgroup, 每開啟一個(gè)子groutine就執(zhí)行 wg.Add(1), 子groutine內(nèi)部執(zhí)行wg.Done(), 父groutine通過wg.Wait()等待所有子協(xié)程
- Context控制。
waitGroup和Context應(yīng)該是Go中較為常用的兩種并發(fā)控制。相較而言,context對(duì)于派生groutine有更強(qiáng)大的控制力,可以控制多級(jí)樹狀分布的groutine。
當(dāng)然waitGroup的子groutine也可以再開啟新的waitGroup并且等待多個(gè)孫groutine, 但是不如context的控制更加方便.
Context:
context包提供了四個(gè)方法創(chuàng)建不同類型的context
- WitchCancel()
- WithDeadline()
- WithTimeout()
- WithValue()
WithValue()主要用于通過context傳遞一些上下文消息,不在本次討論中。WithTimeout和WithDeadLine幾乎是一致的。但無論哪種,控制groutine都需要使用ctx.Done()方法. Done() 方法返回一個(gè) "只讀"的chan <-chan struct{}, 需要編寫代碼監(jiān)聽這個(gè)chan,一旦收到它的消息就說明這個(gè)context應(yīng)當(dāng)結(jié)束了,無論是到達(dá)了超時(shí)時(shí)間還是在某個(gè)地方主動(dòng)cancel()了方法。
看看代碼:
var ch1 chan int
var ch2 chan int
<br>// 任務(wù)A, 通過最外層的for來控制定期執(zhí)行
func TestMe(t *testing.T) {
ch1 = make(chan int, 0)
ch2 = make(chan int, 0)
count := 0
for {
count ++
ctx, cancel := context.WithTimeout(context.Background(), time.Second * 2)<br> // 任務(wù)A的邏輯部分,開啟子任務(wù)B, C。<br> // B,C通過ch1,ch2和A通信。<br> // 同時(shí)監(jiān)聽ctx.Done,如果超時(shí)了立即結(jié)束本次任務(wù)不繼續(xù)執(zhí)行
go func(ctx context.Context) {
go g1(ctx, count)
go g2(ctx, count)
v1, v2 := -1, -1
for v1 == -1 || v2 == -1 {
select {
case <- ctx.Done():
cancel()
fmt.Println("父級(jí)2超時(shí)退出,當(dāng)前count值為", count, "當(dāng)前時(shí)間:", time.Now())
return
case v1 = <- ch1:
case v2 = <- ch2:
}
}
fmt.Println("正常執(zhí)行完成退出, 開啟下次循環(huán),當(dāng)前count值為:", count, "當(dāng)前 v1: ", v1, "當(dāng)前 v2: ", v2)
}(ctx)<br> // 任務(wù)A監(jiān)控ctx是否到達(dá)timeOUT,timeout就終止本次執(zhí)行
select {
case <- ctx.Done():
fmt.Println("父級(jí)1超時(shí)退出,當(dāng)前count值為", count, "當(dāng)前時(shí)間:", time.Now())
}
time.Sleep(time.Second * 3)
}
}
<br>// 改進(jìn)后的任務(wù)B,即使計(jì)算出了結(jié)果,也不會(huì)再向ch1寫數(shù)據(jù)了,不會(huì)造成臟數(shù)據(jù)
func g1 (ctx context.Context, num int) {
fmt.Println("g1 num", num, "time", time.Now())
select {
case <-ctx.Done():
fmt.Println("子級(jí) g1關(guān)閉, 不向channel中寫數(shù)據(jù)")
return
default:
ch1 <- num
}
}
<br>// 改進(jìn)前的任務(wù)C
func g2 (ctx context.Context, num int) {
fmt.Println("g2 num", num, "time", time.Now())
ch2 <- num基于上述代碼,子任務(wù)B, C的處理其實(shí)有一次較大的變動(dòng)。一開始B,C都是類似于子任務(wù)C,即g2的這種寫法。
這種寫法在執(zhí)行完成后就把自身的結(jié)果交給channel, 父groutine通過channel來讀取數(shù)據(jù),正常情況下也能工作。但異常情況下,如子任務(wù)B執(zhí)行完成,子任務(wù)C(即g2)因?yàn)榫W(wǎng)絡(luò)通信等原因執(zhí)行了5s(超過context的最大時(shí)長), 就會(huì)出現(xiàn)比較嚴(yán)重的問題。到達(dá)超時(shí)時(shí)間后,A檢測到了超時(shí)就自動(dòng)結(jié)束了本次任務(wù),但g2還在執(zhí)行過程中。g2執(zhí)行完成后向ch2寫數(shù)據(jù)阻塞了(因?yàn)锳已關(guān)閉,沒有讀取ch2的groutine)。下一個(gè)循環(huán)中A再次開啟讀取ch1與ch2, 實(shí)際上讀取ch1是當(dāng)次的結(jié)果,ch2是上次任務(wù)中g(shù)2返回的結(jié)果,導(dǎo)致兩處依賴的數(shù)據(jù)源不一致。
模擬上述情況,將g2做了一些改動(dòng)如下:
// 在第3次任務(wù)重等待3s, 使得它超時(shí)<br>func g2 (ctx context.Context, num int) {
if num == 3 {
time.Sleep(time.Second * 3)
}
fmt.Println("g2 num", num, "time", time.Now())
ch2 <- num
}
實(shí)際上,如果想要通過context控制groutine, 一定要監(jiān)控Done()方法。如g1所示。相同情況下A超時(shí)退出,C仍在執(zhí)行。C執(zhí)行完成后先檢測Context是否已退出,如果已退出就不再向ch2中寫入本次的數(shù)據(jù)了。(拋磚引玉了,也可能有更好的寫法,希望大佬不吝賜教)
將g2改成和g1類似的寫法后測試結(jié)果如下:
func g2 (ctx context.Context, num int) {
if num == 3 {
time.Sleep(time.Second * 10)
fmt.Println("這次g2 超時(shí),應(yīng)當(dāng)g1, g2都不返回")
}
fmt.Println("g2 num", num, "time", time.Now())
select {
case <-ctx.Done():
fmt.Println("子級(jí) g2關(guān)閉, 不向channel中寫數(shù)據(jù)")
return
default:
ch2 <- num
}
}
總結(jié)
到此這篇關(guān)于Go中g(shù)routine通信與context控制的文章就介紹到這了,更多相關(guān)Go groutine通信與context控制內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
go-zero源碼閱讀之布隆過濾器實(shí)現(xiàn)代碼
布隆過濾器可以用于檢索一個(gè)元素是否在一個(gè)集合中。它的優(yōu)點(diǎn)是空間效率和查詢時(shí)間都比一般的算法要好的多,缺點(diǎn)是有一定的誤識(shí)別率和刪除困難,這篇文章主要介紹了go-zero源碼閱讀-布隆過濾器,需要的朋友可以參考下2023-02-02
一文總結(jié)Go語言切片核心知識(shí)點(diǎn)和坑
都說Go的切片用起來絲滑得很,Java中的List怎么用,切片就怎么用,作為曾經(jīng)的Java選手,因?yàn)榍衅氖褂貌坏卯?dāng),喜提缺陷若干,本文就給大家總結(jié)一下Go語言切片核心知識(shí)點(diǎn)和坑,需要的朋友可以參考下2023-06-06
詳解Golang如何實(shí)現(xiàn)節(jié)假日不打擾用戶
這篇文章主要為大家介紹了Golang如何實(shí)現(xiàn)節(jié)假日不打擾用戶過程詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-01-01
解析golang 標(biāo)準(zhǔn)庫template的代碼生成方法
這個(gè)項(xiàng)目的自動(dòng)生成代碼都是基于 golang 的標(biāo)準(zhǔn)庫 template 的,所以這篇文章也算是對(duì)使用 template 庫的一次總結(jié),本文通過實(shí)例代碼給大家介紹的非常詳細(xì),需要的朋友參考下吧2021-11-11
Go基礎(chǔ)教程系列之?dāng)?shù)據(jù)類型詳細(xì)說明
這篇文章主要介紹了Go基礎(chǔ)教程系列之?dāng)?shù)據(jù)類型詳細(xì)說明,需要的朋友可以參考下2022-04-04

