GOLANG使用Context實現(xiàn)傳值、超時和取消的方法
GO1.7之后,新增了context.Context這個package,實現(xiàn)goroutine的管理。
Context基本的用法參考GOLANG使用Context管理關(guān)聯(lián)goroutine。
實際上,Context還有個非常重要的作用,就是設(shè)置超時。比如,如果我們有個API是這樣設(shè)計的:
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)
那么,它本身就可以支持取消和超時,也就是用戶如果需要取消,比如發(fā)送了SIGINT信號,程序需要退出,可以在收到信號后調(diào)用cancel:
sc := make(chan os.Signal, 0) signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM) go func() { for range sc { cancel() } }()
如果需要超時,這個API也不用改,只需要調(diào)用前設(shè)置超時時間:
ctx,cancel := context.WithTimeout(context.Background(), 3*time.Second) defer cancel() pkt,err := stack.Read(ctx)
如果一個程序在運(yùn)行,比如Read在等待,那么在沒有人工干預(yù)的情況下,那就應(yīng)該自己運(yùn)行就好了。而人工干預(yù),也就是需要取消,比如要升級程序了,或者需要停止服務(wù)了,都屬于這種取消操作。而超時,一般是系統(tǒng)的策略,因為不能一直等下去,就需要在一定時間沒有反應(yīng)時終止服務(wù)。實際上context這兩個都能支持得很好,而且還不影響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)庫的包的緣故了吧,非常之強(qiáng)大和好用,而又非常簡單。一行context,深藏功與名。
另外,Context還可以傳遞上下文的Key-Value對象,比如我們希望日志中,相關(guān)的goroutine都打印一個簡化的CID,那么就可以用context.WithValue,參考go-oryx-lib/logger。
以上就是本文的全部內(nèi)容,希望對大家的學(xué)習(xí)有所幫助,也希望大家多多支持腳本之家。
相關(guān)文章
Golang http包構(gòu)建RESTful API的實現(xiàn)
在Go語言中實現(xiàn)RESTful API可以利用標(biāo)準(zhǔn)庫net/http提供的功能,它允許你輕松地創(chuàng)建和處理HTTP請求,本文主要介紹了Golang http包構(gòu)建RESTful API的實現(xiàn),感興趣的可以了解一下2024-01-01Go Excelize API源碼解讀GetSheetViewOptions與SetPageLayo
這篇文章主要為大家介紹了Go Excelize API源碼解讀GetSheetViewOptions與SetPageLayout方法示例,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2022-08-08golang 實現(xiàn)tcp轉(zhuǎn)發(fā)代理的方法
今天小編就為大家分享一篇golang 實現(xiàn)tcp轉(zhuǎn)發(fā)代理的方法,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-08-08