建站
咨詢(xún)
售后
建站咨詢(xún)
新聞中心
01、介紹
在 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)。
讓客戶(hù)滿(mǎn)意是我們工作的目標(biāo)，不斷超越客戶(hù)的期望值來(lái)自于我們對(duì)這個(gè)行業(yè)的熱愛(ài)。我們立志把好的技術(shù)通過(guò)有效、簡(jiǎn)單的方式提供給客戶(hù)，將通過(guò)不懈努力成為客戶(hù)在信息化領(lǐng)域值得信任、有價(jià)值的長(zhǎng)期合作伙伴，公司提供的服務(wù)項(xiàng)目有：申請(qǐng)域名、網(wǎng)站空間、營(yíng)銷(xiāo)軟件、網(wǎng)站建設(shè)、太和網(wǎng)站維護(hù)、網(wǎng)站推廣。
如果我們需要監(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 的一些使用方式。
02、取消一個(gè) goroutine
使用 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ā)送取消指令。
03、取消多個(gè) goroutine
接下來(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)行。
04、上下文信息傳遞
我們?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 方法獲取傳遞的上下文信息。
05、總結(jié)
本文我們簡(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的使用方式
網(wǎng)址分享：http://m.fisionsoft.com.cn/article/cdiopee.html
