您好,登錄后才能下訂單哦!
本篇內(nèi)容介紹了“Golang語(yǔ)言中Context的使用方法”的有關(guān)知識(shí),在實(shí)際案例的操作過(guò)程中,不少人都會(huì)遇到這樣的困境,接下來(lái)就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧!希望大家仔細(xì)閱讀,能夠?qū)W有所成!
在 Golang 語(yǔ)言并發(fā)編程中,經(jīng)常會(huì)遇到監(jiān)控 goroutine 運(yùn)行結(jié)束的場(chǎng)景,通常我們會(huì)想到使用 WaitGroup 和 chan + select,其中 WaitGroup 用于監(jiān)控一組 goroutine 是否全部運(yùn)行結(jié)束,chan + select 用于監(jiān)控一個(gè) goroutine 是否運(yùn)行結(jié)束(取消一個(gè) goroutine)。
如果我們需要監(jiān)控多個(gè) goroutine 是否運(yùn)行結(jié)束(取消多個(gè) goroutine),通常會(huì)使用 context,當(dāng)然使用 context 也可以用于監(jiān)控一個(gè) goroutine 是否運(yùn)行結(jié)束(取消一個(gè) goroutine)。我們?cè)谥暗奈恼乱呀?jīng)介紹過(guò) Golang 語(yǔ)言標(biāo)準(zhǔn)庫(kù) Context,未閱讀的讀者朋友們可以按需翻閱。本文我們主要介紹 Context 的一些使用方式。
使用 context 取消一個(gè) goroutine,比較類(lèi)似于使用 chan + select 的方式取消一個(gè) goroutine。
示例代碼:
func main () { ctx, cancel := context.WithCancel(context.Background()) go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println("goroutine 已停止") return default: fmt.Println("goroutine 正在運(yùn)行") time.Sleep(time.Second) } } }(ctx) time.Sleep(time.Second * 5) cancel() time.Sleep(time.Second * 5) fmt.Println("main goroutine 已結(jié)束") }
閱讀上面這段代碼,我們首先使用 context.Background() 創(chuàng)建一個(gè) context 樹(shù)的根節(jié)點(diǎn),然后使用 context.WithCancel() 創(chuàng)建一個(gè)可取消的子 context 類(lèi)型的變量 ctx,作為參數(shù)傳遞給子 goroutine,用作跟蹤子 goroutine。
然后在子 goroutine 中,使用 for select 監(jiān)控 <-ctx.Done() 判斷子 goroutine 是否運(yùn)行結(jié)束。
最后使用 context.WithCancel() 返回的第二個(gè)值 CancelFunc 類(lèi)型的 cancel 變量給子 goroutine 發(fā)送取消指令。
接下來(lái),我們?cè)賮?lái)看一個(gè)使用 context 停止多個(gè) goroutine 的示例。
func main () { ctx, cancel := context.WithCancel(context.Background()) // 停止多個(gè) goroutine go worker(ctx, "節(jié)點(diǎn)一") go worker(ctx, "節(jié)點(diǎn)二") go worker(ctx, "節(jié)點(diǎn)三") time.Sleep(time.Second * 5) cancel() time.Sleep(time.Second * 5) fmt.Println("main goroutine 已結(jié)束") } func worker (ctx context.Context, node string) { for { select { case <-ctx.Done(): fmt.Println(node, "goroutine 已停止") return default: fmt.Println(node, "goroutine 正在運(yùn)行") time.Sleep(time.Second) } } }
閱讀上面這段代碼,我們使用 go 關(guān)鍵字啟動(dòng)三個(gè) worker goroutine,和上個(gè)示例一樣,首先創(chuàng)建一個(gè) context 樹(shù)的根節(jié)點(diǎn),使用第一個(gè)返回值 context 類(lèi)型的子 ctx 跟蹤每一個(gè) worker goroutine,在 worker 中使用 for seclect 監(jiān)控 <-ctx.Done() 判斷子 goroutine 是否運(yùn)行結(jié)束,最后通過(guò)調(diào)用第二個(gè)返回值 CancelFunc 類(lèi)型的 cancel 給子 goroutine 發(fā)送取消指令,此時(shí)所有子 context 都會(huì)接收到取消指令,goroutine 結(jié)束運(yùn)行。
我們?cè)谇懊娴氖纠惺褂?WithCancel 函數(shù),用作取消 context,除此之外,可用作取消 Context 的函數(shù)還有 WithDeadline 函數(shù)和 WithTimeout 函數(shù),分別用于定時(shí)取消和超時(shí)取消,限于篇幅,本文不再贅述,感興趣的讀者可以查閱官方標(biāo)準(zhǔn)庫(kù)文檔。除了上述三個(gè)函數(shù)外,還有一個(gè) WithValue 函數(shù),它是用作上下文信息傳遞的一個(gè)函數(shù)。
在 Golang 語(yǔ)言中,Context 包還有一個(gè)重要的作用,就是用作上下文信息傳遞,接下來(lái)我們介紹一下如何使用 WithValue 函數(shù)傳遞上下文信息。
示例代碼:
func main () { ctx, cancel := context.WithCancel(context.Background()) // 傳遞上下文信息 ctxValue := context.WithValue(ctx, "uid", 1) go func(ctx context.Context) { for { select { case <-ctx.Done(): fmt.Println(ctx.Value("uid"), "goroutine 已停止") return default: fmt.Println("goroutine 正在運(yùn)行") time.Sleep(time.Second) } } }(ctxValue) time.Sleep(time.Second * 5) cancel() time.Sleep(time.Second * 5) fmt.Println("main goroutine 已結(jié)束") }
閱讀上面這段代碼,我們使用 WithValue 函數(shù)給子 goroutine 傳遞上下文信息 uid。WithValue 函數(shù)接收三個(gè)參數(shù),分別是 parent context,key 和 value。返回值是一個(gè) context,我們可以在子 goroutine 中調(diào)用 Value 方法獲取傳遞的上下文信息。
本文我們簡(jiǎn)述了監(jiān)控 goroutine 的幾種方式,分別是 WaitGroup,chan + select 和 context。重點(diǎn)介紹了 context 的一些使用方式,分別是取消一個(gè) goroutine,取消多個(gè) goroutine 和傳遞上下文信息。關(guān)于定時(shí)取消和超時(shí)取消,感興趣的讀者可以參閱官方標(biāo)準(zhǔn)庫(kù)文檔。
“Golang語(yǔ)言中Context的使用方法”的內(nèi)容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注億速云網(wǎng)站,小編將為大家輸出更多高質(zhì)量的實(shí)用文章!
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場(chǎng),如果涉及侵權(quán)請(qǐng)聯(lián)系站長(zhǎng)郵箱:is@yisu.com進(jìn)行舉報(bào),并提供相關(guān)證據(jù),一經(jīng)查實(shí),將立刻刪除涉嫌侵權(quán)內(nèi)容。