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在等待，那么在沒有人工干預(yù)的情況下，那就應(yīng)該自己運(yùn)行就好了。而人工干預(yù)，也就是需要取消，比如要升級(jí)程序了，或者需要停止服務(wù)了，都屬于這種取消操作。而超時(shí)，一般是系統(tǒng)的策略，因?yàn)椴荒芤恢钡认氯?，就需要在一定時(shí)間沒有反應(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)庫的包的緣故了吧，非常之強(qiáng)大和好用，而又非常簡單。一行context，深藏功與名。

另外，Context還可以傳遞上下文的Key-Value對象，比如我們希望日志中，相關(guān)的goroutine都打印一個(gè)簡化的CID，那么就可以用context.WithValue，參考go-oryx-lib/logger。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。