GOLANG使用Context實(shí)現(xiàn)傳值、超時(shí)和取消的方法
GO1.7之后,新增了context.Context這個(gè)package,實(shí)現(xiàn)goroutine的管理。
Context基本的用法參考GOLANG使用Context管理關(guān)聯(lián)goroutine。
實(shí)際上,Context還有個(gè)非常重要的作用,就是設(shè)置超時(shí)。比如,如果我們有個(gè)API是這樣設(shè)計(jì)的:
type Packet interface {
encoding.BinaryMarshaler
encoding.BinaryUnmarshaler
}
type Stack struct {
}
func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) {
return
}
一般使用是這樣使用,創(chuàng)建context然后調(diào)用接口:
ctx,cancel := context.WithCancel(context.Background())
stack := &Stack{}
pkt,err := stack.Read(ctx)
那么,它本身就可以支持取消和超時(shí),也就是用戶如果需要取消,比如發(fā)送了SIGINT信號(hào),程序需要退出,可以在收到信號(hào)后調(diào)用cancel:
sc := make(chan os.Signal, 0)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM)
go func() {
for range sc {
cancel()
}
}()
如果需要超時(shí),這個(gè)API也不用改,只需要調(diào)用前設(shè)置超時(shí)時(shí)間:
ctx,cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() pkt,err := stack.Read(ctx)
如果一個(gè)程序在運(yùn)行,比如Read在等待,那么在沒(méi)有人工干預(yù)的情況下,那就應(yīng)該自己運(yùn)行就好了。而人工干預(yù),也就是需要取消,比如要升級(jí)程序了,或者需要停止服務(wù)了,都屬于這種取消操作。而超時(shí),一般是系統(tǒng)的策略,因?yàn)椴荒芤恢钡认氯?,就需要在一定時(shí)間沒(méi)有反應(yīng)時(shí)終止服務(wù)。實(shí)際上context這兩個(gè)都能支持得很好,而且還不影響Read本身的邏輯,在Read中只需要關(guān)注context是否Done:
func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) {
select {
// case <- dataChannel: // Parse packet from data channel.
case <- ctx.Done():
return nil,ctx.Err()
}
return
}
這是為何context被接納成為標(biāo)準(zhǔn)庫(kù)的包的緣故了吧,非常之強(qiáng)大和好用,而又非常簡(jiǎn)單。一行context,深藏功與名。
另外,Context還可以傳遞上下文的Key-Value對(duì)象,比如我們希望日志中,相關(guān)的goroutine都打印一個(gè)簡(jiǎn)化的CID,那么就可以用context.WithValue,參考go-oryx-lib/logger。
以上就是本文的全部?jī)?nèi)容,希望對(duì)大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Golang中time.After的使用理解與釋放問(wèn)題
這篇文章主要給大家介紹了關(guān)于Golang中time.After的使用理解與釋放問(wèn)題,文中通過(guò)示例代碼介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2018-08-08
golang 實(shí)現(xiàn)菜單樹(shù)的生成方式
這篇文章主要介紹了golang 實(shí)現(xiàn)菜單樹(shù)的生成方式,具有很好的參考價(jià)值,希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-04-04
Go語(yǔ)言實(shí)現(xiàn)多協(xié)程文件下載器的過(guò)程詳解
這篇文章主要介紹了Go語(yǔ)言實(shí)現(xiàn)多協(xié)程文件下載器的相關(guān)資料,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2024-08-08
Air實(shí)現(xiàn)Go程序?qū)崟r(shí)熱重載使用過(guò)程解析示例
這篇文章主要為大家介紹了Air實(shí)現(xiàn)Go程序?qū)崟r(shí)熱重載使用過(guò)程解析示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步早日升職加薪2022-04-04
GoFrame框架gcache的緩存控制淘汰策略實(shí)踐示例
這篇文章主要為大家介紹了GoFrame框架gcache的緩存控制淘汰策略的實(shí)踐示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-06-06
探索Golang實(shí)現(xiàn)Redis持久化AOF實(shí)例
這篇文章主要為大家介紹了Golang實(shí)現(xiàn)Redis持久化AOF實(shí)例探索,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01

