詳解Go語言的context包從放棄到入門
一、Context包到底是干嘛用的
我們會(huì)在用到很多東西的時(shí)候都看到context的影子，比如gin框架，比如grpc，這東西到底是做啥的？
大家都在用，沒幾個(gè)知道這是干嘛的，知其然而不知其所以然，
誰都在CRUD，誰都覺得if else就完了，有代碼能copy我也行，原理啥啥不懂不重要，反正就是一把梭
原理說白了就是：
- 當(dāng)前協(xié)程取消了，可以通知所有由它創(chuàng)建的子協(xié)程退出
- 當(dāng)前協(xié)程取消了，不會(huì)影響到創(chuàng)建它的父級(jí)協(xié)程的狀態(tài)
- 擴(kuò)展了額外的功能：超時(shí)取消、定時(shí)取消、可以給子協(xié)程共享數(shù)據(jù)
二、主協(xié)程退出通知子協(xié)程示例演示
主協(xié)程通知子協(xié)程退出
如下代碼展示了，通過一個(gè)叫done的channel通道達(dá)到了這樣的效果
package main
import (
"fmt"
"time"
)
func main() {
done := make(chan string)
//緩沖通道預(yù)先放置10個(gè)消息
messages := make(chan int, 10)
defer close(messages)
for i := 0; i < 10; i++ {
messages <- i
}
//啟動(dòng)3個(gè)子協(xié)程消費(fèi)messages消息
for i := 1; i <= 3; i++ {
go child(i, done, messages)
}
time.Sleep(3 * time.Second) //等待子協(xié)程接收一半的消息
close(done) //結(jié)束前通知子協(xié)程
time.Sleep(2 * time.Second) //等待所有的子協(xié)程輸出
fmt.Println("主協(xié)程結(jié)束")
}
//從messages通道獲取信息，當(dāng)收到結(jié)束信號(hào)的時(shí)候不再接收
func child(i int, done <-chan string, messages <-chan int) {
Consume:
for {
time.Sleep(1 * time.Second)
select {
case <-done:
fmt.Printf("[%d]被主協(xié)程通知結(jié)束...\n", i)
break Consume
default:
fmt.Printf("[%d]接收消息: %d\n", i, <-messages)
}
}
}
運(yùn)行結(jié)束如下
這里，我們用一個(gè)channel的關(guān)閉做到了通知所有的消費(fèi)到一半的子協(xié)程退出。
問題來了，如果子協(xié)程又要啟動(dòng)它的子協(xié)程，這可咋整？
主協(xié)程通知有子協(xié)程，子協(xié)程又有多個(gè)子協(xié)程
這是可哲學(xué)問題，我們還是得建立一個(gè)叫done的channel來監(jiān)測(cè)
下面演示一下這種操作，再在每個(gè)child方法里啟動(dòng)多個(gè)job，如下
全量代碼貼出來
package main
import (
"fmt"
"time"
)
func main() {
done := make(chan string)
//緩沖通道預(yù)先放置10個(gè)消息
messages := make(chan int, 10)
defer close(messages)
for i := 0; i < 10; i++ {
messages <- i
}
//啟動(dòng)3個(gè)子協(xié)程消費(fèi)messages消息
for i := 1; i <= 3; i++ {
go child(i, done, messages)
}
time.Sleep(3 * time.Second) //等待子協(xié)程接收一半的消息
close(done) //結(jié)束前通知子協(xié)程
time.Sleep(2 * time.Second) //等待所有的子協(xié)程輸出
fmt.Println("主協(xié)程結(jié)束")
}
//從messages通道獲取信息，當(dāng)收到結(jié)束信號(hào)的時(shí)候不再接收
func child(i int, done <-chan string, messages <-chan int) {
newDone := make(chan string)
defer close(newDone)
go childJob(i, "a", newDone)
go childJob(i, "b", newDone)
Consume:
for {
time.Sleep(1 * time.Second)
select {
case <-done:
fmt.Printf("[%d]被主協(xié)程通知結(jié)束...\n", i)
break Consume
default:
fmt.Printf("[%d]接收消息: %d\n", i, <-messages)
}
}
}
//任務(wù)
func childJob(parent int, name string, done <-chan string) {
for {
time.Sleep(1 * time.Second)
select {
case <-done:
fmt.Printf("[%d-%v]被結(jié)束...\n", parent, name)
return
default:
fmt.Printf("[%d-%v]執(zhí)行\(zhòng)n", parent, name)
}
}
}
運(yùn)行結(jié)果如下
問題來了，如果job里再啟動(dòng)自己的goroutine，這樣沒完沒了的建立done的通道有點(diǎn)惡心，這時(shí)候context包就來了！
我們先把上面的代碼改成context包的方式
package main
import (
"context"
"fmt"
"time"
)
func main() {
ctx, cancel := context.WithCancel(context.Background())
//緩沖通道預(yù)先放置10個(gè)消息
messages := make(chan int, 10)
defer close(messages)
for i := 0; i < 10; i++ {
messages <- i
}
//啟動(dòng)3個(gè)子協(xié)程消費(fèi)messages消息
for i := 1; i <= 3; i++ {
go child(i, ctx, messages)
}
time.Sleep(3 * time.Second) //等待子協(xié)程接收一半的消息
cancel() //結(jié)束前通知子協(xié)程
time.Sleep(2 * time.Second) //等待所有的子協(xié)程輸出
fmt.Println("主協(xié)程結(jié)束")
}
//從messages通道獲取信息，當(dāng)收到結(jié)束信號(hào)的時(shí)候不再接收
func child(i int, ctx context.Context, messages <-chan int) {
//基于父級(jí)的context建立context
newCtx, _ := context.WithCancel(ctx)
go childJob(i, "a", newCtx)
go childJob(i, "b", newCtx)
Consume:
for {
time.Sleep(1 * time.Second)
select {
case <-ctx.Done():
fmt.Printf("[%d]被主協(xié)程通知結(jié)束...\n", i)
break Consume
default:
fmt.Printf("[%d]接收消息: %d\n", i, <-messages)
}
}
}
//任務(wù)
func childJob(parent int, name string, ctx context.Context) {
for {
time.Sleep(1 * time.Second)
select {
case <-ctx.Done():
fmt.Printf("[%d-%v]被結(jié)束...\n", parent, name)
return
default:
fmt.Printf("[%d-%v]執(zhí)行\(zhòng)n", parent, name)
}
}
}
運(yùn)行結(jié)果如下
可以看到，改成context包還是順利的通過子協(xié)程退出了
主要修改了幾個(gè)地方，再ctx向下傳遞
基于上層context再構(gòu)建當(dāng)前層級(jí)的context
監(jiān)聽context的退出信號(hào)，
這就是context包的核心原理，鏈?zhǔn)絺鬟fcontext，基于context構(gòu)造新的context
三、Context包的核心接口和方法
更多資料可以查看：Go 語言設(shè)計(jì)與實(shí)現(xiàn)
context接口
context是一個(gè)接口，主要包含以下4個(gè)方法
- Deadline
返回當(dāng)前context任務(wù)被取消的時(shí)間，沒有設(shè)定返回ok返回false
- Done
當(dāng)綁定當(dāng)前的context任務(wù)被取消時(shí)，將返回一個(gè)關(guān)閉的channel
- Err
Done返回的channel沒有關(guān)閉，返回nil；
Done返回的channel已經(jīng)關(guān)閉，返回非空值表示任務(wù)結(jié)束的原因；
context被取消，返回Canceled。
context超時(shí)，DeadlineExceeded
- Value
返回context
存儲(chǔ)的鍵
emptyCtx結(jié)構(gòu)體
實(shí)現(xiàn)了context接口，emptyCtx沒有超時(shí)時(shí)間，不能取消，也不能存儲(chǔ)額外信息，所以emptyCtx用來做根節(jié)點(diǎn)，一般用Background和TODO來初始化emptyCtx
Backgroud
通常用于主函數(shù)，初始化以及測(cè)試，作為頂層的context
TODO
不確定使用什么用context的時(shí)候才會(huì)使用
valueCtx結(jié)構(gòu)體
type valueCtx struct{ Context key, val interface{} }
valueCtx利用Context的變量來表示父節(jié)點(diǎn)context，所以當(dāng)前context繼承了父context的所有信息
valueCtx還可以存儲(chǔ)鍵值。
Value
func (c *valueCtx) Value(key interface{}) interface{} {
if c.key == key {
return c.val
}
return c.Context.Value(key)
}
可以用來獲取當(dāng)前context和所有的父節(jié)點(diǎn)存儲(chǔ)的key
如果當(dāng)前的context不存在需要的key，會(huì)沿著context鏈向上尋找key對(duì)應(yīng)的值，直到根節(jié)點(diǎn)
WithValue
可以向context添加鍵值
func WithValue(parent Context, key, val interface{}) Context {
if key == nil {
panic("nil key")
}
if !reflect.TypeOf(key).Comparable() {
panic("key is not comparable")
}
return &valueCtx{parent, key, val}
}
添加鍵值會(huì)返回創(chuàng)建一個(gè)新的valueCtx子節(jié)點(diǎn)
cancelCtx結(jié)構(gòu)體
type cancelCtx struct {
Context
mu sync.Mutex
done chan struct{}
children map[canceler]struct{}
err error
}
type canceler interface {
cancel(removeFromParent bool, err error)
Done() <-chan struct{}
}
和valueCtx類似，有一個(gè)context做為父節(jié)點(diǎn)，
變量done表示一個(gè)channel，用來表示傳遞關(guān)閉;
children表示一個(gè)map，存儲(chǔ)了當(dāng)前context節(jié)點(diǎn)為下的子節(jié)點(diǎn)
err用來存儲(chǔ)錯(cuò)誤信息表示任務(wù)結(jié)束的原因
WithCancel
用來創(chuàng)建一個(gè)可取消的context，返回一個(gè)context和一個(gè)CancelFunc，調(diào)用CancelFunc可以觸發(fā)cancel操作。
timerCtx結(jié)構(gòu)體
timerCtx是基于cancelCtx的context精英，是一種可以定時(shí)取消的context，過期時(shí)間的deadline不晚于所設(shè)置的時(shí)間d
WithDeadline
返回一個(gè)基于parent的可取消的context，并且過期時(shí)間deadline不晚于所設(shè)置時(shí)間d
WithTimeout
創(chuàng)建一個(gè)定時(shí)取消context，和WithDeadline差不多，WithTimeout是相對(duì)時(shí)間
四、總結(jié)核心原理
- Done方法返回一個(gè)channel
- 外部通過調(diào)用<-channel監(jiān)聽cancel方法
- cancel方法會(huì)調(diào)用close(channel)
當(dāng)調(diào)用close方法的時(shí)間，所有的channel再次從通道獲取內(nèi)容，會(huì)返回零值和false
res,ok := <-done:
- 過期自動(dòng)取消，使用了time.AfterFunc方法，到時(shí)調(diào)用cancel方法
c.timer = time.AfterFunc(dur, func() {
c.cancel(true, DeadlineExceeded)
})
授人以漁不如授人以漁，知其然也知其所以然，讓我們共同構(gòu)建美麗新世界，讓人與自然更加和諧，就是這樣，giao~
到此這篇關(guān)于Go語言的context包從放棄到入門的文章就介紹到這了,更多相關(guān)Go語言context包入門內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
這篇文章主要介紹了Gin框架使用panic處理中間件問題，在 Gin 框架中，錯(cuò)誤處理和 panic 處理是非常重要的功能。當(dāng)處理 HTTP 請(qǐng)求時(shí)，可能會(huì)出現(xiàn)各種各樣的錯(cuò)誤，例如數(shù)據(jù)庫連接錯(cuò)誤、網(wǎng)絡(luò)錯(cuò)誤、權(quán)限問題等等2023-04-04
Go語言實(shí)現(xiàn)權(quán)重抽獎(jiǎng)系統(tǒng)的項(xiàng)目實(shí)踐
本文主要介紹了Go語言實(shí)現(xiàn)權(quán)重抽獎(jiǎng)系統(tǒng)的項(xiàng)目實(shí)踐,文中通過示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2025-04-04
Go的條件編譯主要通過構(gòu)建標(biāo)簽（build?tags）和構(gòu)建約束（build?constraints）來實(shí)現(xiàn),這些標(biāo)簽和約束可以讓我們針對(duì)不同的操作系統(tǒng)、架構(gòu)或特定條件編寫特定代碼,本文給大家介紹了如何使用Go語言進(jìn)行條件編譯,需要的朋友可以參考下2024-06-06
這篇文章主要介紹了golang 生成定單號(hào)的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2020-12-12
這篇文章主要介紹了Go語言使用select{}阻塞main函數(shù)介紹，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2021-04-04
接口是一種抽象類型,是對(duì)其他類型行為的概括與抽象,從語法角度來看,接口是一組方法定義的集合,下面這篇文章主要給大家介紹了關(guān)于go語言接口之接口值的相關(guān)資料,文章通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-06-06
golang方法中receiver為指針與不為指針的區(qū)別詳析
這篇文章主要給大家介紹了關(guān)于golang方法中receiver為指針與不為指針區(qū)別的相關(guān)資料，其實(shí)最大的區(qū)別應(yīng)該是指針傳遞的是對(duì)像的引用，文中通過示例代碼介紹的非常詳細(xì)，需要的朋友可以參考借鑒，下面來一起看看吧。2017-10-10
策略是一種行為設(shè)計(jì)模式,?它將一組行為轉(zhuǎn)換為對(duì)象,?并使其在原始上下文對(duì)象內(nèi)部能夠相互替換,本文就將通過代碼示例給大家詳細(xì)的介紹一下Go的策略模式,需要的朋友可以參考下2023-08-08
golang?xorm?自定義日志記錄器之使用zap實(shí)現(xiàn)日志輸出、切割日志(最新)
這篇文章主要介紹了golang?xorm?自定義日志記錄器，使用zap實(shí)現(xiàn)日志輸出、切割日志,包括連接postgresql數(shù)據(jù)庫的操作方法及?zap日志工具?，本文結(jié)合實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值，需要的朋友可以參考下2022-10-10
GoLang channel底層代碼實(shí)現(xiàn)詳解
Channel和goroutine的結(jié)合是Go并發(fā)編程的大殺器。而Channel的實(shí)際應(yīng)用也經(jīng)常讓人眼前一亮，通過與select，cancel，timer等結(jié)合，它能實(shí)現(xiàn)各種各樣的功能。接下來，我們就要梳理一下GoLang channel底層代碼實(shí)現(xiàn)2022-10-10