GOLANG使用Context實現傳值、超時和取消的方法

這篇文章主要介紹了GOLANG使用Context實現傳值、超時和取消的方法,小編覺得挺不錯的,現在分享給大家,也給大家做個參考。一起跟隨小編過來看看吧

GO1.7之後,新增了context.Context這個package,實現goroutine的管理。

Context基本的用法參考GOLANG使用Context管理關聯goroutine

實際上,Context還有個非常重要的作用,就是設置超時。比如,如果我們有個API是這樣設計的:

type Packet interface {
  encoding.BinaryMarshaler
  encoding.BinaryUnmarshaler
}

type Stack struct {
}
func (v *Stack) Read(ctx context.Context) (pkt Packet, err error) {
  return
}

一般使用是這樣使用,創建context然後調用接口:

ctx,cancel := context.WithCancel(context.Background())
stack := &Stack{}
pkt,err := stack.Read(ctx)

那麼,它本身就可以支持取消和超時,也就是用戶如果需要取消,比如發送了SIGINT信號,程序需要退出,可以在收到信號後調用cancel:

sc := make(chan os.Signal, 0)
signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM)
go func() {
  for range sc {
    cancel()
  }
}()

如果需要超時,這個API也不用改,只需要調用前設置超時時間:

ctx,cancel := context.WithTimeout(context.Background(), 3*time.Second)
defer cancel()
pkt,err := stack.Read(ctx)

如果一個程序在運行,比如Read在等待,那麼在沒有人工干預的情況下,那就應該自己運行就好了。而人工干預,也就是需要取消,比如要升級程序了,或者需要停止服務了,都屬於這種取消操作。而超時,一般是系統的策略,因爲不能一直等下去,就需要在一定時間沒有反應時終止服務。實際上context這兩個都能支持得很好,而且還不影響Read本身的邏輯,在Read中只需要關注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被接納成爲標準庫的包的緣故了吧,非常之強大和好用,而又非常簡單。一行context,深藏功與名。

另外,Context還可以傳遞上下文的Key-Value對象,比如我們希望日誌中,相關的goroutine都打印一個簡化的CID,那麼就可以用context.WithValue,參考go-oryx-lib/logger。

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支持神馬文庫。

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章