您好,登錄后才能下訂單哦!
本篇內(nèi)容介紹了“go語言context包功能及操作使用的方法”的有關(guān)知識,在實際案例的操作過程中,不少人都會遇到這樣的困境,接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧!希望大家仔細(xì)閱讀,能夠?qū)W有所成!
我們會在用到很多東西的時候都看到context的影子,比如gin框架,比如grpc,這東西到底是做啥的?
大家都在用,沒幾個知道這是干嘛的,知其然而不知其所以然
原理說白了就是:
當(dāng)前協(xié)程取消了,可以通知所有由它創(chuàng)建的子協(xié)程退出
當(dāng)前協(xié)程取消了,不會影響到創(chuàng)建它的父級協(xié)程的狀態(tài)
擴(kuò)展了額外的功能:超時取消、定時取消、可以和子協(xié)程共享數(shù)據(jù)
這就是context包的核心原理,鏈?zhǔn)絺鬟fcontext,基于context構(gòu)造新的context
每一個 RPC 調(diào)用都應(yīng)該有超時退出的能力,這是比較合理的 API 設(shè)計
不僅僅 是超時,你還需要有能力去結(jié)束那些不再需要操作的行為
context.Context 是 Go 標(biāo)準(zhǔn)的解決方案
任何函數(shù)可能被阻塞,或者需要很長時間來完成的,都應(yīng)該有個 context.Context
在 RPC 開始的時候,使用 context.Background()
有些人把在 main() 里記錄一個 context.Background(),然后把這個放到服務(wù)器的某個變量里,然后請求來了后從這個變量里繼承 context。這么做是不對的。直接每個請求,源自自己的 context.Background() 即可。
如果你沒有 context,卻需要調(diào)用一個 context 的函數(shù)的話,用 context.TODO()
如果某步操作需要自己的超時設(shè)置的話,給它一個獨(dú)立的 sub-context(如前面的例子)
package main import ( "context" "fmt" "time" ) func main() { ctx, cancel := context.WithCancel(context.Background()) //緩沖通道預(yù)先放置10個消息 messages := make(chan int, 10) defer close(messages) for i := 0; i < 10; i++ { messages <- i } //啟動3個子協(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é)束信號的時候不再接收 func child(i int, ctx context.Context, messages <-chan int) { //基于父級的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é)程退出了
主要修改了幾個地方,再ctx向下傳遞
基于上層context再構(gòu)建當(dāng)前層級的context
監(jiān)聽context的退出信號,
這就是context包的核心原理,鏈?zhǔn)絺鬟fcontext,基于context構(gòu)造新的context
type Context interface { Deadline() (deadline time.Time, ok bool) Done() <-chan struct{} Err() error Value(key interface{}) interface{} }
Deadline返回綁定當(dāng)前context的任務(wù)被取消的截止時間;如果沒有設(shè)定期限,將返回ok == false。
Done 當(dāng)綁定當(dāng)前context的任務(wù)被取消時,將返回一個關(guān)閉的channel;如果當(dāng)前context不會被取消,將返回nil。
Err 如果Done返回的channel沒有關(guān)閉,將返回nil;如果Done返回的channel已經(jīng)關(guān)閉,將返回非空的值表示任務(wù)結(jié)束的原因。如果是context被取消,Err將返回Canceled;如果是context超時,Err將返回DeadlineExceeded。
Value 返回context存儲的鍵值對中當(dāng)前key對應(yīng)的值,如果沒有對應(yīng)的key,則返回nil。
實現(xiàn)了context接口,emptyCtx沒有超時時間,不能取消,也不能存儲額外信息,所以emptyCtx用來做根節(jié)點(diǎn),一般用Background和TODO來初始化emptyCtx
通常用于主函數(shù),初始化以及測試,作為頂層的context
context.Background()
不確定使用什么用context的時候才會使用
type valueCtx struct{ Context key, val interface{} }
valueCtx利用Context的變量來表示父節(jié)點(diǎn)context,所以當(dāng)前context繼承了父context的所有信息
valueCtx還可以存儲鍵值。
可以向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} }
添加鍵值會返回創(chuàng)建一個新的valueCtx子節(jié)點(diǎn)
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)存儲的key
如果當(dāng)前的context不存在需要的key,會沿著context鏈向上尋找key對應(yīng)的值,直到根節(jié)點(diǎn)
package main import ( "context" "fmt" "time" ) func main() { ctx := context.WithValue(context.Background(), "name1", "root1") //第一層 go func(parent context.Context) { ctx = context.WithValue(parent, "name2", "root2") //第二層 go func(parent context.Context) { ctx = context.WithValue(parent, "name3", "root3") //第三層 go func(parent context.Context) { //可以獲取所有的父類的值 fmt.Println(ctx.Value("name1")) fmt.Println(ctx.Value("name2")) fmt.Println(ctx.Value("name3")) //不存在 fmt.Println(ctx.Value("name4")) }(ctx) }(ctx) }(ctx) time.Sleep(1 * time.Second) fmt.Println("end") }
運(yùn)行結(jié)果
可以看到,子context是可以獲取所有父級設(shè)置過的key
用來創(chuàng)建一個可取消的context,返回一個context和一個CancelFunc,調(diào)用CancelFunc可以觸發(fā)cancel操作。
package main import ( "context" "fmt" "time" ) func main() { ctx, cancel := context.WithCancel(context.Background()) //第一層 go func(parent context.Context) { ctx, _ := context.WithCancel(parent) //第二層 go func(parent context.Context) { ctx, _ := context.WithCancel(parent) //第三層 go func(parent context.Context) { waitCancel(ctx, 3) }(ctx) waitCancel(ctx, 2) }(ctx) waitCancel(ctx, 1) }(ctx) // 主線程給的結(jié)束時間 time.Sleep(2 * time.Second) cancel() // 調(diào)用取消context time.Sleep(1 * time.Second) } func waitCancel(ctx context.Context, i int) { for { time.Sleep(time.Second) select { case <-ctx.Done(): fmt.Printf("%d end\n", i) return default: fmt.Printf("%d do\n", i) } } }
結(jié)果:
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{} }
返回一個基于parent的可取消的context,并且過期時間deadline不晚于所設(shè)置時間d
創(chuàng)建一個定時取消context,和WithDeadline差不多,WithTimeout是相對時間
timerCtx是基于cancelCtx的context精英,是一種可以定時取消的context,過期時間的deadline不晚于所設(shè)置的時間d
示例:
package main import ( "context" "fmt" "time" ) func main() { // 設(shè)置超時時間 ctx, _ := context.WithTimeout(context.Background(), 2*time.Second) //第一層 go func(parent context.Context) { ctx, _ := context.WithCancel(parent) //第二層 go func(parent context.Context) { ctx, _ := context.WithCancel(parent) //第三層 go func(parent context.Context) { waitCancel(ctx, 3) }(ctx) waitCancel(ctx, 2) }(ctx) waitCancel(ctx, 1) }(ctx) <-ctx.Done() // 給時間調(diào)用end time.Sleep(time.Second) } func waitCancel(ctx context.Context, i int) { for { time.Sleep(time.Second) select { case <-ctx.Done(): fmt.Printf("%d end\n", i) return default: fmt.Printf("%d do\n", i) } } }
運(yùn)行結(jié)果:
1 do 3 do 2 do 1 end 3 end 2 end
可以看到,雖然我們沒有調(diào)用cancel方法,5秒后自動調(diào)用了,所有的子goroutine都已經(jīng)收到停止信號
Done方法返回一個channel
外部通過調(diào)用<-channel監(jiān)聽cancel方法
cancel方法會調(diào)用close(channel)
當(dāng)調(diào)用close方法的時候,所有的channel再次從通道獲取內(nèi)容,會返回零值和false
res,ok := <-done:
過期自動取消,使用了time.AfterFunc方法,到時調(diào)用cancel方法
c.timer = time.AfterFunc(dur, func() { c.cancel(true, DeadlineExceeded) })
“go語言context包功能及操作使用的方法”的內(nèi)容就介紹到這里了,感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識可以關(guān)注億速云網(wǎng)站,小編將為大家輸出更多高質(zhì)量的實用文章!
免責(zé)聲明:本站發(fā)布的內(nèi)容(圖片、視頻和文字)以原創(chuàng)、轉(zhuǎn)載和分享為主,文章觀點(diǎn)不代表本網(wǎng)站立場,如果涉及侵權(quán)請聯(lián)系站長郵箱:is@yisu.com進(jìn)行舉報,并提供相關(guān)證據(jù),一經(jīng)查實,將立刻刪除涉嫌侵權(quán)內(nèi)容。